Add a generic fallback for rotate! and reflect! (#430)

Author: amontoison

src/host/linalg.jl

@@ -441,6 +441,34 @@ function Base.similar(A::Hermitian{<:Any,<:AbstractGPUArray}, ::Type{T}) where T
     return Hermitian(B, ifelse(A.uplo == 'U', :U, :L))
 end
 
+## rotate
+
+function LinearAlgebra.rotate!(x::AbstractGPUArray, y::AbstractGPUArray, c::Number, s::Number)
+    gpu_call(x, y, c, s; name="rotate!") do ctx, x, y, c, s
+        i = @linearidx x
+        @inbounds xi = x[i]
+        @inbounds yi = y[i]
+        @inbounds x[i] =       c  * xi + s * yi
+        @inbounds y[i] = -conj(s) * xi + c * yi
+        return
+    end
+    return x, y
+end
+
+## reflect
+
+function LinearAlgebra.reflect!(x::AbstractGPUArray, y::AbstractGPUArray, c::Number, s::Number)
+    gpu_call(x, y, c, s; name="reflect!") do ctx, x, y, c, s
+        i = @linearidx x
+        @inbounds xi = x[i]
+        @inbounds yi = y[i]
+        @inbounds x[i] =      c  * xi + s * yi
+        @inbounds y[i] = conj(s) * xi - c * yi
+        return
+    end
+    return x, y
+end
+
 ## dot
 
 LinearAlgebra.dot(x::AbstractGPUArray, y::AbstractGPUArray) = mapreduce(dot, +, x, y)

test/testsuite/linalg.jl

@@ -230,6 +230,14 @@
         @test compare(dot, AT, rand(T,5), rand(T, 5))
     end
 
+    @testset "rotate!" for T in eltypes
+        @test compare(rotate!, AT, rand(T,5), rand(T,5), Ref(rand(real(T))), Ref(rand(T)))
+    end
+
+    @testset "reflect!" for T in eltypes
+        @test compare(reflect!, AT, rand(T,5), rand(T,5), Ref(rand(real(T))), Ref(rand(T)))
+    end
+
     @testset "iszero and isone" for T in eltypes
         A = one(AT(rand(T, 2, 2)))
         @test isone(A)