Link mtblas library

Author: Critsium-xy

CMakeLists.txt

@@ -39,6 +39,7 @@ option(ENABLE_RAPIDJSON "Enable rapid-json usage." OFF)
 option(ENABLE_CNPY "Enable cnpy usage." OFF)
 option(ENABLE_PEXSI "Enable support for PEXSI." OFF)
 option(ENABLE_CUSOLVERMP "Enable cusolvermp." OFF)
+option(ENABLE_DSP "Enable DSP usage." OFF)
 
 # enable json support
 if(ENABLE_RAPIDJSON)
@@ -119,6 +120,12 @@ elseif(ENABLE_LCAO AND NOT ENABLE_MPI)
   set(ABACUS_BIN_NAME abacus_serial)
 endif()
 
+if (USE_DSP)
+  set(USE_ELPA OFF)
+  set(ENABLE_LCAO OFF)
+  set(ABACUS_BIN_NAME abacus_dsp)
+endif()
+
 list(APPEND CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/cmake)
 
 if(ENABLE_COVERAGE)
@@ -240,6 +247,10 @@ if(ENABLE_MPI)
   list(APPEND math_libs MPI::MPI_CXX)
 endif()
 
+if (USE_DSP)
+  target_link_libraries(${ABACUS_BIN_NAME} DIR_MTBLAS_LIBRARY)
+endif()
+
 find_package(Threads REQUIRED)
 target_link_libraries(${ABACUS_BIN_NAME} Threads::Threads)
 

source/module_base/kernels/dsp/dsp_connector.h

@@ -0,0 +1,63 @@
+#ifdef __DSP
+
+// Base dsp functions
+void createMtblasHandle(int id);
+void destroyMtblasHandle();
+void *malloc_ht(size_t bytes);
+void free_ht(void* ptr);
+
+
+// mtblas functions
+
+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 const float *beta,
+	const float *c, const int *ldc);
+
+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,
+	const double *c, const int *ldc);
+
+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);
+
+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);
+
+
+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 const float *beta,
+	const float *c, const int *ldc);
+
+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,
+	const double *c, const int *ldc);
+
+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);
+
+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);
+
+//#define zgemm_ zgemm_mt
+
+#endif