Add scalar-VecUnroll method for log_fast

JuliaSIMD · Mar 20, 2021 · 4562f05 · 4562f05
1 parent b6b86da
commit 4562f05
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "SLEEFPirates"
 uuid = "476501e8-09a2-5ece-8869-fb82de89a1fa"
 authors = ["chriselrod <[email protected]>"]
-version = "0.6.12"
+version = "0.6.13"
 
 [deps]
 IfElse = "615f187c-cbe4-4ef1-ba3b-2fcf58d6d173"

diff --git a/src/log.jl b/src/log.jl
@@ -236,4 +236,11 @@ else
     @inline log_fast(d::AbstractSIMD) = log_fast(float(d), False())
 end
 @inline log_fast(d::Union{Float32,Float64}) = log_fast(d, False())
+@generated function log_fast(x::VecUnroll{N,1,T,T}) where {N,T}
+    quote
+        $(Expr(:meta,:inline))
+        lx = log_fast(VectorizationBase.transpose_vecunroll(x))
+        VecUnroll(Base.Cartesian.@ntuple $(N+1) n -> lx(n))
+    end
+end