fix errors about array index out of bounds in kgetColRowStats

修艺 · 修艺 · commit 51d3091365ab · 2023-12-09T19:02:31.000+08:00
diff --git a/csrc/kernels.cu b/csrc/kernels.cu
@@ -2178,7 +2178,12 @@ template<typename T, int THREADS, int ITEMS_PER_THREAD, int TILE_ROWS, int TILE_
   {
     //smem_col_absmax_values[threadIdx.x + (j*THREADS)] = -FLT_MAX;
     smem_row_absmax_values[threadIdx.x + (j*THREADS)] = -FLT_MAX;
-    smem_row_nnz_values[threadIdx.x + (j*THREADS)] = 0;
+    // smem_row_nnz_values[threadIdx.x + (j*THREADS)] = 0;
+  }
+
+  #pragma unroll TILE_ROWS
+  for (int j = 0; j < TILE_ROWS; j++) {
+    smem_row_nnz_values[j] = 0;
   }
 
   #pragma unroll ITEMS_PER_THREAD

Original file line number	Diff line number	Diff line change
`@@ -2178,7 +2178,12 @@ template<typename T, int THREADS, int ITEMS_PER_THREAD, int TILE_ROWS, int TILE_`
`2178`	`2178`	`{`
`2179`	`2179`	`//smem_col_absmax_values[threadIdx.x + (j*THREADS)] = -FLT_MAX;`
`2180`	`2180`	`smem_row_absmax_values[threadIdx.x + (j*THREADS)] = -FLT_MAX;`
`2181`		`- smem_row_nnz_values[threadIdx.x + (j*THREADS)] = 0;`
	`2181`	`+ // smem_row_nnz_values[threadIdx.x + (j*THREADS)] = 0;`
	`2182`	`+ }`
	`2183`	`+`
	`2184`	`+ #pragma unroll TILE_ROWS`
	`2185`	`+ for (int j = 0; j < TILE_ROWS; j++) {`
	`2186`	`+ smem_row_nnz_values[j] = 0;`
`2182`	`2187`	`}`
`2183`	`2188`
`2184`	`2189`	`#pragma unroll ITEMS_PER_THREAD`