Implement feature common to all `AbstractGPUArrays` through KernelAbstractions.jl

albertomercurio
Sun, 15 Dec 2024 10:00:43 +0000
GPU

Hello,

Let’s imagine I have a very common function

LinearAlgebra.kron(a::VT, b::VT) where VT<:AbstractGPUArray

which can support any AbstractGPUArray since I’m using KernelAbstractions.jl. Where should I implement such feature? Is there a general place? Or should I implement it on every platform as CUDA.jl, Metal.jl etc?

3 posts - 2 participants