We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 04340b1 commit 9926813Copy full SHA for 9926813
test/integration/optimisers.jl
@@ -24,5 +24,7 @@ end
24
gs = Reactant.to_rarray((a=rand(4), b=rand(2), c=rand(4)))
25
26
hlo = @code_hlo Optimisers.update(st_opt, ps, gs)
27
- @test length(findall("stablehlo.broadcast_in_dim", repr(hlo))) == 2
+ if !Reactant.TRACE_CALLS[]
28
+ @test length(findall("stablehlo.broadcast_in_dim", repr(hlo))) == 2
29
+ end
30
end
test/ops.jl
@@ -1154,15 +1154,17 @@ end
1154
hlo_ir = repr(mod)
1155
csts = collect(x for x in eachsplit(hlo_ir, "\n") if occursin("stablehlo.constant", x))
1156
# calls to similar give rise to dense<0> constants (that are not deduplicated):
1157
- csts = filter(x -> !occursin("dense<0>", x), csts)
1158
-
1159
- @test length(csts) == 2
1160
- idx = findfirst(x -> occursin("1, 2, 3, 4", x), csts)
1161
- @test idx !== nothing
1162
- if idx == 1
1163
- @test occursin("6, 2, 3, 4", csts[2])
1164
- else
1165
- @test occursin("6, 2, 3, 4", csts[1])
+ csts = filter(x -> !occursin("dense<0>", x), csts)
+
+ @test length(csts) == 2
+ idx = findfirst(x -> occursin("1, 2, 3, 4", x), csts)
+ @test idx !== nothing
+ if idx == 1
+ @test occursin("6, 2, 3, 4", csts[2])
+ else
1166
+ @test occursin("6, 2, 3, 4", csts[1])
1167
1168
1169
1170
0 commit comments