Skip to content

Commit

Permalink
Replace to_device() in docs with DeviceBuffer.to_device()
Browse files Browse the repository at this point in the history
  • Loading branch information
shwina authored Nov 1, 2021
1 parent c07f1c8 commit 6043b88
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python/docs/basics.md
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,7 @@ DeviceBuffers can also be created by copying data from host memory:
>>> import rmm
>>> import numpy as np
>>> a = np.array([1, 2, 3], dtype='float64')
>>> buf = rmm.to_device(a.tobytes())
>>> buf = rmm.DeviceBuffer.to_device(a.view("int8")) # to_device expects an 8-bit type or `bytes`
>>> buf.size
24
```
Expand Down

0 comments on commit 6043b88

Please sign in to comment.