added explicit nullptr

ROCm · Jun 19, 2024 · 95465d6 · 95465d6
1 parent 616b18d
commit 95465d6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/xla/stream_executor/rocm/rocm_blas.cc b/xla/stream_executor/rocm/rocm_blas.cc
@@ -344,9 +344,9 @@ absl::Status ROCMBlas::DoBlasInternalImpl(FuncT rocblas_func, Stream *stream,
     }
   }
 #if TF_ROCM_VERSION >= 60000
-  if (auto *workspace = GetWorkspace();
-      workspace != nullptr && workspace->opaque() && workspace->size() > 0) {
-      (void)wrap::rocblas_set_workspace(blas_, workspace->opaque(), 
+  if (auto *workspace = GetWorkspace(); workspace != nullptr && 
+            workspace->opaque() != nullptr && workspace->size() > 0) {
+    (void)wrap::rocblas_set_workspace(blas_, workspace->opaque(), 
               workspace->size());
   }
 #endif