|
33 | 33 | function matmuladd!(C::AbstractMatrix, ::LoopedArrayOp, A::AbstractMatrix, |
34 | 34 | B::AbstractMatrix, bias::AbstractVector) |
35 | 35 | if unrolled_any(≤(256), (size(C, 1), size(A, 2), size(B, 2))) && |
36 | | - unrolled_all(≤(1024), (size(C, 1), size(A, 2), size(B, 2))) && |
37 | 36 | LoopVectorization.check_args(C, A, B) |
38 | 37 | __matmuladd_loopvec!(C, A, B, bias) |
39 | 38 | return |
@@ -93,7 +92,6 @@ function matmul!(C::AbstractMatrix, ::AbstractInternalArrayOpMode, |
93 | 92 | end |
94 | 93 | function matmul!(C::AbstractMatrix, ::LoopedArrayOp, A::AbstractMatrix, B::AbstractMatrix) |
95 | 94 | if unrolled_any(≤(256), (size(C, 1), size(A, 2), size(B, 2))) && |
96 | | - unrolled_all(≤(1024), (size(C, 1), size(A, 2), size(B, 2))) && |
97 | 95 | LoopVectorization.check_args(C, A, B) |
98 | 96 | __matmul_loopvec!(C, A, B) |
99 | 97 | return |
|
0 commit comments