Precompile Matrix*Vector

JuliaLinearAlgebra · Jan 26, 2023 · 033dd22 · 033dd22
1 parent 946e0de
commit 033dd22
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 8 deletions.
diff --git a/src/BandedMatrices.jl b/src/BandedMatrices.jl
@@ -95,4 +95,14 @@ include("interfaceimpl.jl")
 
 # _precompile_()
 
+# precompile instructions
+let B = BandedMatrix(0=>zeros(0)), v = zeros(size(B,2))
+    BT = typeof(B)
+    vT = typeof(v)
+    @assert precompile(+, (BT, BT))
+    @assert precompile(-, (BT,))
+    @assert precompile(-, (BT, BT))
+    @assert precompile(*, (BT, vT))
+end
+
 end #module
diff --git a/src/banded/BandedMatrix.jl b/src/banded/BandedMatrix.jl
@@ -825,11 +825,3 @@ end
 
 bandedbroadcaststyle(_) = BandedStyle()
 BroadcastStyle(::Type{<:BandedMatrix{<:Any,Dat}}) where Dat = bandedbroadcaststyle(BroadcastStyle(Dat))
-
-# precompile instructions
-let B = BandedMatrix(0=>zeros(0))
-    BT = typeof(B)
-    @assert precompile(+, (BT, BT))
-    @assert precompile(-, (BT,))
-    @assert precompile(-, (BT, BT))
-end