AlenkaF commented on code in PR #41889:
URL: https://github.com/apache/arrow/pull/41889#discussion_r1636423725
##########
python/pyarrow/tests/test_cuda.py:
##########
@@ -313,7 +313,8 @@ def test_CudaBuffer(size):
assert cbuf[s].to_pybytes() == arr[s].tobytes()
sbuf = cbuf.slice(size//4, size//2)
- assert sbuf.parent == cbuf
+ cbuf_parent = cuda.CudaBuffer.from_buffer(sbuf.parent)
+ assert cbuf_parent.to_pybytes() == cbuf.to_pybytes()
Review Comment:
`sbuf.parent` is pyarrow Buffer with data on GPU and so the `equals` method
can not work as we have added a check for non-cpu data.
Why it was working before I am not sure - it segfaulted when testing buffers
on GPU for tests in _test_io.py_ 🤷♀️
--
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]
