Restore DeviceMemory constructor

nlesc-recruit · Aug 14, 2024 · 6b13265 · 6b13265
1 parent dff5959
commit 6b13265
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/include/cudawrappers/cu.hpp b/include/cudawrappers/cu.hpp
@@ -576,6 +576,14 @@ class DeviceMemory : public Wrapper<CUdeviceptr> {
     checkCudaCall(cuMemHostGetDevicePointer(&_obj, hostMemory, 0));
   }
 
+  explicit DeviceMemory(const DeviceMemory &other, size_t offset, size_t size)
+      : _size(size) {
+    if (size + offset > other.size()) {
+      throw Error(CUDA_ERROR_INVALID_VALUE);
+    }
+    _obj = reinterpret_cast<CUdeviceptr>(reinterpret_cast<char *>(other._obj) +
+                                         offset);
+
   void memset(unsigned char value, size_t size) {
     checkCudaCall(cuMemsetD8(_obj, value, size));
   }