Skip to content

Commit b85b321

Browse files
committed
Fix bug in export.cpp when grid/block size was derived from tunable parameters
1 parent afc33a0 commit b85b321

File tree

1 file changed

+10
-2
lines changed

1 file changed

+10
-2
lines changed

src/export.cpp

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff 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
};

0 commit comments

Comments
 (0)