update to cuTENSOR v2

JuliaGPU · Jan 21, 2024 · 2202f41 · 2202f41
1 parent 724a6c7
commit 2202f41
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/simple_contraction.jl b/examples/simple_contraction.jl
@@ -107,7 +107,7 @@ function main()
     )
 
     # Executing the tensor contraction.
-    cuTENSOR.contraction!(
+    cuTENSOR.contract!(
         1,
         (A), tensorModes[2], cuTENSOR.CUTENSOR_OP_IDENTITY,
         (B), tensorModes[3], cuTENSOR.CUTENSOR_OP_IDENTITY,

diff --git a/test/contraction.jl b/test/contraction.jl
@@ -88,7 +88,7 @@ function testContraction(extents, tensorModes)
     )
 
     # Executing the tensor contraction.
-    cuTENSOR.contraction!(
+    cuTENSOR.contract!(
         1,
         (A), tensorModes[2], cuTENSOR.CUTENSOR_OP_IDENTITY,
         (B), tensorModes[3], cuTENSOR.CUTENSOR_OP_IDENTITY,