Function TEST

tensorflow/core/common_runtime/gpu_tensorpool_allocator_test.cc:35–44

namespace {

TEST(GPUTensorPoolAllocatorTest, SmallAllocationWithAlignment32B) {
  PlatformGpuId platform_gpu_id(0);
  GPUMemAllocator* sub_allocator = new GPUMemAllocator(
      GpuIdUtil::ExecutorForPlatformGpuId(platform_gpu_id).ValueOrDie(),
      platform_gpu_id, false /*use_unified_memory*/, {}, {});
  GPUTensorPoolAllocator allocator(sub_allocator, "GPU_0_tensorpool", 1 << 30);
  void* p = allocator.AllocateRaw(32, 100);
  EXPECT_TRUE(p != nullptr);
  allocator.DeallocateRaw(p);
}

TEST(GPUTensorPoolAllocatorTest, SmallAllocationWithAlignment8B) {
  PlatformGpuId platform_gpu_id(0);