-
Notifications
You must be signed in to change notification settings - Fork 54
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
Array API #1022
base: main
Are you sure you want to change the base?
Conversation
This fixes a bug where negative numbers could be represented by uint8.
In cases involving large numbers where the `dtype` parameter was unchanged, precision loss occurred due to intermediate casting to float32 (in `__local_op`) before casting back to the original `dtype`. Fixed by not doing the intermediate casting.
Fixed bug in `ht.any` where `keepdim` argument was not working when `axis=None` (same fix as 136e1b7)
f0a169b introduced a change in ht.abs where the dtype is not promoted to float by default. This caused the tests to fail.
Thank you for the PR! |
All Linalg tests are skipped for now. |
heat/array_api/test/skips.txt
Outdated
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.
This file contains all function tests that sometimes fail and sometimes not. It skips them. Might be easier to fix later.
heat/array_api/test/xfails.txt
Outdated
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.
This file contains function tests that failed every single time. These are expected to fail again. Mostly not implemented or not compliant functions.
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.
Let's add @neosunhan 's name to the list of contributors . Otherwise this can be merged, great job @mtar and @neosunhan !
Thank you for the PR! |
Thank you for the PR! |
Description
Implement array API according to the Python array API standard (version 2021.12)
See #774 for more information.
Changes proposed:
EDIT (Claudia): This PR also introduces a workflow to test compliance whenever a PR is reviewed.
EDIT2: workflow triggered manually, API version 2023.12
Type of change
Due Diligence
Does this change modify the behaviour of other functions? If so, which?
no
skip ci