Skip to content

Commit

Permalink
Fix tensorflow/core/util/gpu_kernel_helper_test.cu.cc build
Browse files Browse the repository at this point in the history
  • Loading branch information
draganmladjenovic committed Oct 16, 2024
1 parent 07cab8c commit 35a6567
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tensorflow/core/util/gpu_kernel_helper_test.cu.cc
Original file line number Diff line number Diff line change
Expand Up @@ -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<void**>(&outbuf), sizeof(int) * bufsize);
outbuf_host = hostbuf;
#endif
ASSERT_EQ(cudaSuccess, err) << cudaGetErrorString(err);
Expand Down Expand Up @@ -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<void**>(&failure_count), sizeof(unsigned), 0), cudaSuccess);
#endif
*failure_count = 0;
#if TENSORFLOW_USE_ROCM
Expand Down

0 comments on commit 35a6567

Please sign in to comment.