anjakefala commented on code in PR #43590:
URL: https://github.com/apache/arrow/pull/43590#discussion_r1767459756
##########
python/pyarrow/tests/test_cuda.py:
##########
@@ -119,6 +119,24 @@ def make_random_buffer(size, target='host'):
raise ValueError('invalid target value')
+def test_copy_from_buffer():
+ arr, buf = make_random_buffer(128)
+ cudabuf = global_context.buffer_from_data(buf)
+
+ mm2 = global_context1.memory_manager
+ for dest in [mm2, mm2.device]:
+ buf2 = cudabuf.copy(dest)
+ assert buf2.device_type == pa.DeviceAllocationType.CUDA
Review Comment:
Ah, so the `pa.default_memory_manager()` is the memory manager for CPU.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
