Merge pull request #1861 from cwpearson/fix/rocm-5.2.0-hang-quick

Use `rocblas_pointer_mode_host` in rocBLAS `scal`
kokkos · Jun 12, 2023 · cec953f · cec953f
2 parents 22b5f4e + 478a56b
commit cec953f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/blas/tpls/KokkosBlas1_scal_tpl_spec_decl.hpp b/blas/tpls/KokkosBlas1_scal_tpl_spec_decl.hpp
@@ -299,7 +299,7 @@ namespace Impl {
         KOKKOS_ROCBLAS_SAFE_CALL_IMPL(                                         \
             rocblas_get_pointer_mode(s.handle, &pointer_mode));                \
         KOKKOS_ROCBLAS_SAFE_CALL_IMPL(                                         \
-            rocblas_set_pointer_mode(s.handle, rocblas_pointer_mode_device));  \
+            rocblas_set_pointer_mode(s.handle, rocblas_pointer_mode_host));    \
         KOKKOS_ROCBLAS_SAFE_CALL_IMPL(ROCBLAS_FN(                              \
             s.handle, N, reinterpret_cast<const ROCBLAS_SCALAR_TYPE*>(&alpha), \
             reinterpret_cast<ROCBLAS_SCALAR_TYPE*>(R.data()), one));           \