Vectorize loop broadcast cond, broken in many cases

apache · Aug 26, 2019 · 1585533 · 1585533
1 parent 8a42430
commit 1585533
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/pass/vectorize_loop.cc b/src/pass/vectorize_loop.cc
@@ -220,7 +220,7 @@ class Vectorizer : public IRMutator {
       int lanes = std::max(std::max(
           cond.type().lanes(),
           t.type().lanes()), f.type().lanes());
-      return Select::make(cond, BroadcastTo(t, lanes), BroadcastTo(f, lanes));
+      return Select::make(BroadcastTo(cond, lanes), BroadcastTo(t, lanes), BroadcastTo(f, lanes));
     }
   }
   Expr Mutate_(const Cast *op, const Expr& e) final {