Skip to content

Commit

Permalink
Update cudf.Buffer pointer access method
Browse files Browse the repository at this point in the history
Fix test that reads directly from `cudf.Buffer` pointer to new
`get_ptr(mode="read")`, in accordance with changes from
rapidsai/cudf#12587 .
  • Loading branch information
pentschev committed Jan 27, 2023
1 parent 9ff3996 commit 07bab4e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion dask_cuda/tests/test_proxify_host_file.py
Original file line number Diff line number Diff line change
Expand Up @@ -281,7 +281,7 @@ def test_dataframes_share_dev_mem(root_dir):
# Even though the two dataframe doesn't point to the same cudf.Buffer object
assert view1["a"].data is not view2["a"].data
# They still share the same underlying device memory
view1["a"].data.ptr == view2["a"].data.ptr
view1["a"].data.get_ptr(mode="read") == view2["a"].data.get_ptr(mode="read")

dhf = ProxifyHostFile(
worker_local_directory=root_dir, device_memory_limit=160, memory_limit=1000
Expand Down

0 comments on commit 07bab4e

Please sign in to comment.