Skip to content

T
tic
Commit f11f6b4e by Tianqi Chen Committed by GitHub
Options

prevent aggressive unrolling in vthread (#983)

parent f13421cf
Showing with 2 additions and 1 deletions
src/pass/inject_virtual_thread.cc
...@@ -418,7 +418,8 @@ class VTInjector : public IRMutator { ...@@ -418,7 +418,8 @@ class VTInjector : public IRMutator {
// reset the flags after processing. // reset the flags after processing.
vt_loop_injected_ = false; vt_loop_injected_ = false;
visit_touched_var_ = false; visit_touched_var_ = false;
if (max_loop_depth_ == 0) { // only unroll if number of vthreads are small
if (max_loop_depth_ == 0 && num_threads_ < 16) {
// do unrolling if it is inside innermost content. // do unrolling if it is inside innermost content.
Stmt blk = Substitute(stmt, {{var_, make_zero(var_.type())}}); Stmt blk = Substitute(stmt, {{var_, make_zero(var_.type())}});
for (int i = 1; i < num_threads_; ++i) { for (int i = 1; i < num_threads_; ++i) {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment