Skip to content

Commit

Permalink
Update cudf.Buffer pointer access method (#1094)
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 .

Authors:
  - Peter Andreas Entschev (https://github.com/pentschev)

Approvers:
  - GALI PREM SAGAR (https://github.com/galipremsagar)
  - Lawrence Mitchell (https://github.com/wence-)

URL: #1094
  • Loading branch information
pentschev authored Jan 27, 2023
1 parent 9ff3996 commit 4f0922c
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 4f0922c

Please sign in to comment.