sort big ilp nl to get the same results

brucefan1983 · Jan 7, 2025 · 6410e40 · 6410e40
1 parent 24b42f8
commit 6410e40
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/force/neighbor.cu b/src/force/neighbor.cu
@@ -499,6 +499,10 @@ void find_neighbor_ilp(
   const int MN = NL.size() / NN.size();
   gpu_sort_neighbor_list_ilp<<<N, min(1024, MN), MN * sizeof(int)>>>(N, NN.data(), NL.data());
   GPU_CHECK_KERNEL
+
+  const int big_ilp_MN = big_ilp_NL.size() / big_ilp_NN.size();
+  gpu_sort_neighbor_list<<<N, big_ilp_MN, big_ilp_MN * sizeof(int)>>>(N, big_ilp_NN.data(), big_ilp_NL.data());
+  GPU_CHECK_KERNEL
 }
 
 static __global__ void gpu_find_neighbor_ON1_SW(