Skip to content

Commit 585a37b

Browse files
committed
test: increase batch size to > unroll limit
1 parent 9ff1b0f commit 585a37b

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

test/batching.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -80,8 +80,8 @@ function naive_batched_matmul(x, y)
8080
end
8181

8282
@testset "Naive Batched Matmul => Single Dot General" begin
83-
x = Reactant.to_rarray(Reactant.TestUtils.construct_test_array(Float32, 3, 256, 5))
84-
y = Reactant.to_rarray(Reactant.TestUtils.construct_test_array(Float32, 256, 7, 5))
83+
x = Reactant.to_rarray(Reactant.TestUtils.construct_test_array(Float32, 3, 256, 8))
84+
y = Reactant.to_rarray(Reactant.TestUtils.construct_test_array(Float32, 256, 7, 8))
8585

8686
run_auto_batching_tests(naive_batched_matmul, x, y)
8787
end

0 commit comments

Comments
 (0)