Skip to content

Commit b4553f4

Browse files
committed
Attempt to make dimensions consistent
1 parent bc82c1f commit b4553f4

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

src/GramMatrix.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -78,12 +78,12 @@ function GramMatrix(μ::AbstractVector{T}, X::XT, p0::T) where {T, XT <: Abstrac
7878
W[m, 2] = (X[m-1, m]*W[m-1, 1] + (X[m, m]-X[1, 1])*W[m, 1] + X[m+1, m]*W[m+1, 1])/X[2, 1]
7979
end
8080
end
81-
@inbounds @simd for n in 3:n
82-
for m in n:min(N-n+1, b+n)
83-
W[m, n] = (X[m-1, m]*W[m-1, n-1] + (X[m, m]-X[n-1, n-1])*W[m, n-1] + X[m+1, m]*W[m+1, n-1] - X[n-2, n-1]*W[m, n-2])/X[n, n-1]
81+
@inbounds @simd for j in 3:N
82+
for m in j:min(N, b+j)
83+
W[m, j] = (X[m-1, m]*W[m-1, j-1] + (X[m, m]-X[j-1, j-1])*W[m, j-1] + (m < N ? X[m+1, m]*W[m+1, j-1] : zero(T)) - X[j-2, j-1]*W[m, j-2])/X[j, j-1]
8484
end
8585
end
86-
return GramMatrix(Symmetric(W[1:n, 1:n], :L), eval(XT.name.name)(view(X, 1:n, 1:n)))
86+
return GramMatrix(Symmetric(W, :L), X)
8787
end
8888

8989
#

test/grammatrixtests.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ using FastTransforms, BandedMatrices, LazyArrays, LinearAlgebra, Test
2929
F = cholesky(G)
3030
@test F.L*F.L' W
3131

32-
@test G[1:66,1:66] GramMatrix(W[1:5, 1], X)
32+
@test G GramMatrix(W[1:5, 1], )
3333

3434
X = BandedMatrix(SymTridiagonal(T[2n-1 for n in 1:n+b], T[-n for n in 1:n+b-1])) # Laguerre X, tests nonzero diagonal
3535
W = I+X^2+X^4

0 commit comments

Comments
 (0)