Internal Code Change

PiperOrigin-RevId: 570296006
openxla · Oct 3, 2023 · 8ad01d1 · 8ad01d1
1 parent 3ab72e5
commit 8ad01d1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/xla/service/gpu/nccl_utils.cc b/xla/service/gpu/nccl_utils.cc
@@ -216,7 +216,7 @@ void CheckNcclAsyncError(NcclComm& lockable_comm) {
     return XLA_CUDA_STATUS(async_err);
   }();
 
-  if (!status.ok()) LOG(ERROR) << status.ToString();
+  if (!status.ok()) LOG(ERROR) << status;
 }
 
 }  // namespace

diff --git a/xla/service/gpu/precompiled_kernels.cc b/xla/service/gpu/precompiled_kernels.cc
@@ -119,7 +119,7 @@ class LazyKernel {
         static absl::once_flag logged_once;
         absl::call_once(logged_once, [&]() {
           LOG(WARNING)
-              << compiled_ptx_or.status().ToString()
+              << compiled_ptx_or.status()
               << "\nRelying on driver to perform ptx compilation. "
               << "\nSetting XLA_FLAGS=--xla_gpu_cuda_data_dir=/path/to/cuda "
               << " or modifying $PATH can be used to set the location of ptxas."