From 899ea941095dcf0d0337d3516ede7a4b6b606df4 Mon Sep 17 00:00:00 2001
From: "Nicholas J. Browning" <nickjbrowning@gmail.com>
Date: Fri, 10 May 2024 17:40:12 +0200
Subject: [PATCH] fixed sap cstream

---
 mops/src/sap/sap.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mops/src/sap/sap.cu b/mops/src/sap/sap.cu
index f370265..305f9d8 100644
--- 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,