Bug fixed.

Fix the bugs in FlushToMem functions.

source/tensor/core/utilities/FlushToMem.cu

@@ -104,7 +104,7 @@ void CudaGPUToCPUFlush(XTensor * tensor, int devID, XMem * CPUMem)
 
     /* compute the requried memory size */
     int size = 0;
-    if (m->isSparse)
+    if (tensor->isSparse)
         size = sizeof(int) + (sizeof(int) + tensor->unitSize) * tensor->unitNumNonZero;
     else
         size = tensor->unitSize * tensor->unitNum;

source/tensor/core/utilities/FlushToMem.cuh

@@ -31,7 +31,10 @@ namespace nts { // namespace nts(NiuTrans.Tensor)
 /* flush a list of XTensor to GPU memory */
 void CudaCPUToGPUFlush(TensorList * mList, int devID, XMem * GPUMem);
 
-/* copy the data from GPU memory to CPU memory */
+/* copy the data from GPU memory to CPU memory (memory pool) */
+void CudaGPUToCPUFlush(XTensor * tensor, int devID, XMem * CPUMem);
+
+/* copy the data from GPU memory to CPU memory ((dataHost)) and do not delete the data */
 void CudaGPUToCPUFlush(XTensor * tensor);
 
 #endif // USE_CUDA