Changes to MallocV to take into consideration the memory space of a v…

…ariable

source/adios2/toolkit/format/buffer/malloc/MallocV.cpp

@@ -7,6 +7,7 @@
  */
 
 #include "MallocV.h"
+#include "adios2/helper/adiosFunctions.h"
 #include "adios2/toolkit/format/buffer/BufferV.h"
 
 #include <algorithm>
@@ -74,7 +75,12 @@ size_t MallocV::AddToVec(const size_t size, const void *buf, size_t align, bool
             m_InternalBlock = (char *)realloc(m_InternalBlock, NewSize);
             m_AllocatedSize = NewSize;
         }
-        memcpy(m_InternalBlock + m_internalPos, buf, size);
+#ifdef ADIOS2_HAVE_GPU_SUPPORT
+        if (MemSpace == MemorySpace::GPU)
+            helper::CopyFromGPUToBuffer(m_InternalBlock, m_internalPos, buf, MemSpace, size);
+#endif
+        if (MemSpace == MemorySpace::Host)
+            memcpy(m_InternalBlock + m_internalPos, buf, size);
 
         if (DataV.size() && !DataV.back().External &&
             (m_internalPos == (DataV.back().Offset + DataV.back().Size)))