From 31ee8391b85cef28a38d1ebbe49794eef125b84b Mon Sep 17 00:00:00 2001
From: Berenger Bramas <Berenger.Bramas@inria.fr>
Date: Wed, 30 Oct 2024 16:54:12 +0100
Subject: [PATCH] add sync after gemm
---
Benchmark/cholesky_gemm/gemm.cpp | 1 +
1 file changed, 1 insertion(+)
diff --git a/Benchmark/cholesky_gemm/gemm.cpp b/Benchmark/cholesky_gemm/gemm.cpp
index 41c3280..c12c65e 100644
--- a/Benchmark/cholesky_gemm/gemm.cpp
+++ b/Benchmark/cholesky_gemm/gemm.cpp
@@ -116,6 +116,7 @@ auto gemm(const int NbLoops, SpBlas::Block blocksC[], const SpBlas::Block blocks
inBlockDim, inBlockDim, inBlockDim, &alphaBeta, (const double*)paramA.getRawPtr(), inBlockDim,
(const double*)paramB.getRawPtr(), inBlockDim,
&alphaBeta, (double*)paramC.getRawPtr(), inBlockDim ) );
+ CUDA_ASSERT(cudaStreamSynchronize(SpCudaUtils::GetCurrentStream()));
})
#endif
#ifdef SPECX_COMPILE_WITH_HIP
--
GitLab