vifelse for masked calls

JuliaSIMD · Jan 4, 2023 · c3dbca5 · c3dbca5 · chriselrod · Jan 5, 2023
1 parent ac31711
commit c3dbca5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "LoopVectorization"
 uuid = "bdcacae8-1622-11e9-2a5c-532679323890"
 authors = ["Chris Elrod <[email protected]>"]
-version = "0.12.144"
+version = "0.12.145"
 
 [deps]
 ArrayInterface = "4fba245c-0d91-5ea0-9b3e-6abc04ee57a9"

diff --git a/src/codegen/lower_compute.jl b/src/codegen/lower_compute.jl
@@ -166,7 +166,8 @@ function ifelselastexpr(hasf::Bool, M::Int, vargtypes, K::Int, S::Int, maskearly
   end
   for m ∈ start:M
     call = if hasf
-      (maskearly | (m == M)) ? Expr(:call, :ifelse, :f, :m) : Expr(:call, :f)
+      (maskearly | (m == M)) ? Expr(:call, VectorizationBase.vifelse, :f, :m) :
+      Expr(:call, :f)
     else# m == M because !hasf
       Expr(:call, :ifelse, :m)
     end