Fix parallel usage

Author: Critsium-xy

source/module_base/kernels/dsp/dsp_connector.h

@@ -4,8 +4,8 @@
 
 // Base dsp functions
 void dspInitHandle(int id);
-void dspDestoryHandle();
-void *malloc_ht(size_t bytes);
+void dspDestoryHandle(int id);
+void *malloc_ht(size_t bytes, int cluster_id);
 void free_ht(void* ptr);
 
 

source/module_base/module_device/memory_op.cpp

@@ -4,6 +4,7 @@
 #include "module_base/tool_threading.h"
 #ifdef __DSP
 #include "module_base/kernels/dsp/dsp_connector.h"
+#include "module_base/global_variable.h"
 #endif
 
 #include <complex>
@@ -28,7 +29,7 @@ struct resize_memory_op<FPTYPE, base_device::DEVICE_CPU>
 #endif
         }
 #ifdef __DSP
-        arr = (FPTYPE*)malloc_ht(sizeof(FPTYPE) * size);
+        arr = (FPTYPE*)malloc_ht(sizeof(FPTYPE) * size, GlobalV::MY_RANK);
 #else
         arr = (FPTYPE*)malloc(sizeof(FPTYPE) * size);
 #endif

source/module_esolver/esolver_ks_pw.cpp

@@ -73,7 +73,7 @@ ESolver_KS_PW<T, Device>::ESolver_KS_PW()
 #endif
 #ifdef __DSP
     std::cout << " ** Initializing DSP Hardware..." << std::endl;
-    dspInitHandle(GlobalV::MY_RANK % 4);
+    dspInitHandle(GlobalV::MY_RANK);
 #endif
 }
 
@@ -102,7 +102,7 @@ ESolver_KS_PW<T, Device>::~ESolver_KS_PW()
     }
 #ifdef __DSP
     std::cout << " ** Closing DSP Hardware..." << std::endl;
-    dspDestoryHandle();
+    dspDestoryHandle(GlobalV::MY_RANK);
 #endif
     if (PARAM.inp.precision == "single")
     {