Skip to content

Commit af3bfe0

Browse files
author
Ralph Kube
committed
Fixed reducedim operations for CuQRPackedQ
1 parent afe8179 commit af3bfe0

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

lib/cusolver/linalg.jl

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,13 @@ CUDA.CuMatrix(A::CuQRPackedQ) = orgqr!(copy(A.factors), A.τ)
2727
CUDA.CuArray(A::CuQRPackedQ) = CuMatrix(A)
2828
Base.Matrix(A::CuQRPackedQ) = Matrix(CuMatrix(A))
2929

30+
Base.similar(A::CuQRPackedQ{S,T}) where {S,T} = CuArray{S, ndims(a)}(undef, size(a))
31+
Base.sum(xs::CuQRPackedQ) = sum(CuArray(xs))
32+
Base.prod(xs::CuQRPackedQ) = prod(CuArray(xs))
33+
Base.maximum(xs::CuQRPackedQ) = maximum(CuArray(xs))
34+
Base.minimum(xs::CuQRPackedQ) = minimum(CuArray(xs))
35+
36+
3037
function Base.getproperty(A::CuQR, d::Symbol)
3138
m, n = size(getfield(A, :factors))
3239
if d == :R

0 commit comments

Comments
 (0)