Add a generic dot for GPUArrays (#427)

Author: amontoison

src/host/linalg.jl

@@ -441,6 +441,9 @@ function Base.similar(A::Hermitian{<:Any,<:AbstractGPUArray}, ::Type{T}) where T
     return Hermitian(B, ifelse(A.uplo == 'U', :U, :L))
 end
 
+## dot
+
+LinearAlgebra.dot(x::AbstractGPUArray, y::AbstractGPUArray) = mapreduce(dot, +, x, y)
 
 ## axp{b}y
 

test/testsuite/linalg.jl

@@ -222,11 +222,12 @@
     end
 
     @testset "axp{b}y" for T in eltypes
-        alpha, beta = 0.5, 2.0
-        x = T.([2,4,6])
-        y = T.([3,4,5])
-        @test axpby!(alpha,x,beta,y) ≈ T.([7,10,13])
-        @test axpy!(alpha,x,y) ≈ T.([8,12,16])
+        @test compare(axpby!, AT, Ref(rand(T)), rand(T,5), Ref(rand(T)), rand(T,5))
+        @test compare(axpy!, AT, Ref(rand(T)), rand(T,5), rand(T,5))
+    end
+
+    @testset "dot" for T in eltypes
+        @test compare(dot, AT, rand(T,5), rand(T, 5))
     end
 
     @testset "iszero and isone" for T in eltypes