-
Notifications
You must be signed in to change notification settings - Fork 2.8k
46 lines (42 loc) · 1.37 KB
/
jax-array-api.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
name: JAX Array API
on:
workflow_dispatch: # allows triggering the workflow run manually
pull_request: # Automatically trigger on pull requests affecting this file
branches:
- main
paths:
- '**workflows/jax-array-api.yml'
- '**experimental/array_api/**'
jobs:
build:
runs-on: ubuntu-latest
strategy:
matrix:
python-version: [3.11]
steps:
- name: Checkout jax
uses: actions/checkout@v3
- name: Checkout array-api-tests
uses: actions/checkout@v3
with:
repository: data-apis/array-api-tests
submodules: 'true'
path: 'array-api-tests'
- name: Fix array-apis bug
# Temporary workaround for https://github.com/data-apis/array-api/issues/631
run: |
sed -i -e 's/\\/\\\\/g' array-api-tests/array-api/spec/API_specification/signatures/*.py
- name: Set up Python ${{ matrix.python-version }}
uses: actions/setup-python@v1
with:
python-version: ${{ matrix.python-version }}
- name: Install dependencies
run: |
python -m pip install .[cpu]
python -m pip install -r array-api-tests/requirements.txt
- name: Run the test suite
env:
ARRAY_API_TESTS_MODULE: jax.experimental.array_api
JAX_ENABLE_X64: 'true'
run: |
pytest --ci array-api-tests/array_api_tests/ --skips-file array-api-skips.txt