add FlushToDevice()

source/tensor/XTensor.cpp

@@ -1985,6 +1985,19 @@ void XTensor::FlushToMem(XMem* targetMem)
     }
 }
 
+/* 
+flush the data to the target device (with id) 
+>> myDevID - id of the target device
+*/
+void XTensor::FlushToDevice(int myDevID)
+{
+    if (myDevID == devID)
+        return;
+
+    XMem * myMem = GMems.GetMem(myDevID);
+    FlushToMem(myMem);
+}
+
 /*
 allocate the memory space of the tensor (in the global memory) 
 >> tensor - the tensor we intend to process

source/tensor/XTensor.h

@@ -457,6 +457,9 @@ public:
     /* flush the data to the target device */
     void FlushToMem(XMem * targetMem);
 
+    /* flush the data to the target device (with id) */
+    void FlushToDevice(int myDevID);
+
     /* allocate the memory space of the tensor (in the global memory) */
     static
     void AllocateData(XTensor * tensor, XMem * myMem = NULL, bool useBuf = false);