fix permutedims for CuArrays and update Project.toml

under-Peter · Mar 24, 2020 · d8f824d · d8f824d
1 parent ae8e039
commit d8f824d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "OMEinsum"
 uuid = "ebe7aa44-baf0-506c-a96f-8464559b3922"
 authors = ["Andreas Peter <[email protected]>"]
-version = "0.3.1"
+version = "0.3.2"
 
 [deps]
 BatchedRoutines = "a9ab73d0-e05c-5df1-8fde-d6a4645b8d8e"

diff --git a/src/cueinsum.jl b/src/cueinsum.jl
@@ -55,3 +55,5 @@ end
         batched_contract(ixs1, xs1, ixs2, xs2, $(Val(iy)))
     end
 end
+
+tensorpermute(A::CuArray, perm) = permutedims(A, perm)
-Original file line number
+Diff line change
@@ Expand Up / @@ -55,3 +55,5 @@ end @@
             batched_contract(ixs1, xs1, ixs2, xs2, $(Val(iy)))
         end
     end
+    tensorpermute(A::CuArray, perm) = permutedims(A, perm)