Skip to content

Commit 87f0d36

Browse files
committed
Revert change of einsum in rotate_wf:
1 parent fe19e07 commit 87f0d36

File tree

1 file changed

+15
-15
lines changed

1 file changed

+15
-15
lines changed

source/module_hsolver/diago_bpcg.cpp

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -216,25 +216,25 @@ void DiagoBPCG<T, Device>::rotate_wf(
216216
{
217217
ct::EinsumOption option(
218218
/*conj_x=*/false, /*conj_y=*/false, /*alpha=*/1.0, /*beta=*/0.0, /*Tensor out=*/&workspace_in);
219-
// workspace_in = ct::op::einsum("ij,jk->ik", hsub_in, psi_out, option);
219+
workspace_in = ct::op::einsum("ij,jk->ik", hsub_in, psi_out, option);
220220

221221
// this->rotate_wf(hsub_out, psi_out, workspace_in);
222222
// this->orth_cholesky(this->work, this->psi, this->hpsi, this->hsub);
223223
// gemm: workspace_in(n_basis x n_band) = psi_out(n_basis x n_band) * hsub_in(n_band x n_band)
224-
gemm_op()(this->ctx,
225-
'N',
226-
'N',
227-
this->n_basis, //m
228-
this->n_band, //n
229-
this->n_band, //k
230-
this->one, //1.0
231-
psi_out.data<T>(),
232-
this->n_basis, //lda
233-
hsub_in.data<T>(),
234-
this->n_band, //ldb
235-
this->zero, //0.0
236-
workspace_in.data<T>(),
237-
this->n_basis); //ldc
224+
// gemm_op()(this->ctx,
225+
// 'N',
226+
// 'N',
227+
// this->n_basis, //m
228+
// this->n_band, //n
229+
// this->n_band, //k
230+
// this->one, //1.0
231+
// psi_out.data<T>(),
232+
// this->n_basis, //lda
233+
// hsub_in.data<T>(),
234+
// this->n_band, //ldb
235+
// this->zero, //0.0
236+
// workspace_in.data<T>(),
237+
// this->n_basis); //ldc
238238

239239
syncmem_complex_op()(psi_out.template data<T>(), workspace_in.template data<T>(), this->n_band * this->n_basis);
240240

0 commit comments

Comments
 (0)