Add testcase

Jutho · Oct 7, 2024 · f82aada · f82aada
1 parent cd0fd61
commit f82aada
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/test/cutensor.jl b/test/cutensor.jl
@@ -336,4 +336,14 @@ if cuTENSOR.has_cutensor()
             @test copy(C) ≈ Ccopy
         end
     end
+
+    @testset "Issues" verbose = true begin
+        @testset "Issue PR #186" begin
+            # https://github.com/Jutho/TensorOperations.jl/pull/186
+            A = randn(Float32, (5, 5, 5, 5))
+            Atr = @tensor A[a, b, b, a]
+            Atr2 = @cutensor A[a, b, b, a]
+            @test Atr ≈ Atr2
+        end
+    end
 end