add stream sync

sneaxiy · Dec 12, 2023 · c0e4c7c · c0e4c7c
1 parent 3897a46
commit c0e4c7c
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/test_aadiff.cu b/test_aadiff.cu
@@ -183,6 +183,7 @@ class Matrix {
     std::vector<float> out(height_ * width_);
     resource.SyncStream();
     CUDA_CHECK(cudaMemcpyAsync(out.data(), data_, nbytes, cudaMemcpyDeviceToHost, resource.stream)); 
+    resource.SyncStream();
     return out;
   }