Skip to content

Add "multi device" support #59

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.

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 26 commits into from
Oct 16, 2024
Merged
Show file tree
Hide file tree
Changes from 22 commits
Commits
Show all changes
26 commits
Select commit Hold shift + click to select a range
600df5e
Add "multi device" support
betatim Sep 4, 2024
325b9d0
Loop device through elementwise functions
betatim Sep 27, 2024
6cc7bac
Define __hash__
betatim Sep 27, 2024
bca670a
More device pass through
betatim Sep 27, 2024
426609f
Fix meshgrid
betatim Sep 27, 2024
727072f
Add testing and small typo fixes
betatim Oct 2, 2024
23f390e
Add a comment about atanh special casing
betatim Oct 3, 2024
405b7e7
Add conversion to NumPy test
betatim Oct 3, 2024
03e1ae7
Add multi-device support to sorting functions
betatim Oct 3, 2024
3bc8199
More multi-device support
betatim Oct 3, 2024
032f3bb
Formatting
betatim Oct 3, 2024
724e071
Add multi-device test for take
betatim Oct 3, 2024
e0b2a64
Multi-device support in linear algebra functions
betatim Oct 3, 2024
9323324
Multi-device support for array manipulation
betatim Oct 3, 2024
ff37de7
Add multi-device support for searching
betatim Oct 3, 2024
bae7482
Add multi-device support to stats and sets
betatim Oct 3, 2024
cca1785
Add multi-device support for utils
betatim Oct 3, 2024
a96c497
More FFT multi-device
betatim Oct 7, 2024
58334e5
Fix weird ruff error
betatim Oct 7, 2024
1c77ba0
Merge branch 'main' into multiple-devices
betatim Oct 7, 2024
9c5436c
New default version
betatim Oct 7, 2024
0dbabcc
Fix result device
betatim Oct 7, 2024
8e6365b
Make device= a required argument to create an Array
betatim Oct 16, 2024
635e14d
Merge branch 'main' into betatim-multiple-devices
asmeurer Oct 16, 2024
78def19
Add device check to repeat()
asmeurer Oct 16, 2024
33450f3
Use ValueError for different device errors
asmeurer Oct 16, 2024
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
3 changes: 3 additions & 0 deletions array_api_strict/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -309,6 +309,9 @@

__all__ += ["all", "any"]

from ._array_object import Device
__all__ += ["Device"]
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I'd rather not add this to the __init__.py since it isn't part of the array API. If it is necessary to have some public APIs to create device objects we should make APIs that are more obviously array-api-strict specific (similar to the flags APIs).


# Helper functions that are not part of the standard

from ._flags import (
Expand Down
Loading
Loading