Skip to content

Commit 4d44420

Browse files
Fix tensorflow/core/util/gpu_kernel_helper_test.cu.cc build
1 parent 07cab8c commit 4d44420

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorflow/core/util/gpu_kernel_helper_test.cu.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -164,7 +164,7 @@ class GpuLaunchConfigTest : public ::testing::Test {
164164
cudaError_t err = cudaMallocManaged(&outbuf, sizeof(int) * bufsize);
165165
outbuf_host = outbuf;
166166
#else
167-
hipError_t err = hipMalloc(&outbuf, sizeof(int) * bufsize);
167+
hipError_t err = hipMalloc(reinterpret_cast<void**>(&outbuf), sizeof(int) * bufsize);
168168
outbuf_host = hostbuf;
169169
#endif
170170
ASSERT_EQ(cudaSuccess, err) << cudaGetErrorString(err);
@@ -331,7 +331,7 @@ TEST(CudaDeviceFunctionsTest, ShuffleGetSrcLane) {
331331
#if GOOGLE_CUDA
332332
ASSERT_EQ(cudaMallocManaged(&failure_count, sizeof(unsigned)), cudaSuccess);
333333
#else
334-
ASSERT_EQ(hipHostMalloc(&failure_count, sizeof(unsigned), 0), cudaSuccess);
334+
ASSERT_EQ(hipHostMalloc(reinterpret_cast<void**>(&failure_count), sizeof(unsigned), 0), cudaSuccess);
335335
#endif
336336
*failure_count = 0;
337337
#if TENSORFLOW_USE_ROCM

0 commit comments

Comments
 (0)