mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 11:56:07 +00:00
49 lines
1.5 KiB
YAML
49 lines
1.5 KiB
YAML
name: JAX Array API
|
|
|
|
on:
|
|
push:
|
|
branches:
|
|
- main
|
|
pull_request:
|
|
branches:
|
|
- main
|
|
|
|
concurrency:
|
|
group: ${{ github.workflow }}-${{ github.head_ref || github.ref }}
|
|
cancel-in-progress: true
|
|
|
|
jobs:
|
|
build:
|
|
|
|
runs-on: ubuntu-latest
|
|
strategy:
|
|
matrix:
|
|
python-version: [3.11]
|
|
|
|
steps:
|
|
- name: Checkout jax
|
|
uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
|
|
- name: Checkout array-api-tests
|
|
uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
|
|
with:
|
|
repository: data-apis/array-api-tests
|
|
# TODO(jakevdp) update this to a stable release/tag when available.
|
|
ref: 'd982a6245400295477f5da5afa1c4a2a5e641ea4' # Latest commit as of 2025-01-30
|
|
submodules: 'true'
|
|
path: 'array-api-tests'
|
|
- name: Set up Python ${{ matrix.python-version }}
|
|
uses: actions/setup-python@42375524e23c412d93fb67b49958b491fce71c38 # v5.4.0
|
|
with:
|
|
python-version: ${{ matrix.python-version }}
|
|
- name: Install dependencies
|
|
run: |
|
|
pip install uv~=0.5.30
|
|
uv pip install --system .[ci] --system pytest-xdist -r array-api-tests/requirements.txt
|
|
- name: Run the test suite
|
|
env:
|
|
ARRAY_API_TESTS_MODULE: jax.numpy
|
|
JAX_ENABLE_X64: 'true'
|
|
run: |
|
|
cd ${GITHUB_WORKSPACE}/array-api-tests
|
|
pytest -n auto array_api_tests --derandomize --disable-deadline --skips-file ${GITHUB_WORKSPACE}/tests/array_api_skips.txt
|