Skip to content

T
tic
Commit 536ba30b by Wu Zhao Committed by Tianqi Chen
Options

[LLVM CodeGen] Solve LLVM CodeGen br instruction accept not-i1 type issue (#2381)

parent b3375702
Showing with 4 additions and 2 deletions
src/pass/rewrite_unsafe_select.cc
......@@ -95,8 +95,10 @@ class UnsafeSelectRewriter : public IRMutator {
Expr expr = IRMutator::Mutate_(op, e);
op = expr.as<Select>();
UnsafeExprDetector unsafe;
if (unsafe.VisitExpr(op->true_value) ||
unsafe.VisitExpr(op->false_value)) {
bool cond_is_scalar_bool = op->condition.type().is_bool() && op->condition.type().is_scalar();
if ((unsafe.VisitExpr(op->true_value) ||
unsafe.VisitExpr(op->false_value)) &&
cond_is_scalar_bool) {
return Call::make(
op->type,
intrinsic::tvm_if_then_else,
......
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