diff --git a/NDTensors/ext/NDTensorsCUDAExt/permutedims.jl b/NDTensors/ext/NDTensorsCUDAExt/permutedims.jl index eeca2441f0..b7830cc576 100644 --- a/NDTensors/ext/NDTensorsCUDAExt/permutedims.jl +++ b/NDTensors/ext/NDTensorsCUDAExt/permutedims.jl @@ -1,3 +1,3 @@ function NDTensors.permutedims!!(::Type{<:CuArray}, M, perm) return permutedims(M, perm) -end \ No newline at end of file +end diff --git a/NDTensors/src/abstractarray/permutedims.jl b/NDTensors/src/abstractarray/permutedims.jl index 6fdd9b322b..4bf1e39dd8 100644 --- a/NDTensors/src/abstractarray/permutedims.jl +++ b/NDTensors/src/abstractarray/permutedims.jl @@ -1,3 +1,3 @@ function permutedims!!(::Type{<:AbstractArray}, M, perm) return @strided Mdest = permutedims(M, perm) -end \ No newline at end of file +end diff --git a/NDTensors/src/arraytensor/array.jl b/NDTensors/src/arraytensor/array.jl index c6e1b2b82f..c2950b2c20 100644 --- a/NDTensors/src/arraytensor/array.jl +++ b/NDTensors/src/arraytensor/array.jl @@ -61,6 +61,6 @@ end function permutedims!( output_array::MatrixOrArrayStorage, array::MatrixOrArrayStorage, perm, f::Function ) - output_array .= f.(output_array, permutedims!!(leaf_parenttype(array),array, perm)) + output_array .= f.(output_array, permutedims!!(leaf_parenttype(array), array, perm)) return output_array end diff --git a/NDTensors/src/dense/tensoralgebra/contract.jl b/NDTensors/src/dense/tensoralgebra/contract.jl index 44b6787992..8b764c1913 100644 --- a/NDTensors/src/dense/tensoralgebra/contract.jl +++ b/NDTensors/src/dense/tensoralgebra/contract.jl @@ -378,7 +378,10 @@ function _contract!( # we need to make sure C is permuted to the same # ordering as A B which is the inverse of props.PC if β ≠ 0 - CM = reshape(permutedims!!(leaf_parenttype(CT), CT, invperm(props.PC)), (props.dleft, props.dright)) + CM = reshape( + permutedims!!(leaf_parenttype(CT), CT, invperm(props.PC)), + (props.dleft, props.dright), + ) else # Need to copy here since we will be permuting # into C later