File tree Expand file tree Collapse file tree 1 file changed +10
-2
lines changed Expand file tree Collapse file tree 1 file changed +10
-2
lines changed Original file line number Diff line number Diff line change @@ -204,13 +204,21 @@ struct KernelBuilderSerializerHack {
204204
205205 result[" name" ] = builder.kernel_name_ ;
206206 result[" compile_flags" ] = expr_list_to_json (builder.compile_flags_ );
207- result[" block_size" ] = expr_list_to_json (builder.block_size_ );
208- result[" grid_size" ] = expr_list_to_json (builder.grid_size_ );
209207 result[" shared_memory" ] = expr_to_json (builder.shared_mem_ );
210208 result[" template_args" ] = expr_list_to_json (builder.template_args_ );
211209 result[" defines" ] = std::move (defines);
212210 result[" headers" ] = std::move (headers);
213211
212+ result[" block_size" ] = expr_list_to_json (std::array<Expr, 3 > {
213+ builder.determine_block_size (0 ),
214+ builder.determine_block_size (1 ),
215+ builder.determine_block_size (2 )});
216+
217+ result[" grid_size" ] = expr_list_to_json (std::array<Expr, 3 > {
218+ builder.determine_block_size (0 ),
219+ builder.determine_block_size (1 ),
220+ builder.determine_block_size (2 )});
221+
214222 return result;
215223 }
216224};
You can’t perform that action at this time.
0 commit comments