caching test

test/control_flow.jl

@@ -631,3 +631,28 @@ end
     @test length(ops) == 5 # call3, .+, .*, _call3 (2X)
 end
 
+struct Foo
+    x
+end
+struct Bar
+    x
+end
+
+_call4(foobar::Union{Foo, Bar}) = foobar.x
+function call4(foo, foo2, bar)
+    @trace _call4(foo)
+    @trace _call4(foo2)
+    @trace _call4(bar)
+end
+
+begin
+    a = rand(10)
+    b = rand(10)
+    foo = Foo(Reactant.to_rarray(a))
+    foo2 = Foo(Reactant.to_rarray(b))
+    bar = Foo(Bar(Reactant.to_rarray(b))) # typeof(foo) == typeof(bar), but these don't match!
+    ir = @code_hlo optimize=false call4(foo, foo2, bar)
+    ops = [op for op in Reactant.MLIR.IR.OperationIterator(Reactant.MLIR.IR.body(ir))]
+    @test length(ops) == 3 # call4, _call4 for {foo, foo2}, and _call4 for bar
+end
+