Skip to content

Commit 2ef492c

Browse files
committed
Actually run tests locally
1 parent 259fe26 commit 2ef492c

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/gemm.jl

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -559,8 +559,8 @@
559559
return C
560560
end
561561
function dense!(f::F, C, A, B) where {F}
562-
K = LoopVectorization.size(A, StaticInt(2))
563-
Kp1 = K + StaticInt(1)
562+
Kp1 = LoopVectorization.size(A, LoopVectorization.StaticInt(2))
563+
K = Kp1 - LoopVectorization.StaticInt(1)
564564
@turbo for n indices((B,C),2), m indices((A,C),1)
565565
Cmn = zero(eltype(C))
566566
for k 1:K
@@ -723,7 +723,7 @@
723723
At = copy(A');
724724
Bt = copy(B');
725725
C2 = similar(C);
726-
A2 = rand(R, M, K+1)
726+
A2 = rand(R, M, K+1);
727727
dense!(LoopVectorization.relu, C, A2, B);
728728
@test C LoopVectorization.relu.(@view(A2[:,begin:end-1]) * B .+ @view(A2[:,end]))
729729
@testset "avx $T dynamc gemm" begin

0 commit comments

Comments
 (0)