Skip to content

Commit

Permalink
[Fix][MetaSchedule] Fix redundant stages in async pipeline for mlt
Browse files Browse the repository at this point in the history
This PR fixes redundant stages if visiting `InitializeWithTuneContext`
multiple times.
  • Loading branch information
cblmemo committed Feb 28, 2023
1 parent 2b2cb96 commit 976135d
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion src/meta_schedule/schedule_rule/multi_level_tiling.cc
Original file line number Diff line number Diff line change
Expand Up @@ -96,7 +96,9 @@ void MultiLevelTilingNode::InitializeWithTuneContext(const TuneContext& context)
if (std::stoi(sm) >= 80) {
// only stage = 4 & 5 is tested. all integer that is bigger than 2
// is theoretically feasible, but no guarantee for great performance.
this->stages.insert(this->stages.end(), {4, 5});
this->stages = {4, 5};
} else {
this->stages.clear();
}
} catch (const std::invalid_argument& e) {
LOG(WARNING) << "ValueError: Unable to parse `target.arch`: " << sm
Expand Down

0 comments on commit 976135d

Please sign in to comment.