cudf ptr

luweizheng · Jul 6, 2024 · 275fa7d · 275fa7d
1 parent f06f408
commit 275fa7d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/xorbits/_mars/storage/cuda.py b/python/xorbits/_mars/storage/cuda.py
@@ -275,7 +275,7 @@ async def get(self, object_id: str, **kwargs) -> object:
  if isinstance(buf, cupy.ndarray):
  new_buffers.append(DeviceBuffer(ptr=buf.data.ptr, size=buf.size))
  elif isinstance(buf, CPBuffer):
- new_buffers.append(DeviceBuffer(ptr=buf.ptr, size=buf.size))
+ new_buffers.append(DeviceBuffer(ptr=buf.owner._ptr, size=buf.size))
  else:
  new_buffers.append(buf)
  return deserialize(headers, new_buffers)