Advanced Computing Platform for Theoretical Physics

Commit f6f14409 authored by Shuang Liang's avatar Shuang Liang
Browse files

fix: eigensolver

parent d04917e7
......@@ -129,8 +129,10 @@ LanczosEigen(init_vector::AbstractVector{T},
function eigensolver(A::itFOLMResult{T,S,U,V}) where {T,S,U,V}
@unpack init_vector, symtridiagonal, krylov_basis = A
values, Q = eigen(symtridiagonal)
values = convert(S, values)
Q = convert(V, Q)
if S<:CuArray
values = convert(CuArray, values)
Q = convert(CuArray, Q)
end
vectors = krylov_basis * Q
weight = Q[1,:]
return LanczosEigen(init_vector, weight, values, vectors)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment