From b8aff5e5e9e13bdc8077a886fedce36f528c37e1 Mon Sep 17 00:00:00 2001 From: Zeng Jinle <32832641+sneaxiy@users.noreply.github.com> Date: Wed, 25 Sep 2019 20:27:29 +0800 Subject: [PATCH] fix buddy_allocator_test, test=develop (#19967) --- paddle/fluid/memory/detail/buddy_allocator_test.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/memory/detail/buddy_allocator_test.cc b/paddle/fluid/memory/detail/buddy_allocator_test.cc index f02d204e1f8..dce48ac9ee8 100644 --- a/paddle/fluid/memory/detail/buddy_allocator_test.cc +++ b/paddle/fluid/memory/detail/buddy_allocator_test.cc @@ -220,7 +220,7 @@ TEST(BuddyAllocator, AllocFromAvailableWhenFractionIsOne) { FLAGS_reallocate_gpu_memory_in_mb = 0; void* p = nullptr; - EXPECT_TRUE(cudaMalloc(&p, static_cast(4) << 30) == cudaSuccess); + EXPECT_TRUE(cudaMalloc(&p, static_cast(3) << 30) == cudaSuccess); // BuddyAllocator should be able to alloc the remaining GPU BuddyAllocator buddy_allocator( @@ -228,7 +228,7 @@ TEST(BuddyAllocator, AllocFromAvailableWhenFractionIsOne) { platform::GpuMinChunkSize(), platform::GpuMaxChunkSize()); TestBuddyAllocator(&buddy_allocator, static_cast(1) << 30); - TestBuddyAllocator(&buddy_allocator, static_cast(5) << 30); + TestBuddyAllocator(&buddy_allocator, static_cast(2) << 30); if (p) { EXPECT_TRUE(cudaFree(p) == cudaSuccess); -- GitLab