Skip to content

Commit e86a4b4

Browse files
committed
test different eltypes
1 parent f7c63ae commit e86a4b4

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

test/integration/omeinsum.jl

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -60,6 +60,16 @@ end
6060
c = f(a, b)
6161
c_re = @jit f(a_re, b_re)
6262
@test c c_re
63+
64+
@testset "with different eltype" begin
65+
b = Reactant.TestUtils.construct_test_array(Float32, 3, 4)
66+
b_re = Reactant.to_rarray(b)
67+
68+
f = ein"ij,jk->ik"
69+
c = f(a, b)
70+
c_re = @jit f(a_re, b_re)
71+
@test c c_re
72+
end
6373
end
6474

6575
@testset "hadamard product" begin

0 commit comments

Comments
 (0)