From 80e0f547bb11d8d1b942e65a50bc99d557a04730 Mon Sep 17 00:00:00 2001 From: Zeng Jinle <32832641+sneaxiy@users.noreply.github.com> Date: Tue, 24 Sep 2019 15:03:48 +0800 Subject: [PATCH] fix allocator ut,test=develop (#19945) --- paddle/fluid/memory/allocation/retry_allocator_test.cc | 2 +- paddle/fluid/memory/detail/system_allocator_test.cc | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/memory/allocation/retry_allocator_test.cc b/paddle/fluid/memory/allocation/retry_allocator_test.cc index 670dc9ffba..11a8dfdc47 100644 --- a/paddle/fluid/memory/allocation/retry_allocator_test.cc +++ b/paddle/fluid/memory/allocation/retry_allocator_test.cc @@ -127,7 +127,7 @@ TEST(RetryAllocator, RetryAllocatorLastAllocFailure) { { platform::CUDAPlace p(0); RetryAllocator allocator(std::make_shared(p), retry_ms); - size_t allocate_size = -1UL; // Very large number + size_t allocate_size = (static_cast(1) << 40); // Very large number try { auto allocation = allocator.Allocate(allocate_size); ASSERT_TRUE(false); diff --git a/paddle/fluid/memory/detail/system_allocator_test.cc b/paddle/fluid/memory/detail/system_allocator_test.cc index ed5022cac6..34bb40d549 100644 --- a/paddle/fluid/memory/detail/system_allocator_test.cc +++ b/paddle/fluid/memory/detail/system_allocator_test.cc @@ -73,7 +73,7 @@ TEST(CUDAPinnedAllocator, Alloc) { TEST(GPUAllocator, AllocFailure) { paddle::memory::detail::GPUAllocator allocator(0); size_t index; - size_t alloc_size = -1UL; // very large size + size_t alloc_size = (static_cast(1) << 40); // Very large number try { allocator.Alloc(&index, alloc_size); ASSERT_TRUE(false);