row/colsupport for Cholesky factors with AbstractInfUnitRange

Author: dlfivefifty

Project.toml

@@ -1,6 +1,6 @@
 name = "InfiniteLinearAlgebra"
 uuid = "cde9dba0-b1de-11e9-2c62-0bab9446c55c"
-version = "0.9.0"
+version = "0.9.1"
 
 [deps]
 ArrayLayouts = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"

src/infcholesky.jl

@@ -54,7 +54,8 @@ ArrayLayouts._cholesky(::SymTridiagonalLayout, ::NTuple{2,OneToInf{Int}}, A, ::C
 ArrayLayouts._cholesky(::SymmetricLayout{<:AbstractBandedLayout}, ::NTuple{2,OneToInf{Int}}, A, ::CNoPivot) = adaptivecholesky(A)
 
 colsupport(::AdaptiveCholeskyFactors, ::OneToInf) = OneToInf()
-rowsupport(::AdaptiveCholeskyFactors, ::OneToInf) = OneToInf()
+colsupport(F::AdaptiveCholeskyFactors, kr::AbstractInfUnitRange) = max(1,kr[1]-bandwidth(F,2)):∞
+rowsupport(::AdaptiveCholeskyFactors, kr::AbstractInfUnitRange) = kr
 
 function colsupport(F::AdaptiveCholeskyFactors, j)
     partialcholesky!(F, maximum(j)+bandwidth(F,2))

test/test_infcholesky.jl

@@ -64,6 +64,9 @@ import InfiniteLinearAlgebra: SymmetricBandedLayouts, AdaptiveCholeskyFactors
         F = cholesky(S)
         @test colsupport(F.factors,5) == rowsupport(F.factors,3) == 3:5
         @test rowsupport(F.factors) == colsupport(F.factors) == axes(F.factors,1)
+        @test rowsupport(F.factors,2:∞) == 2:∞
+        @test colsupport(F.factors,2:∞) == 1:∞
+        @test colsupport(F.factors,5:∞) == 3:∞
 
         @test (F.U * F.U')[1:10,1:10]  ≈ F.U[1:10,1:12] * F.U[1:10,1:12]'
         @test (F.U' * F.U)[1:10,1:10]  ≈ S[1:10,1:10]