fix wrong dimension of gemm_op in rotate_wf

deepmodeling · Jan 6, 2025 · fe19e07 · fe19e07
1 parent 14c3693
commit fe19e07
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/source/module_hsolver/diago_bpcg.cpp b/source/module_hsolver/diago_bpcg.cpp
@@ -224,7 +224,7 @@ void DiagoBPCG<T, Device>::rotate_wf(
     gemm_op()(this->ctx,
                 'N',
                 'N',
-                this->n_dim,        //m
+                this->n_basis,        //m
                 this->n_band,       //n
                 this->n_band,       //k
                 this->one,          //1.0