JuliaGPU/CUDA.jl

Support general eltypes in matrix factorizations (LU, QR, Cholesky)

Open

#1,458 创建于 2022年3月29日

在 GitHub 查看
 (0 评论) (0 反应) (0 负责人)Julia (274 fork)batch import
cuda librariesenhancementgood first issue

仓库指标

Star
 (1,408 star)
PR 合并指标
 (平均合并 5天 5小时) (30 天内合并 16 个 PR)

描述

Matrices with eltypes other than T <: CublasFloat should be supported in matrix factorizations (LU, QR, Cholesky) by promotion to a supported eltype during the copy that the non-mutating functions have to make anyway. This already works in many cases by reusing non-mutating methods in Base.LinearAlgebra and only specializing the mutating methods; however, Base tends to promote to Float64 when cuda would be better served by Float32, so it would be better to systematically handle this within CUDA.CUSOLVER.

Already implemented for matrix division and SVD in #1453. Submitting this issue to make sure it's not forgotten for the other factorizations.

贡献者指南