File tree Expand file tree Collapse file tree
MatrixAlgebraKitAMDGPUExt Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -14,13 +14,7 @@ using LinearAlgebra: BlasFloat
1414
1515include (" yarocsolver.jl" )
1616
17- function MatrixAlgebraKit. default_qr_algorithm (:: Type{T} ; kwargs... ) where {T <: StridedROCMatrix }
18- return ROCSOLVER_HouseholderQR (; kwargs... )
19- end
20- function MatrixAlgebraKit. default_lq_algorithm (:: Type{T} ; kwargs... ) where {T <: StridedROCMatrix }
21- qr_alg = ROCSOLVER_HouseholderQR (; kwargs... )
22- return LQViaTransposedQR (qr_alg)
23- end
17+ MatrixAlgebraKit. default_householder_driver (:: StridedROCMatrix{<:BlasFloat} ) = ROCSOLVER ()
2418function MatrixAlgebraKit. default_svd_algorithm (:: Type{T} ; kwargs... ) where {T <: StridedROCMatrix }
2519 return ROCSOLVER_QRIteration (; kwargs... )
2620end
Original file line number Diff line number Diff line change @@ -15,13 +15,7 @@ using LinearAlgebra: BlasFloat
1515
1616include (" yacusolver.jl" )
1717
18- function MatrixAlgebraKit. default_qr_algorithm (:: Type{T} ; kwargs... ) where {TT <: BlasFloat , T <: StridedCuVecOrMat{TT} }
19- return CUSOLVER_HouseholderQR (; kwargs... )
20- end
21- function MatrixAlgebraKit. default_lq_algorithm (:: Type{T} ; kwargs... ) where {TT <: BlasFloat , T <: StridedCuVecOrMat{TT} }
22- qr_alg = CUSOLVER_HouseholderQR (; kwargs... )
23- return LQViaTransposedQR (qr_alg)
24- end
18+ MatrixAlgebraKit. default_householder_driver (:: StridedCuMatrix{<:BlasFloat} ) = CUSOLVER ()
2519function MatrixAlgebraKit. default_svd_algorithm (:: Type{T} ; kwargs... ) where {TT <: BlasFloat , T <: StridedCuVecOrMat{TT} }
2620 return CUSOLVER_QRIteration (; kwargs... )
2721end
You can’t perform that action at this time.
0 commit comments