Fix condition compile

cblas_new
liaogang 8 years ago
parent 365b457a04
commit 21b7915d91

@ -34,13 +34,6 @@ size_t align(size_t size, paddle::platform::CPUPlace place) {
return remaining == 0 ? size : size + (alignment - remaining); return remaining == 0 ? size : size + (alignment - remaining);
} }
size_t align(size_t size, paddle::platform::GPUPlace place) {
size += sizeof(paddle::memory::detail::Metadata);
size_t alignment = paddle::platform::GpuMinChunkSize();
size_t remaining = size % alignment;
return remaining == 0 ? size : size + (alignment - remaining);
}
void update_size(size_t &total_size, const size_t size) {} void update_size(size_t &total_size, const size_t size) {}
TEST(BuddyAllocator, CPUAllocation) { TEST(BuddyAllocator, CPUAllocation) {
@ -91,6 +84,13 @@ TEST(BuddyAllocator, CPUMultAlloc) {
#ifndef PADDLE_ONLY_CPU #ifndef PADDLE_ONLY_CPU
size_t align(size_t size, paddle::platform::GPUPlace place) {
size += sizeof(paddle::memory::detail::Metadata);
size_t alignment = paddle::platform::GpuMinChunkSize();
size_t remaining = size % alignment;
return remaining == 0 ? size : size + (alignment - remaining);
}
TEST(BuddyAllocator, GPUAllocation) { TEST(BuddyAllocator, GPUAllocation) {
void *p = nullptr; void *p = nullptr;

Loading…
Cancel
Save