From f94693fdc2d6fdf34e02f43909802ffd713fc3cd Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Thu, 28 May 2026 14:28:24 +0000
Subject: [PATCH] Fix: add back cudaDeviceSynchronize

---
 tests/unit/test_fmha_smem_p.cu | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/unit/test_fmha_smem_p.cu b/tests/unit/test_fmha_smem_p.cu
index 001836d7..87368f76 100644
--- a/tests/unit/test_fmha_smem_p.cu
+++ b/tests/unit/test_fmha_smem_p.cu
@@ -190,6 +190,7 @@ int main() {
 
     cudaError_t launch_err = cudaGetLastError();
     if (launch_err != cudaSuccess) { printf("LAUNCH ERROR: %s\n", cudaGetErrorString(launch_err)); return 1; }
+    cudaError_t err = cudaDeviceSynchronize();
     if (err != cudaSuccess) { printf("CUDA ERROR: %s\n", cudaGetErrorString(err)); return 1; }
 
     cudaMemcpy(h_o, d_o, HD*sizeof(bf16_t), cudaMemcpyDeviceToHost);