8000 Add Python API, semantics and implementation details for DLPack by rgommers · Pull Request #106 · data-apis/array-api · GitHub
[go: up one dir, main page]
Skip to content

Add Python API, semantics and implementation details for DLPack #106

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 12 commits into from
Feb 21, 2021
Merged

Add Python API, semantics and implementation details for DLPack #106

Changes from 1 commit
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
30276e6
Update data interchange section with Python API and semantics
rgommers Dec 30, 2020
79bb689
temporary commit to enable MyST feature
rgommers Dec 30, 2020
4e2c49d
Add DLPack synchronization semantics; add from_dlpack/__dlpack__ to API
rgommers Jan 2, 2021
0b701c4
Update stream numbering for `__dlpack__`
rgommers Jan 12, 2021
a4549af
Add __dlpack__ device and update description of stream=None
rgommers Feb 16, 2021
a719b18
Add more device-specific notes for CUDA/ROCm stream handling
rgommers Feb 17, 2021
693b15a
Fix issue where producer/consumer were reversed
rgommers Feb 17, 2021
897ca2e
Improve the description of the stream keyword for `__dlpack__`
rgommers Feb 17, 2021
d3b9a79
Update __dlpack_device__ to use IntEnum for device type
rgommers Feb 20, 2021
75261cc
Add -1 as a sentinel value for DLPack stream handling
rgommers Feb 20, 2021
5cde9aa
Add supported DLPack version range.
rgommers Feb 20, 2021
603ad2e
Add details on strides null and size 0 arrays.
rgommers Feb 21, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Improve the description of the stream keyword for __dlpack__
  • Loading branch information
@rgommers
rgommers committed Feb 17, 2021
commit 897ca2eae442e969a7910ee4dd852f1de1fc9c7f
2 changes: 1 addition & 1 deletion spec/API_specification/array_object.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -375,7 +375,7 @@ Exports the array as a DLPack capsule, for consumption by {ref}`function-from_dl

- **stream**: _Optional\[int\]_

- If given, the stream number the consumer will use on the device the array is present on, as a Python integer. Device-specific notes:
- An optional pointer to a stream, as a Python integer, provided by the consumer that the producer will use to make the array safe to operate on. Device-specific notes:

:::{admonition} CUDA
- `None`: producer must assume the legacy default stream (default),
Expand Down
0