fix issue NVIDIA#161 (warning in cub/device/dispatch/dispatch_spmv_or…

…ig.cuh)
seunghwak · May 9, 2019 · 3158210 · 3158210
1 parent c3cceac
commit 3158210
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cub/device/dispatch/dispatch_spmv_orig.cuh b/cub/device/dispatch/dispatch_spmv_orig.cuh
@@ -636,7 +636,7 @@ struct DispatchSpmv
 
             // Get max x-dimension of grid
             int max_dim_x;
-            if (CubDebug(error = cudaDeviceGetAttribute(&max_dim_x, cudaDevAttrMaxGridDimX, device_ordinal))) break;;
+            if (CubDebug(error = cudaDeviceGetAttribute(&max_dim_x, cudaDevAttrMaxGridDimX, device_ordinal))) break;
 
             // Total number of spmv work items
             int num_merge_items = spmv_params.num_rows + spmv_params.num_nonzeros;
@@ -664,12 +664,12 @@ struct DispatchSpmv
 
             // Get grid dimensions
             dim3 spmv_grid_size(
-                CUB_MIN(num_merge_tiles, max_dim_x),
+                CUB_MIN(num_merge_tiles, static_cast<unsigned int>(max_dim_x)),
                 (num_merge_tiles + max_dim_x - 1) / max_dim_x,
                 1);
 
             dim3 segment_fixup_grid_size(
-                CUB_MIN(num_segment_fixup_tiles, max_dim_x),
+                CUB_MIN(num_segment_fixup_tiles, static_cast<unsigned int>(max_dim_x)),
                 (num_segment_fixup_tiles + max_dim_x - 1) / max_dim_x,
                 1);