Skip to content

Commit aad7ebe

Browse files
committed
rm patch because it is in OMEinsum.jl
1 parent 0b29f18 commit aad7ebe

File tree

1 file changed

+0
-17
lines changed

1 file changed

+0
-17
lines changed

src/cuda.jl

Lines changed: 0 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -27,20 +27,3 @@ for TT in [:(Tropical{<:NativeTypes}), :TropicalTypes]
2727
end
2828
end
2929
end
30-
31-
using .CUDA: @cartesianidx, AbstractGPUArray, gpu_call
32-
33-
@inline @generated function map_index(dest, src, I, perm::NTuple{N,T}) where {N,T}
34-
Expr(:(=), Expr(:ref, :dest, [:(@inbounds I[perm[$i]]) for i in 1:N]...), Expr(:ref, :src, :I))
35-
end
36-
function LinearAlgebra.permutedims!(dest::AbstractGPUArray, src::AbstractGPUArray,
37-
perm::NTuple)
38-
Base.checkdims_perm(dest, src, perm)
39-
function permutedims_kernel(ctx, dest, src, perm)
40-
I = @cartesianidx src
41-
map_index(dest, src, I, perm)
42-
return
43-
end
44-
gpu_call(permutedims_kernel, dest, src, perm)
45-
return dest
46-
end

0 commit comments

Comments
 (0)