add missing LinearAlgebra methods on GPU (#226)

jonniedie · Oct 25, 2023 · d22a56c · d22a56c · jonniedie · Oct 25, 2023
1 parent 6d4ea42
commit d22a56c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/ext/ComponentArraysGPUArraysExt.jl b/ext/ComponentArraysGPUArraysExt.jl
@@ -21,6 +21,9 @@ function Base.fill!(A::GPUComponentArray{T}, x) where {T}
 end
 
 LinearAlgebra.dot(x::GPUComponentArray, y::GPUComponentArray) = dot(getdata(x), getdata(y))
+LinearAlgebra.dot(x::GPUComponentArray, y::AbstractGPUArray) = dot(getdata(x), y)
+LinearAlgebra.dot(x::AbstractGPUArray, y::GPUComponentArray) = dot(x, getdata(y))
+
 LinearAlgebra.norm(ca::GPUComponentArray, p::Real) = norm(getdata(ca), p)
 LinearAlgebra.rmul!(ca::GPUComponentArray, b::Number) = GPUArrays.generic_rmul!(ca, b)