INITIAL COMMIT

Author: Critsium-xy

source/module_base/CMakeLists.txt

@@ -10,7 +10,10 @@ add_library(
     base
     OBJECT
     assoc_laguerre.cpp
-    blas_connector.cpp
+    #blas_connector.cpp
+    blas_connector_base.cpp
+    blas_connector_vector.cpp
+    blas_connector_matrix.cpp
     clebsch_gordan_coeff.cpp
     complexarray.cpp
     complexmatrix.cpp

source/module_base/blas_connector.h

@@ -368,9 +368,26 @@ class BlasConnector
 
 #ifdef __CUDA
 
+#include <cuda_runtime.h>
+#include "cublas_v2.h"
+
+// If you want to use cublas, you need these functions to create and destroy the cublas/hipblas handle.
+// You also need to use these functions to translate the transpose parameter into cublas/hipblas datatype.
+
 namespace BlasUtils{
-	void createGpuBlasHandle();
-	void destoryBLAShandle();
+
+	static cublasHandle_t cublas_handle = nullptr;
+
+	void createGpuBlasHandle(); // Create a cublas/hipblas handle.
+
+	void destoryBLAShandle(); // Destroy the cublas/hipblas handle. Do this when the software is about to end.
+
+	cublasOperation_t judge_trans(bool is_complex, const char& trans, const char* name); // Translate a normal transpose parameter to a cublas/hipblas type.
+
+	cublasSideMode_t judge_side(const char& trans); // Translate a normal side parameter to a cublas/hipblas type.
+
+	cublasFillMode_t judge_fill(const char& trans); // Translate a normal fill parameter to a cublas/hipblas type.
+
 }
 
 #endif

source/module_base/blas_connector_base.cpp

@@ -0,0 +1,79 @@
+#include "blas_connector.h"
+#include "macros.h"
+
+#ifdef __CUDA
+#include <base/macros/macros.h>
+#include <cuda_runtime.h>
+#include "cublas_v2.h"
+#include "module_base/kernels/math_kernel_op.h"
+#include "module_base/module_device/memory_op.h"
+
+
+namespace BlasUtils{
+
+	static cublasHandle_t cublas_handle = nullptr;
+
+	void createGpuBlasHandle(){
+		if (cublas_handle == nullptr) {
+			cublasErrcheck(cublasCreate(&cublas_handle));
+		}
+	}
+
+	void destoryBLAShandle(){
+		if (cublas_handle != nullptr) {
+			cublasErrcheck(cublasDestroy(cublas_handle));
+			cublas_handle = nullptr;
+		}
+	}
+
+
+	cublasOperation_t judge_trans(bool is_complex, const char& trans, const char* name)
+	{
+		if (trans == 'N')
+		{
+			return CUBLAS_OP_N;
+		}
+		else if(trans == 'T')
+		{
+			return CUBLAS_OP_T;
+		}
+		else if(is_complex && trans == 'C')
+		{
+			return CUBLAS_OP_C;
+		}
+		return CUBLAS_OP_N;
+	}
+
+	cublasSideMode_t judge_side(const char& trans)
+	{
+		if (trans == 'L')
+		{
+			return CUBLAS_SIDE_LEFT;
+		}
+		else if (trans == 'R')
+		{
+			return CUBLAS_SIDE_RIGHT;
+		}
+		return CUBLAS_SIDE_LEFT;
+	}
+
+	cublasFillMode_t judge_fill(const char& trans)
+	{
+		if (trans == 'F')
+		{
+			return CUBLAS_FILL_MODE_FULL;
+		}
+		else if (trans == 'U')
+		{
+			return CUBLAS_FILL_MODE_UPPER;
+		}
+		else if (trans == 'D')
+		{
+			return CUBLAS_FILL_MODE_LOWER;
+		}
+		return CUBLAS_FILL_MODE_FULL;
+	}
+
+} // namespace BlasUtils
+
+#endif