fixed sap cstream

lab-cosmo · May 10, 2024 · 899ea94 · 899ea94
1 parent e654289
commit 899ea94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mops/src/sap/sap.cu b/mops/src/sap/sap.cu
@@ -701,7 +701,7 @@ void mops::cuda::sparse_accumulation_of_products_vjp_vjp(
 
     int32_t* packed_indices = shared_array<int32_t>(indices_A.shape[0], sptr, &space);
 
-    sparse_accumulation_of_products_vjp_vjp_kernel<scalar_t><<<block_dim, thread_block, space>>>(
+    sparse_accumulation_of_products_vjp_vjp_kernel<scalar_t><<<block_dim, thread_block, space, cstream>>>(
         grad_grad_output,
         grad_A_2,
         grad_B_2,