forked from pytorch/xla
-
Notifications
You must be signed in to change notification settings - Fork 0
48 lines (46 loc) · 1.3 KB
/
torch_xla2.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
47
48
on:
pull_request:
branches:
- master
- r[0-9]+.[0-9]+
paths:
- 'experimental/torch_xla2/**'
push:
branches:
- master
- r[0-9]+.[0-9]+
paths:
- 'experimental/torch_xla2/**'
workflow_dispatch:
concurrency:
group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.sha }}-${{ github.event_name == 'workflow_dispatch' }}-${{ github.event_name == 'schedule' }}
cancel-in-progress: true
jobs:
torchxla2-cpu:
runs-on: ubuntu-20.04
strategy:
matrix:
python-version: ['3.10']
steps:
- name: Checkout repo
uses: actions/checkout@v4
with:
sparse-checkout: |
experimental/torch_xla2
- name: Setup Python
uses: actions/setup-python@v4
with:
python-version: ${{ matrix.python-version }}
- name: Install
shell: bash
working-directory: experimental/torch_xla2
run: |
pip install -r test-requirements.txt
pip install -e .[cpu]
pip install tensorflow-cpu # for TF integrations tests
- name: Run tests
working-directory: experimental/torch_xla2
shell: bash
run: |
pytest test/
XLA_FLAGS=--xla_force_host_platform_device_count=4 pytest -n 0 test_dist/