update with files

YukeWang96 · Apr 25, 2023 · e281fba · e281fba
1 parent 07e800c
commit e281fba
Showing 1 changed file with 9 additions and 4 deletions.
diff --git a/include/neighbor_utils.cuh b/include/neighbor_utils.cuh
@@ -2105,8 +2105,11 @@ void SAG_host_UVM_updated(float* d_out,
 {
 
     // d_output, d_input, d_row_ptr, d_col_ind, lb_src, ub_src, dim.
-    const int partSize = 2;
-    const int warpPerBlock = 32;
+    // const int partSize = 2;
+    // const int warpPerBlock = 32;
+
+    const int partSize = 16;
+    const int warpPerBlock = 1;
 
     const nidType block = warpPerBlock * WARP_SIZE;
     const nidType grid = ub_src - lb_src;
@@ -2145,8 +2148,10 @@ void GIN_host_UVM_updated(float* d_out,
 {
 
     // d_output, d_input, d_row_ptr, d_col_ind, lb_src, ub_src, dim.
-    const int partSize = 2;
-    const int warpPerBlock = 32;
+    // const int partSize = 2;
+    // const int warpPerBlock = 32;
+    const int partSize = 32;
+    const int warpPerBlock = 1;
 
     const nidType block = warpPerBlock * WARP_SIZE;
     const nidType grid = ub_src - lb_src;