More fixes

JuliaSmoothOptimizers · Nov 28, 2023 · 9fa3503 · 9fa3503
1 parent d722cfc
commit 9fa3503
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/ext/CUDA/operators.jl b/ext/CUDA/operators.jl
@@ -12,7 +12,8 @@ eltype(A::CUDA_KrylovOperator{T}) where T = T
 size(A::CUDA_KrylovOperator) = (A.m, A.n)
 
 for (SparseMatrixType, BlasType) in ((:(CuSparseMatrixCSR{T}), :BlasFloat),
-                                     (:(CuSparseMatrixCSC{T}), :BlasFloat))
+                                     (:(CuSparseMatrixCSC{T}), :BlasFloat),
+                                     (:(CuSparseMatrixCOO{T}), :BlasFloat)))
     @eval begin
         function KP.KrylovOperator(A::$SparseMatrixType; nrhs::Int=1, transa::Char='N') where T <: $BlasType
             m,n = size(A)

diff --git a/ext/KrylovPreconditionersCUDAExt.jl b/ext/KrylovPreconditionersCUDAExt.jl
@@ -4,6 +4,7 @@ using CUDA
 using CUDA.CUSPARSE
 using LinearAlgebra: checksquare, BlasReal, BlasFloat
 import LinearAlgebra: ldiv!, mul!
+import Base: size, eltype
 
 using KrylovPreconditioners
 const KP = KrylovPreconditioners