diff --git a/csrc/device_lower/pass/unroll.cpp b/csrc/device_lower/pass/unroll.cpp index 51234493e3c..5254a087dde 100644 --- a/csrc/device_lower/pass/unroll.cpp +++ b/csrc/device_lower/pass/unroll.cpp @@ -64,7 +64,7 @@ void UnrollPass::registerReplace(Expr* reference, Expr* new_expr) { } void UnrollPass::dispatch(Expr* expr) { - if (ir_utils::isTvOp(expr) && !ir_utils::isCpAsyncBulk(expr)) { + if (ir_utils::isTvOp(expr)) { // If tv op, predicate it const auto out_tv = ir_utils::getTvOutput(expr); const bool should_predicate = !for_loops_.empty() ||