Fix bugs #1

Author: Critsium-xy

source/module_base/kernels/dsp/dsp_connector.h

@@ -1,3 +1,5 @@
+#ifndef DSP_CONNECTOR_H
+#define DSP_CONNECTOR_H
 #ifdef __DSP
 
 // Base dsp functions
@@ -12,7 +14,7 @@ void free_ht(void* ptr);
 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 *b, const int *ldb, const float *beta,
 	const float *c, const int *ldc);
 
 void dgemm_mt_(const char *transa, const char *transb,
@@ -37,7 +39,7 @@ void cgemm_mt_(const char *transa, const char *transb,
 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 *b, const int *ldb, const float *beta,
 	const float *c, const int *ldc);
 
 void dgemm_mth_(const char *transa, const char *transb,
@@ -60,4 +62,5 @@ void cgemm_mth_(const char *transa, const char *transb,
 
 //#define zgemm_ zgemm_mt
 
+#endif
 #endif

source/module_base/module_device/memory_op.cpp

@@ -3,7 +3,7 @@
 #include "module_base/memory.h"
 #include "module_base/tool_threading.h"
 #ifdef __DSP
-#include "module_base/kernels/dsp_connector.h"
+#include "module_base/kernels/dsp/dsp_connector.h"
 #endif
 
 #include <complex>

source/module_esolver/esolver_ks_pw.cpp

@@ -49,6 +49,10 @@
 #include <ATen/kernels/blas.h>
 #include <ATen/kernels/lapack.h>
 
+#ifdef __DSP
+#include "module_base/kernels/dsp/dsp_connector.h"
+#endif
+
 namespace ModuleESolver
 {