Skip to content

Commit eb109f8

Browse files
committed
Test 0-dropping as well as non-zero int dropping
1 parent bb2aef7 commit eb109f8

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

test/shuffleloadstores.jl

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -243,15 +243,16 @@ function my_gemm_manual_unroll(out, s::Matrix{UInt8}, V)
243243
for l in 1:k
244244
block = s[l, j]
245245
# unrolled loop
246+
thisiszero = 0
246247
p = 1
247248
Aij = (block >> (2 * (p - 1))) & 3
248-
out[4*(l - 1) + p, c] += ((Aij >= 2) + (Aij == 3)) * V[j, c]
249+
out[4*(l - 1) + p, c + thisiszero] += ((Aij >= 2) + (Aij == 3)) * V[j, c]
249250
p = 2
250251
Aij = (block >> (2 * (p - 1))) & 3
251252
out[4*(l - 1) + p, c] += ((Aij >= 2) + (Aij == 3)) * V[j, c]
252253
p = 3
253254
Aij = (block >> (2 * (p - 1))) & 3
254-
out[4*(l - 1) + p, c] += ((Aij >= 2) + (Aij == 3)) * V[j, c]
255+
out[4*(l - 1) + p + thisiszero, c] += ((Aij >= 2) + (Aij == 3)) * V[j, c]
255256
p = 4
256257
Aij = (block >> (2 * (p - 1))) & 3
257258
out[4*(l - 1) + p, c] += ((Aij >= 2) + (Aij == 3)) * V[j, c]

0 commit comments

Comments
 (0)