Add one and zero object for gemm

Author: Cstandardlib

source/module_hsolver/diago_bpcg.cpp

@@ -94,24 +94,24 @@ void DiagoBPCG<T, Device>::orth_cholesky(
 		ct::Tensor& hsub_out)
 {
     // hsub_out = psi_out * transc(psi_out)
-    // ct::EinsumOption option(
-    //     /*conj_x=*/false, /*conj_y=*/true, /*alpha=*/1.0, /*beta=*/0.0, /*Tensor out=*/&hsub_out);
-    // hsub_out = ct::op::einsum("ij,kj->ik", psi_out, psi_out, option);
+    ct::EinsumOption option(
+        /*conj_x=*/false, /*conj_y=*/true, /*alpha=*/1.0, /*beta=*/0.0, /*Tensor out=*/&hsub_out);
+    hsub_out = ct::op::einsum("ij,kj->ik", psi_out, psi_out, option);
     // using gemm instead einsum for different leading dimension and nbasis
-    gemm_op<T, Device>()(this->ctx,
-                         'N',
-                         'C',
-                         this->n_band,
-                         this->n_band,
-                         this->n_dim,
-                         this->one,
-                         psi_out.data<T>(),
-                         this->n_basis,
-                         psi_out.data<T>(),
-                         this->n_basis,
-                         this->zero,
-                         hsub_out.data<T>(),
-                         this->n_band);
+    // gemm_op()(this->ctx,
+    //                      'N',
+    //                      'C',
+    //                      this->n_band,
+    //                      this->n_band,
+    //                      this->n_dim,
+    //                      this->one,
+    //                      psi_out.data<T>(),
+    //                      this->n_basis,
+    //                      psi_out.data<T>(),
+    //                      this->n_basis,
+    //                      this->zero,
+    //                      hsub_out.data<T>(),
+    //                      this->n_band);
 
     // set hsub matrix to lower format;
     ct::kernels::set_matrix<T, ct_Device>()(

source/module_hsolver/diago_bpcg.h

@@ -80,6 +80,10 @@ class DiagoBPCG
     /// cg convergence thr
     Real all_band_cg_thr = 1E-5;
 
+    // Pointer to objects of 1 and 0 for gemm 
+    const T *one = nullptr, *zero = nullptr, *neg_one = nullptr;
+    const T one_ = static_cast<T>(1.0), zero_ = static_cast<T>(0.0), neg_one_ = static_cast<T>(-1.0);
+
     ct::DataType r_type  = ct::DataType::DT_INVALID;
     ct::DataType t_type  = ct::DataType::DT_INVALID;
     ct::DeviceType device_type = ct::DeviceType::UnKnown;
@@ -334,6 +338,7 @@ class DiagoBPCG
 
     using calc_grad_with_block_op = hsolver::calc_grad_with_block_op<T, Device>;
     using line_minimize_with_block_op = hsolver::line_minimize_with_block_op<T, Device>;
+    using gemm_op = hsolver::gemm_op<T, Device>;
 
 };