From f704b5759ffaca9e33cd0a644d84290f3997c266 Mon Sep 17 00:00:00 2001 From: "Gao, Xiang" Date: Mon, 11 Mar 2024 15:54:59 -0700 Subject: [PATCH] Generate predicates for cp.async.bulk normally --- csrc/device_lower/pass/unroll.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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() ||