add missing LinearAlgebra.dot(::GPUArray, ::GPUComponentArray)

jonniedie · Oct 24, 2023 · d4d2a3b · d4d2a3b
1 parent df9bd66
commit d4d2a3b
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)