Fixes #498: Support num_blocks() as an alias of grid_size() as la…

…unch config builder methods
eyalroz · Apr 23, 2023 · c4c1e88 · c4c1e88
1 parent 6b5a0db
commit c4c1e88
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/cuda/api/launch_config_builder.hpp b/src/cuda/api/launch_config_builder.hpp
@@ -441,7 +441,8 @@ class launch_config_builder_t {
 		return grid_dimensions(grid::dimensions_t{x, y, z});
 	}
 
-	launch_config_builder_t& grid_size(grid::dimension_t size) {return grid_dimensions(size, 1, 1);	}
+	launch_config_builder_t& grid_size(grid::dimension_t size) {return grid_dimensions(size, 1, 1); }
+	launch_config_builder_t& num_blocks(grid::dimension_t size) {return grid_size(size); }
 
 	launch_config_builder_t& overall_dimensions(grid::overall_dimensions_t dims)
 	{