-
Notifications
You must be signed in to change notification settings - Fork 10
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
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 325b9d0
Loop device through elementwise functions
betatim 6cc7bac
Define __hash__
betatim bca670a
More device pass through
betatim 426609f
Fix meshgrid
betatim 727072f
Add testing and small typo fixes
betatim 23f390e
Add a comment about atanh special casing
betatim 405b7e7
Add conversion to NumPy test
betatim 03e1ae7
Add multi-device support to sorting functions
betatim 3bc8199
More multi-device support
betatim 032f3bb
Formatting
betatim 724e071
Add multi-device test for take
betatim e0b2a64
Multi-device support in linear algebra functions
betatim 9323324
Multi-device support for array manipulation
betatim ff37de7
Add multi-device support for searching
betatim bae7482
Add multi-device support to stats and sets
betatim cca1785
Add multi-device support for utils
betatim a96c497
More FFT multi-device
betatim 58334e5
Fix weird ruff error
betatim 1c77ba0
Merge branch 'main' into multiple-devices
betatim 9c5436c
New default version
betatim 0dbabcc
Fix result device
betatim 8e6365b
Make device= a required argument to create an Array
betatim 635e14d
Merge branch 'main' into betatim-multiple-devices
asmeurer 78def19
Add device check to repeat()
asmeurer 33450f3
Use ValueError for different device errors
asmeurer File filter
Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
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).