From 4d44420020592708b05273f5a8538304522e78e5 Mon Sep 17 00:00:00 2001 From: Dragan Mladjenovic Date: Wed, 16 Oct 2024 12:42:28 +0000 Subject: [PATCH] Fix tensorflow/core/util/gpu_kernel_helper_test.cu.cc build --- tensorflow/core/util/gpu_kernel_helper_test.cu.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tensorflow/core/util/gpu_kernel_helper_test.cu.cc b/tensorflow/core/util/gpu_kernel_helper_test.cu.cc index 6612fa5c225ddb..cdaf3dde2ea2cd 100644 --- a/tensorflow/core/util/gpu_kernel_helper_test.cu.cc +++ b/tensorflow/core/util/gpu_kernel_helper_test.cu.cc @@ -164,7 +164,7 @@ class GpuLaunchConfigTest : public ::testing::Test { cudaError_t err = cudaMallocManaged(&outbuf, sizeof(int) * bufsize); outbuf_host = outbuf; #else - hipError_t err = hipMalloc(&outbuf, sizeof(int) * bufsize); + hipError_t err = hipMalloc(reinterpret_cast(&outbuf), sizeof(int) * bufsize); outbuf_host = hostbuf; #endif ASSERT_EQ(cudaSuccess, err) << cudaGetErrorString(err); @@ -331,7 +331,7 @@ TEST(CudaDeviceFunctionsTest, ShuffleGetSrcLane) { #if GOOGLE_CUDA ASSERT_EQ(cudaMallocManaged(&failure_count, sizeof(unsigned)), cudaSuccess); #else - ASSERT_EQ(hipHostMalloc(&failure_count, sizeof(unsigned), 0), cudaSuccess); + ASSERT_EQ(hipHostMalloc(reinterpret_cast(&failure_count), sizeof(unsigned), 0), cudaSuccess); #endif *failure_count = 0; #if TENSORFLOW_USE_ROCM