add one test

under-Peter · Apr 19, 2022 · d0dc91b · d0dc91b
1 parent b5e06ca
commit d0dc91b
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/test/cueinsum.jl b/test/cueinsum.jl
@@ -122,4 +122,9 @@ end
     @test array_match(gradient(a->Array(einsum(EinCode(((1,2), (2,3)), ()), (a, b)))[] |> abs, a)[1], a)
     b = CUDA.randn(3,3)
     @test array_match(gradient(a->Array(einsum(EinCode(((1,2), (2,3)), ()), (a, b)))[] |> abs, a)[1], a)
+end
+
+@testset "adjoint dispatch" begin
+    u = CUDA.rand(2,2); A = CUDA.rand(2,2,3);
+    @test Array(ein"(ip,pql),qj -> ijl"(u', A, u)) ≈ ein"(ip,pql),qj -> ijl"(Array(CuArray(u')), Array(A), Array(u))
 end