Disable workspace setting

ROCm · Aug 28, 2024 · dd85325 · dd85325
1 parent be97509
commit dd85325
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/xla/stream_executor/rocm/rocm_blas.cc b/xla/stream_executor/rocm/rocm_blas.cc
@@ -362,12 +362,12 @@ absl::Status ROCMBlas::DoBlasInternalImpl(FuncT rocblas_func, Stream *stream,
                  << ": " << ToString(ret);
     }
   }
-#if TF_ROCM_VERSION >= 60000
+#if 0
   {
     auto *workspace = GetWorkspace();
     auto *wptr = workspace != nullptr ? workspace->opaque() : nullptr;
     size_t wsize = workspace != nullptr ? workspace->size() : 0;
-    auto ret = wrap::rocblas_set_workspace(blas_, wptr, wsize); 
+    auto ret = wrap::rocblas_set_workspace(blas_, wptr, wsize);
     if (err_on_failure && ret != rocblas_status_success) {
       LOG(ERROR) << "Failed to set workspace before " << FuncT::kName
                  << ": " << ToString(ret);