Repository navigation

PyTorch compatibility layer - #14

Merged

asmeurer merged 95 commits into

data-apis:mainfrom

asmeurer:pytorch

Feb 25, 2023

Merged

PyTorch compatibility layer#14
asmeurer merged 95 commits into
data-apis:mainfrom
asmeurer:pytorch

Commits on Jan 7, 2023
- Start pytorch compatibility layer
  asmeurer
  committed
  f49f42b
- Add vendor tests for torch
  asmeurer
  committed
  4b95748
Commits on Jan 9, 2023
- Replace torch expand_dims wrapper to a wrapper around unsqueeze
  asmeurer
  committed
  1ecb7ca
- Add torch support to the helper functions
  asmeurer
  committed
  52b6054
- Add max and min wrappers for torch
  asmeurer
  committed
  c484dbf
- Add a wrapper for torch.prod
  asmeurer
  committed
  3023a89
- Add the torch prod wrapper to __all__
  asmeurer
  committed
  a1bbd9b
Commits on Jan 10, 2023
- Return a copy from max and min with axis=()
  asmeurer
  committed
  dba7fa7
- Add a size() helper function
  asmeurer
  committed
  44d91e1
- Add any and all torch wrappers and fix some issues with prod
  asmeurer
  committed
  1faea7b
Commits on Jan 11, 2023
- Add astype torch wrapper
  asmeurer
  committed
  ad4484d
- Cast the input to prod/all/any to tensor
  asmeurer
  committed
  db3241d
- More logical order for some functions
  asmeurer
  committed
  3f0d913
Commits on Jan 24, 2023
- Add wrappers for two-argument elementwise functions
  asmeurer
  committed
  2d25dd2
- Add bitwise_invert to torch
  asmeurer
  committed
  c4c0cfa
- Add torch wrappers for broadcast_to and can_cast
  asmeurer
  committed
  a1917f8
- Add torch arange wrapper
  asmeurer
  committed
  ae28ce0
- Add a wrapper for torch.eye
  asmeurer
  committed
  c3d9334
- Add pytorch linspace wrapper
  asmeurer
  committed
  f4d6df1
- Add torch squeeze wrapper
  asmeurer
  committed
  14b6519
- Add torch flip and roll wrappers
  asmeurer
  committed
  db3f579
- Add a torch wrapper for nonzero
  asmeurer
  committed
  27b7e8c
- Fix pyflakes warning
  asmeurer
  committed
  e886644
- Add torch wrapper for where
  asmeurer
  committed
  fd5d179
Commits on Feb 2, 2023
- Add sort wrapper to torch
  asmeurer
  committed
  a5f3253
- Pass kwargs through some torch wrappers
  asmeurer
  committed
  4a71e63
- Add torch mean(), std(), and var() wrappers
  asmeurer
  committed
  157fc1e
Commits on Feb 3, 2023
- Add torch sum() and prod() wrappers
  asmeurer
  committed
  c3efe6a
Commits on Feb 4, 2023
- Add unique_* wrappers to torch
  asmeurer
  committed
  ed46247
- Just raise NotImplementedError in pytorch unique_all()
  asmeurer
  committed
  eaf5358
Commits on Feb 10, 2023
- Fix to_device for pytorch tensors
  asmeurer
  committed
  cd25f47
- Restrict the names imported from torch into the compat submodule
  asmeurer
  committed
  98ed0b2
- Allow torch sum and prod to upcast uint8 to int64
  asmeurer
  committed
  be4b534
- Don't unnecessarily flip the axes in flip()
  asmeurer
  committed
  ecda017
- Comment out dead code in the torch unique_all() wrapper
  asmeurer
  committed
  3ccee1b
- Use flatten instead of ravel
  asmeurer
  committed
  7699755
- Improve some error messages
  asmeurer
  committed
  d38bad5
Commits on Feb 13, 2023
- Use a better function name and use unsqueeze instead of None indexing
  asmeurer
  committed
  24c0ea3
Commits on Feb 18, 2023
- Add pytorch-xfails.txt (still need to validate)
  asmeurer
  committed
  48d1ae1
- Move main namespace linear algebra helpers to _aliases.py
  asmeurer
  committed
  866647d
- Merge branch 'master' into pytorch
  asmeurer
  committed
  85b71de
- Fix main namespace linalg functions in numpy and cupy
  asmeurer
  committed
  c441f33
- Add main namespace linalg functions to the torch wrapper
  asmeurer
  committed
  04eef18
- Add torch wrapper for matmul
  asmeurer
  committed
  453ecb8
Commits on Feb 20, 2023
- Finish torch wrappers for matmul, vecdot, and tensordot
  asmeurer
  committed
  5a3bbbe
Commits on Feb 21, 2023
- Clean up pytorch-xfails file
  asmeurer
  committed
  b8bbdc8
- Update pytorch-xfails.txt
  asmeurer
  committed
  7d176b9
- Merge branch 'main' into pytorch
  asmeurer
  committed
  17c0a91
