Skip to content

PyTorch compatibility layer #14

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
Merged
merged 95 commits into from
Feb 25, 2023
Merged

PyTorch compatibility layer #14

merged 95 commits into from
Feb 25, 2023

Conversation

asmeurer
Copy link
Member

@asmeurer asmeurer commented Jan 9, 2023

No description provided.

@honno honno mentioned this pull request Jan 10, 2023
Merged
@asmeurer asmeurer mentioned this pull request Jan 10, 2023
Closed
asmeurer added 20 commits January 10, 2023 13:32
@asmeurer
Return a copy from max and min with axis=()
dba7fa7
@asmeurer
Add a size() helper function
44d91e1
@asmeurer
Add any and all torch wrappers and fix some issues with prod
1faea7b
@asmeurer
Add astype torch wrapper
ad4484d
@asmeurer
Cast the input to prod/all/any to tensor
db3241d
@asmeurer
More logical order for some functions
3f0d913
@asmeurer
Add wrappers for two-argument elementwise functions
2d25dd2 
PyTorch doesn't do correct type promotion when one of the arguments is a
scalar. We unfortunately can't do anything about this for the actual
operators.
@asmeurer
Add bitwise_invert to torch
c4c0cfa
@asmeurer
Add torch wrappers for broadcast_to and can_cast
a1917f8
@asmeurer
Add torch arange wrapper
ae28ce0
@asmeurer
Add a wrapper for torch.eye
c3d9334
@asmeurer
Add pytorch linspace wrapper
f4d6df1
@asmeurer
Add torch squeeze wrapper
14b6519
@asmeurer
Add torch flip and roll wrappers
db3f579
@asmeurer
Add a torch wrapper for nonzero
27b7e8c
@asmeurer
Fix pyflakes warning
e886644
@asmeurer
Add torch wrapper for where
fd5d179
@asmeurer
Add sort wrapper to torch
a5f3253
@asmeurer
Pass kwargs through some torch wrappers
4a71e63
@asmeurer
Add torch mean(), std(), and var() wrappers
157fc1e
@asmeurer asmeurer mentioned this pull request Feb 3, 2023
Closed
@asmeurer asmeurer mentioned this pull request Feb 23, 2023
Merged
@asmeurer
Add a script to manually run the cupy tests
c8a5a70
@asmeurer
Add a torch skip
7261dae
@asmeurer
Merge branch 'pytorch' of github.com:asmeurer/array-api-compat into p…
795dbea 
…ytorch
@asmeurer
Use cupy specific skips and xfails
0368c8f
@asmeurer
Allow passing pytest args through in test_cupy.sh
2b456c2
@asmeurer
Add a shebang to test_cupy.sh
046ffd0
@asmeurer
Make the hypothesis examples database persistent in test_cupy.sh
c3eb0d5
@asmeurer
Fix sort() and argsort() with cupy
111a122
@asmeurer
Add comments for the rest of the cupy xfails
09b5a6f
@asmeurer
Merge branch 'pytorch' of github.com:asmeurer/array-api-compat into p…
b780b9a 
…ytorch
@asmeurer
Fix argument quoting in test_cupy.sh
1908a00
@asmeurer
Update cupy skips and xfails
5cbd1c0
@asmeurer
Update cupy xfails
6a63d5c
@asmeurer
Update test_cupy.sh to run the vendoring tests
175f195
@asmeurer
Add a minor CHANGELOG entry
d1c7999
@asmeurer
Merge branch 'pytorch' of github.com:asmeurer/array-api-compat into p…
0e923b6 
…ytorch
@asmeurer
Bump the version to 1.1
2fb0a0a
@asmeurer
Add a missing torch skip
3470b36
@asmeurer asmeurer enabled auto-merge February 25, 2023 01:29
@asmeurer asmeurer merged commit b143d02 into data-apis:main Feb 25, 2023
@lezcano
Copy link

lezcano commented Feb 26, 2023

how come this one was merged before the final review?

@asmeurer
Copy link
Member Author

Because I am on PTO this week and I wanted to get this merged and released before then. If you have any further comments feel free to comment here or open a new issue.

@rgommers rgommers changed the title WIP: PyTorch compatibility layer PyTorch compatibility layer Feb 27, 2023
albanD
albanD reviewed Mar 6, 2023
View reviewed changes
@rgommers rgommers mentioned this pull request Jul 4, 2023
Closed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@lezcano lezcano lezcano left review comments

+1 more reviewer

@albanD albanD albanD left review comments

Reviewers whose approvals may not affect merge requirements
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@asmeurer @lezcano @albanD