@@ -60,29 +60,21 @@ void copy_sweep_grid_shape(nvbench::state &state)
6060 num_values);
6161 });
6262}
63- void naive_copy_sweep_grid_shape (nvbench::state &state)
64- {
65- copy_sweep_grid_shape (state);
66- }
67- void tied_copy_sweep_grid_shape (nvbench::state &state)
68- {
69- copy_sweep_grid_shape (state);
70- }
7163
7264// ==============================================================================
7365// Naive iteration of both the BlockSize and NumBlocks axes.
7466// Will generate the full cartesian product of the two axes for a total of
7567// 16 invocations of copy_sweep_grid_shape.
76- NVBENCH_BENCH (naive_copy_sweep_grid_shape )
77- // Full combinatorial of Every power of two from 64->1024:
68+ NVBENCH_BENCH (copy_sweep_grid_shape )
69+ .set_name( " naive_copy_sweep_grid_shape " )
7870 .add_int64_axis(" BlockSize" , {32 , 64 , 128 , 256 })
7971 .add_int64_axis(" NumBlocks" , {1024 , 512 , 256 , 128 });
8072
8173// ==============================================================================
8274// Zipped iteration of BlockSize and NumBlocks axes.
8375// Will generate only 4 invocations of copy_sweep_grid_shape
84- NVBENCH_BENCH (tied_copy_sweep_grid_shape )
85- // Every power of two from 64->1024:
76+ NVBENCH_BENCH (copy_sweep_grid_shape )
77+ .set_name( " tied_copy_sweep_grid_shape " )
8678 .add_zip_axes(nvbench::int64_axis{" BlockSize" , {32 , 64 , 128 , 256 }},
8779 nvbench::int64_axis{" NumBlocks" , {1024 , 512 , 256 , 128 }});
8880
@@ -160,11 +152,8 @@ struct under_diag final : nvbench::user_axis_space
160152 }
161153};
162154
163- void user_copy_sweep_grid_shape (nvbench::state &state)
164- {
165- copy_sweep_grid_shape (state);
166- }
167- NVBENCH_BENCH (user_copy_sweep_grid_shape)
155+ NVBENCH_BENCH (copy_sweep_grid_shape)
156+ .set_name(" user_copy_sweep_grid_shape" )
168157 .add_user_iteration_axes(
169158 [](auto ... args) -> std::unique_ptr<nvbench::axis_space_base> {
170159 return std::make_unique<under_diag>(args...);
0 commit comments