- Install pytorch in on CI
  asmeurer
  committed
  1ffcb15
- Make the GitHub Actions workflow reusable so that we can test pytorch
  asmeurer
  committed
  0db034d
- Fix workflow path
  asmeurer
  committed
  39cbfd4
- Fix variable interpolation syntax
  asmeurer
  committed
  c3d0d8e
- Allow specifying extra pytest args in the test yamls
  asmeurer
  committed
  cf21cea
- Enable verbose output for the torch tests
  asmeurer
  committed
  a95eeb6
- Revert "Enable verbose output for the torch tests"
  asmeurer
  committed
  4737dc0
- Skip the torch test that crashes the CI
  asmeurer
  committed
  ea9c1e2
Commits on Feb 22, 2023
- Skip another test that crashes on CI
  asmeurer
  committed
  4904411
- Disable linalg in the torch CI tests
  asmeurer
  committed
  039af59
- Add missing torch xfails
  asmeurer
  committed
  367c4b6
- Do a verbose CI run for the pytorch array API tests
  asmeurer
  committed
  f3ee38c
- Revert "Do a verbose CI run for the pytorch array API tests"
  asmeurer
  committed
  c631fa3
- Add some missing torch xfails
  asmeurer
  committed
  33dacf9
- Do a verbose output run of the torch array API tests (with the correct flag this time)
  asmeurer
  committed
  82b8def
- Revert "Do a verbose output run of the torch array API tests (with the correct flag this time)"
  asmeurer
  committed
  b014c1b
- Add a missing torch xfail
  asmeurer
  committed
  124d6c3
- Add a missing torch xfail
  asmeurer
  committed
  847a9e5
Commits on Feb 23, 2023
- Skip test_floor_divide, which core dumps on CI
  asmeurer
  committed
  0857b86
- Add a missing torch xfail
  asmeurer
  committed
  6354cd9
- Update the README
  asmeurer
  committed
  0565dee
- Fix some formatting in the README
  asmeurer
  committed
  e9b447c
- Typo fix
  asmeurer
  committed
  bb4d3af
- Update torch reduction functions that don't support multiple axes
  asmeurer
  committed
  5545635
- Add a test skip that crashes on CI
  asmeurer
  committed
  a78f733
- Add a CHANGELOG for the upcoming 1.1 release
  asmeurer
  committed
  0cefa7b
- Add more skips for tests that crash on CI
  asmeurer
  committed
  1aceff5
- Skip a torch test that crashes CI
  asmeurer
  committed
  b016d4c
- Add a torch xfail
  asmeurer
  committed
  1cd43f2
- Add a script to manually run the cupy tests
  asmeurer
  committed
  c8a5a70
- Add a torch skip
  asmeurer
  committed
  7261dae
- Merge branch 'pytorch' of github.com:asmeurer/array-api-compat into pytorch
  asmeurer
  committed
  795dbea
Commits on Feb 24, 2023
- Use cupy specific skips and xfails
  asmeurer
  committed
  0368c8f
- Allow passing pytest args through in test_cupy.sh
  asmeurer
  committed
  2b456c2
- Add a shebang to test_cupy.sh
  asmeurer
  committed
  046ffd0
- Make the hypothesis examples database persistent in test_cupy.sh
  asmeurer
  committed
  c3eb0d5
- Fix sort() and argsort() with cupy
  asmeurer
  committed
  111a122
- Add comments for the rest of the cupy xfails
  asmeurer
  committed
  09b5a6f
- Merge branch 'pytorch' of github.com:asmeurer/array-api-compat into pytorch
  asmeurer
  committed
  b780b9a
Commits on Feb 25, 2023
- Fix argument quoting in test_cupy.sh
  asmeurer
  committed
  1908a00
- Update cupy skips and xfails
  asmeurer
  committed
  5cbd1c0
- Update cupy xfails
  asmeurer
  committed
  6a63d5c
- Update test_cupy.sh to run the vendoring tests
  asmeurer
  committed
  175f195
- Add a minor CHANGELOG entry
  asmeurer
  committed
  d1c7999
- Merge branch 'pytorch' of github.com:asmeurer/array-api-compat into pytorch
  asmeurer
  committed
  0e923b6
- Bump the version to 1.1
  asmeurer
  committed
  2fb0a0a
- Add a missing torch skip
  asmeurer
  committed
  3470b36

Commits

Commits on Jan 7, 2023

Commits on Jan 9, 2023

Commits on Jan 10, 2023

Commits on Jan 11, 2023

Commits on Jan 24, 2023

Commits on Feb 2, 2023

Commits on Feb 3, 2023

Commits on Feb 4, 2023

Commits on Feb 10, 2023

Commits on Feb 13, 2023

Commits on Feb 18, 2023

Commits on Feb 20, 2023

Commits on Feb 21, 2023

Commits on Feb 22, 2023

Commits on Feb 23, 2023

Commits on Feb 24, 2023

Commits on Feb 25, 2023