Fix parallel function

Author: Critsium-xy

source/module_base/blas_connector.cpp

@@ -2,6 +2,7 @@
 
 #ifdef __DSP
 #include "module_base/kernels/dsp/dsp_connector.h"
+#include "module_base/global_variable.h"
 #endif
 
 void BlasConnector::axpy( const int n, const float alpha, const float *X, const int incX, float *Y, const int incY, base_device::AbacusDevice_t device_type)
@@ -94,7 +95,7 @@ void BlasConnector::gemm(const char transa, const char transb, const int m, cons
 	else if (device_type == base_device::AbacusDevice_t::DspDevice){
 		sgemm_mt_(&transb, &transa, &n, &m, &k,
 		&alpha, b, &ldb, a, &lda,
-		&beta, c, &ldc);
+		&beta, c, &ldc, GlobalV::MY_RANK);
 	}
 	#endif
 }
@@ -112,7 +113,7 @@ void BlasConnector::gemm(const char transa, const char transb, const int m, cons
 	else if (device_type == base_device::AbacusDevice_t::DspDevice){
 		dgemm_mt_(&transb, &transa, &n, &m, &k,
 		&alpha, b, &ldb, a, &lda,
-		&beta, c, &ldc);
+		&beta, c, &ldc, GlobalV::MY_RANK);
 	}
 	#endif
 }
@@ -130,7 +131,7 @@ void BlasConnector::gemm(const char transa, const char transb, const int m, cons
 	else if (device_type == base_device::AbacusDevice_t::DspDevice) {
     	cgemm_mt_(&transb, &transa, &n, &m, &k,
         &alpha, b, &ldb, a, &lda,
-        &beta, c, &ldc);
+        &beta, c, &ldc, GlobalV::MY_RANK);
 	}
 	#endif
 }
@@ -148,7 +149,7 @@ void BlasConnector::gemm(const char transa, const char transb, const int m, cons
 	else if (device_type == base_device::AbacusDevice_t::DspDevice) {
     	zgemm_mt_(&transb, &transa, &n, &m, &k,
         &alpha, b, &ldb, a, &lda,
-        &beta, c, &ldc);
+        &beta, c, &ldc, GlobalV::MY_RANK);
 	}
 	#endif
 }

source/module_base/kernels/dsp/dsp_connector.h

@@ -15,50 +15,50 @@ void sgemm_mt_(const char *transa, const char *transb,
 	const int *m, const int *n, const int *k,
 	const float *alpha, const float *a, const int *lda,
 	const float *b, const int *ldb, const float *beta,
-	float *c, const int *ldc);
+	float *c, const int *ldc, int cluster_id);
 
 void dgemm_mt_(const char *transa, const char *transb,
 	const int *m, const int *n, const int *k,
 	const double *alpha,const double *a, const int *lda,
 	const double *b, const int *ldb, const double *beta,
-	double *c, const int *ldc);
+	double *c, const int *ldc, int cluster_id);
 
 void zgemm_mt_(const char *transa, const char *transb,
 	const int *m, const int *n, const int *k,
 	const std::complex<double> *alpha, const std::complex<double> *a, const int *lda,
 	const std::complex<double> *b, const int *ldb, const std::complex<double> *beta,
-	std::complex<double> *c, const int *ldc);
+	std::complex<double> *c, const int *ldc, int cluster_id);
 
 void cgemm_mt_(const char *transa, const char *transb,
 	const int *m, const int *n, const int *k,
 	const std::complex<float> *alpha, const std::complex<float> *a, const int *lda,
 	const std::complex<float> *b, const int *ldb, const std::complex<float> *beta,
-	std::complex<float> *c, const int *ldc);
+	std::complex<float> *c, const int *ldc, int cluster_id);
 
 
 void sgemm_mth_(const char *transa, const char *transb,
 	const int *m, const int *n, const int *k,
 	const float *alpha, const float *a, const int *lda,
 	const float *b, const int *ldb, const float *beta,
-	float *c, const int *ldc);
+	float *c, const int *ldc, int cluster_id);
 
 void dgemm_mth_(const char *transa, const char *transb,
 	const int *m, const int *n, const int *k,
 	const double *alpha,const double *a, const int *lda,
 	const double *b, const int *ldb, const double *beta,
-	double *c, const int *ldc);
+	double *c, const int *ldc, int cluster_id);
 
 void zgemm_mth_(const char *transa, const char *transb,
 	const int *m, const int *n, const int *k,
 	const std::complex<double> *alpha, const std::complex<double> *a, const int *lda,
 	const std::complex<double> *b, const int *ldb, const std::complex<double> *beta,
-	std::complex<double> *c, const int *ldc);
+	std::complex<double> *c, const int *ldc, int cluster_id);
 
 void cgemm_mth_(const char *transa, const char *transb,
 	const int *m, const int *n, const int *k,
 	const std::complex<float> *alpha, const std::complex<float> *a, const int *lda,
 	const std::complex<float> *b, const int *ldb, const std::complex<float> *beta,
-	std::complex<float> *c, const int *ldc);
+	std::complex<float> *c, const int *ldc, int cluster_id);
 
 //#define zgemm_ zgemm_mt