From d883ac3d9367aab480b1c8763fa8cf6489198c1d Mon Sep 17 00:00:00 2001 From: Tom Benson Date: Tue, 7 Nov 2023 16:54:14 -0500 Subject: [PATCH] Match cuBLAS sync behavior (#169) --- src/hydrogen/device/rocBLAS_API.cpp | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/hydrogen/device/rocBLAS_API.cpp b/src/hydrogen/device/rocBLAS_API.cpp index 8be85e6aa..75c5ddcdb 100644 --- a/src/hydrogen/device/rocBLAS_API.cpp +++ b/src/hydrogen/device/rocBLAS_API.cpp @@ -75,6 +75,10 @@ class ResultMgr H_CHECK_HIP(hipFreeAsync(device_, stream)); #endif // HYDROGEN_HAVE_CUB + // Sync stream to match cuBLAS behavior (cuBLAS docs here: + // https://docs.nvidia.com/cuda/cublas/#scalar-parameters) + H_CHECK_HIP(hipStreamSynchronize(stream)); + // Reset pointer mode H_CHECK_ROCBLAS(rocblas_set_pointer_mode(handle_, rocblas_pointer_mode_host)); }