From 2263c26215145ce2cc963b286a12442fc4fb961a Mon Sep 17 00:00:00 2001
From: ysyapa <ysyapa@ncsu.edu>
Date: Sun, 27 Aug 2023 23:44:22 -0400
Subject: [PATCH] Jacobi runs on GPU

---
 Hamiltonian.jl | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/Hamiltonian.jl b/Hamiltonian.jl
index c23942b..d159680 100644
--- a/Hamiltonian.jl
+++ b/Hamiltonian.jl
@@ -19,7 +19,7 @@ struct Hamiltonian{T}
         ∂ = ∂_1DOF.(Ref(s), k, k')
         K = exp(-2im * convert(T, ϕ)) .* (∂ * ∂)    # TODO: Calculate K matrix elements directly
         if mode == gpu_cutensor
-            K = CuTensor(K, ['a', 'A'])
+            K = CuTensor(CuArray(K), ['a', 'A'])
             Vs = CuArray(Vs)
         end
         return new{T}(s, K, Vs, hermitian, mode)
@@ -55,7 +55,7 @@ function LinearAlgebra.mul!(out::Array{Complex{T}}, H::Hamiltonian{T}, v::Array{
 end
 
 "cuTENSOR contraction and accumulation (C = A * B + C)"
-function contract_accumulate!(alpha::Numer, C::CuTensor, A::CuTensor, B::CuTensor)::CuTensor
+function contract_accumulate!(alpha::Number, C::CuTensor, A::CuTensor, B::CuTensor)::CuTensor
     cuTENSOR.contraction!(alpha, A.data, A.inds, cuTENSOR.CUTENSOR_OP_IDENTITY, B.data, B.inds, cuTENSOR.CUTENSOR_OP_IDENTITY,
                           one(eltype(C)), C.data, C.inds, cuTENSOR.CUTENSOR_OP_IDENTITY, cuTENSOR.CUTENSOR_OP_IDENTITY)
     return C
@@ -77,7 +77,7 @@ function LinearAlgebra.mul!(out::CuArray{Complex{T}}, H::Hamiltonian{T}, v::CuAr
         for coord = 1:coords
             i = which_index(H.s, dim, coord)
             @assert v_t.inds == inds_template "v indices permuted"
-            @assert H.K_diag.inds[2] == 'A' "K_diag indices permuted"
+            @assert H.K.inds[2] == 'A' "K_diag indices permuted"
             H.K.inds[1] = 'a' - 1 + i
             v_t.inds[i] = 'A'
             #synchronize(ctx)