compat api

ROCm · Jan 10, 2024 · c7cb061 · c7cb061
1 parent 3752ebf
commit c7cb061
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/xla/stream_executor/gpu/gpu_cudamallocasync_allocator.cc b/xla/stream_executor/gpu/gpu_cudamallocasync_allocator.cc
@@ -95,9 +95,14 @@ namespace stream_executor {
 #if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
 static std::string GetCudaErrorMessage(gpu::GpuStatus result) {
   const char* error;
-  GpuGetErrorString(result, &error);
   const char* name;
+#if GOOGLE_CUDA
+  GpuGetErrorString(result, &error);
   GpuGetErrorName(result, &name);
+#elif
+  error = GpuGetErrorString(result);
+  name = GpuGetErrorName(result);
+#endif
   return absl::StrCat("CUDA error: ", error ? error : "<unknown>", " (",
                       name ? name : "Unknown", ")");
 }