1. add all Label::ab_ab for LRI::cal_loop3()

Author: PeizeLin

include/RI/global/Tensor_Multiply-23.hpp

@@ -29,6 +29,23 @@ namespace Tensor_Multiply
 		return Txy;
 	}
 
+	// Txy(x0,y0,y1) = Tx(x0,a) * Ty(y0,y1,a)
+	template<typename Tdata>
+	Tensor<Tdata> x0y0y1_x0a_y0y1a(const Tensor<Tdata> &Tx, const Tensor<Tdata> &Ty)
+	{
+		assert(Tx.shape.size()==2);
+		assert(Ty.shape.size()==3);
+		Tensor<Tdata> Txy({Tx.shape[0], Ty.shape[0], Ty.shape[1]});
+		Blas_Interface::gemm(
+			'N', 'T',
+			Tx.shape[0],
+			Ty.shape[0] * Ty.shape[1],
+			Tx.shape[1],
+			Tdata(1.0), Tx.ptr(), Ty.ptr(),
+			Tdata(0.0), Txy.ptr());
+		return Txy;
+	}
+
 }
 
 }

include/RI/global/Tensor_Multiply-33.hpp

@@ -46,6 +46,23 @@ namespace Tensor_Multiply
 		return Txy;
 	}
 
+	// Txy(x2,y2) = Tx(a,b,x2) * Ty(a,b,y2)
+	template<typename Tdata>
+	Tensor<Tdata> x2y2_abx2_aby2(const Tensor<Tdata> &Tx, const Tensor<Tdata> &Ty)
+	{
+		assert(Tx.shape.size()==3);
+		assert(Ty.shape.size()==3);
+		Tensor<Tdata> Txy({Tx.shape[2], Ty.shape[2]});
+		Blas_Interface::gemm(
+			'T', 'N',
+			Tx.shape[2],
+			Ty.shape[2],
+			Tx.shape[0] * Tx.shape[1],
+			Tdata(1.0), Tx.ptr(), Ty.ptr(),
+			Tdata(0.0), Txy.ptr());
+		return Txy;
+	}
+
 }
 
 }