GPUArrays.jl
GPUArrays.jl copied to clipboard
Support `LinearAlgebra.kron` for all GPU backends
As @maleadt mentioned in https://github.com/JuliaGPU/Metal.jl/issues/422.
I re-open a new issue here.
The current LinearAlgebra.kron
only supports for CuArray
, and the other GPUArray
uses scalar indexing.
Also, the methods for Kronecker producting Transpose{T,<:AbstractGPUArray}
or Adjoint{T,<:AbstractGPUArray}
are missing.