Skip to content
This repository was archived by the owner on Feb 11, 2023. It is now read-only.

Commit 6d33f89

Browse files
committed
fix a cuda issue
1 parent 6a74859 commit 6d33f89

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

Project.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
name = "SimpleTensorNetworks"
22
uuid = "4456351a-5be3-4067-ade9-541926a41e04"
33
authors = ["GiggleLiu <[email protected]> and contributors"]
4-
version = "0.1.2"
4+
version = "0.1.3"
55

66
[deps]
77
LightGraphs = "093fc24a-ae57-5d10-9952-331d41423f4d"

src/cuda.jl

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,9 @@ function genperm(I::NTuple{N}, perm::NTuple{N}) where N
1717
end
1818

1919
function LinearAlgebra.permutedims!(dest::GPUArrays.AbstractGPUArray, src::GPUArrays.AbstractGPUArray, perm)
20+
LinearAlgebra.permutedims!(dest::GPUArrays.AbstractGPUArray, src::GPUArrays.AbstractGPUArray, Tuple(perm))
21+
end
22+
function LinearAlgebra.permutedims!(dest::GPUArrays.AbstractGPUArray, src::GPUArrays.AbstractGPUArray, perm::NTuple)
2023
perm isa Tuple || (perm = Tuple(perm))
2124
size_dest = size(dest)
2225
size_src = size(src)

0 commit comments

Comments
 (0)