Merge pull request #14 from asmeurer/pytorch

WIP: PyTorch compatibility layer

Author: asmeurer

.github/workflows/array-api-tests-numpy.yml

@@ -0,0 +1,9 @@
+name: Array API Tests (NumPy)
+
+on: [push, pull_request]
+
+jobs:
+  array-api-tests-numpy:
+    uses: ./.github/workflows/array-api-tests.yml
+    with:
+      package-name: numpy

.github/workflows/array-api-tests-torch.yml

@@ -0,0 +1,10 @@
+name: Array API Tests (PyTorch)
+
+on: [push, pull_request]
+
+jobs:
+  array-api-tests-torch:
+    uses: ./.github/workflows/array-api-tests.yml
+    with:
+      package-name: torch
+      pytest-extra-args: "--disable-extension linalg"

.github/workflows/array-api-tests.yml

@@ -1,9 +1,18 @@
 name: Array API Tests
 
-on: [push, pull_request]
+on:
+  workflow_call:
+    inputs:
+      package-name:
+        required: true
+        type: string
+      pytest-extra-args:
+        required: false
+        type: string
+
 
 env:
-  PYTEST_ARGS: "--max-examples 200 -v -rxXfE --ci"
+  PYTEST_ARGS: "--max-examples 200 -v -rxXfE --ci ${{ inputs.pytest-extra-args }}"
 
 jobs:
   tests:
@@ -34,15 +43,15 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
-        python -m pip install numpy
+        python -m pip install ${{ inputs.package-name }}
         python -m pip install -r ${GITHUB_WORKSPACE}/array-api-tests/requirements.txt
-    - name: Run the array API testsuite (NumPy)
+    - name: Run the array API testsuite (${{ inputs.package-name }})
       env:
-        ARRAY_API_TESTS_MODULE: array_api_compat.numpy
+        ARRAY_API_TESTS_MODULE: array_api_compat.${{ inputs.package-name }}
         # This enables the NEP 50 type promotion behavior (without it a lot of
         # tests fail on bad scalar type promotion behavior)
         NPY_PROMOTION_STATE: weak
       run: |
         export PYTHONPATH="${GITHUB_WORKSPACE}/array-api-compat"
         cd ${GITHUB_WORKSPACE}/array-api-tests
-        pytest ${PYTEST_ARGS} --xfails-file ${GITHUB_WORKSPACE}/array-api-compat/numpy-xfails.txt array_api_tests/
+        pytest ${PYTEST_ARGS} --xfails-file ${GITHUB_WORKSPACE}/array-api-compat/${{ inputs.package-name }}-xfails.txt --skips-file ${GITHUB_WORKSPACE}/array-api-compat/${{ inputs.package-name }}-skips.txt array_api_tests/

.github/workflows/tests.yml

@@ -15,7 +15,7 @@ jobs:
       - name: Install Dependencies
         run: |
           python -m pip install --upgrade pip
-          python -m pip install pytest numpy
+          python -m pip install pytest numpy torch
 
       - name: Run Tests
         run: |

CHANGELOG.md

@@ -0,0 +1,24 @@
+# 1.1 (2023-02-24)
+
+## Major Changes
+
+- Added support for PyTorch.
+
+- Add helper function `size()` (required if torch is used as
+  `torch.Tensor.size` is a method that is incompatible with the array API
+  [`.size`](https://data-apis.org/array-api/latest/API_specification/generated/array_api.array.size.html#array_api.array.size)).
+
+- All wrapper functions that wrap existing library functions now pass through
+  arbitrary `**kwargs`.
+
+## Minor Changes
+
+- Added CI to run against the [array API testsuite](https://github.com/data-apis/array-api-tests).
+
+- Fix `sort(stable=False)` and `argsort(stable=False)` with CuPy.
+
+# 1.0 (2022-12-05)
+
+## Major Changes
+
+- Initial release. Includes support for NumPy and CuPy.