From 7cc20a9c27b7d3b198c732a2d5532116ce8b64ea Mon Sep 17 00:00:00 2001 From: ysyapa Date: Sun, 27 Aug 2023 23:02:51 -0400 Subject: [PATCH] Fixed GPU related bug --- Hamiltonian.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Hamiltonian.jl b/Hamiltonian.jl index a206677..e09cafa 100644 --- a/Hamiltonian.jl +++ b/Hamiltonian.jl @@ -68,8 +68,8 @@ end "cuTENSOR contraction and accumulation (C = A * B + C)" function contract_accumulate!(C::CuTensor, A::CuTensor, B::CuTensor)::CuTensor - CUTENSOR.contraction!(one(eltype(C)), 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) + cuTENSOR.contraction!(one(eltype(C)), 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 end