We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 26b3cec commit 06a6219Copy full SHA for 06a6219
src/autoschedulers/mullapudi2016/AutoSchedule.cpp
@@ -3302,8 +3302,8 @@ void Partitioner::generate_group_cpu_schedule(
3302
}
3303
3304
// Find the level at which group members will be computed.
3305
- internal_assert(dims.size() > outer_dims.size());
3306
- const auto tile_inner_index = dims.size() - outer_dims.size() - 1;
+ internal_assert(dims.size() >= outer_dims.size());
+ const auto tile_inner_index = std::max(int(dims.size() - outer_dims.size()) - 1, 0);
3307
VarOrRVar tile_inner_var(Var::outermost());
3308
if (!outer_dims.empty()) {
3309
string var_name = get_base_name(dims[tile_inner_index].var);
0 commit comments