Skip to content

Array API #1022

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

Jump to bottom
Open
wants to merge 179 commits into
base: main
Choose a base branch
from
Open

Array API #1022

wants to merge 179 commits into from

Conversation

@neosunhan
Copy link
Collaborator

@neosunhan neosunhan commented Sep 4, 2022 •
edited by mtar
Loading

Description

Implement array API according to the Python array API standard (version 2021.12)

See #774 for more information.

Changes proposed:

  • In conformance with the specification, the new standard-compliant API will be offered in a new namespace. This namespace can be accessed through a method on the array object.
xp = x.__array_namespace__(api_version='2020.10')

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

  • New feature (non-breaking change which adds functionality)
  • Documentation update

Due Diligence

  • All split configurations tested
  • Multiple dtypes tested in relevant functions
  • Documentation updated (if needed)
  • Updated changelog.md under the title "Pending Additions"

Does this change modify the behaviour of other functions? If so, which?

no

skip ci

neosunhan added 30 commits June 14, 2022 18:51
@neosunhan
Create array_api namespace
ed334f8
@neosunhan
Enable copy parameter for asarray
ec0ebdc
@neosunhan
Add iinfo and finfo
77813fc
@neosunhan
Add constants
38461c7
@neosunhan
Modify iinfo and finfo to read minimum value from torch object
ae1defb 
This fixes a bug where negative numbers could be represented by uint8.
@neosunhan
Add docstrings
313d22d
@neosunhan
Fix bug where keepdim was not working for all
136e1b7
@neosunhan
Implement all
9c6b344
@neosunhan
Implement Array.__eq__
ab30f56
@neosunhan
Implement astype
198a38e
@neosunhan
Implement Array object
2e8cfc3
@neosunhan
Implement add
fed2952
@neosunhan
Standardize import order
accbd15
@neosunhan
Document function parameters
de16eac
@neosunhan
Fix bug causing precision loss in abs
f0a169b 
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.
@neosunhan
Implement abs
1e0c29b
@neosunhan
Implement add
9bcbb01
@neosunhan
Implement bitwise_and
ff53a95
@neosunhan
Implement bitwise_left_shift, less, any
afe0b95 
Fixed bug in `ht.any` where `keepdim` argument was not working
when `axis=None` (same fix as 136e1b7)
@neosunhan
Implement bitwise_invert
2ef6e3a
@neosunhan
Implement bitwise_or
574f469
@neosunhan
Fix documentation for _normalize_two_args
b684c86
@neosunhan
Fix bug in linalg.matrix_norm
6c50d7b 
f0a169b introduced a change in ht.abs where the dtype is not
promoted to float by default. This caused the tests to fail.
@neosunhan
Implement bitwise_xor
9fb9bdd
@neosunhan
Implement equal
ad655ee
@neosunhan
Implement floor_divide
41143f7
@neosunhan
Implement greater_equal
aeb97e4
@neosunhan
Implement greater
58308db
@neosunhan
Implement less_equal
544ab0e
@neosunhan
Implement remainder
7ff5119
@ClaudiaComito
Copy link
Contributor

How are we going to keep track of progress here? i.e. what functions are missing from the coverage, what are being worked on, etc. @mtar @JuanPedroGHM

@mtar
Copy link
Collaborator

mtar commented Dec 10, 2024

The progress is tracked in the skips.txt and xfails.txt files themselves. At the moment

  • skips.txt contains tests that sometimes failed
  • xfail.txt comprises tests that always failed

when the test suite was running.
In the end they should only have tests that we want to skip or expect to fail.

@ClaudiaComito
Copy link
Contributor

The progress is tracked in the skips.txt and xfails.txt files themselves. At the moment

* `skips.txt` contains tests that sometimes failed

* `xfail.txt` comprises tests that always failed

when the test suite was running. In the end they should only have tests that we want to skip or expect to fail.

By tracking I mean something along these lines: pytorch/pytorch#58743

@github-actions
Copy link
Contributor

Thank you for the PR!

@mtar mtar mentioned this pull request Mar 10, 2025
Open
@JuanPedroGHM JuanPedroGHM modified the milestones: 1.6, 1.7.0 Mar 17, 2025
@JuanPedroGHM JuanPedroGHM linked an issue Apr 8, 2025 that may be closed by this pull request
Array API compliance #774
Open
@github-actions
Copy link
Contributor

Thank you for the PR!

@github-actions
Copy link
Contributor

Thank you for the PR!

@github-actions
Copy link
Contributor

github-actions bot commented Jun 4, 2025

Thank you for the PR!

@github-actions
Copy link
Contributor

github-actions bot commented Jul 7, 2025

Thank you for the PR!

@github-actions
Copy link
Contributor

Thank you for the PR!

@github-actions
Copy link
Contributor

Thank you for the PR!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@mtar mtar mtar left review comments

@Markus-Goetz Markus-Goetz Awaiting requested review from Markus-Goetz

@JuanPedroGHM JuanPedroGHM Awaiting requested review from JuanPedroGHM

@ClaudiaComito ClaudiaComito Awaiting requested review from ClaudiaComito

Requested changes must be addressed to merge this pull request.

Assignees

@mtar mtar

Labels

array API enhancement New feature or request GSoC interoperability workflows

Projects

Roadmap
Status: In Progress

Milestone

1.7.0

Development

Successfully merging this pull request may close these issues.

Array API compliance

5 participants

@neosunhan @mtar @ClaudiaComito @JuanPedroGHM