Skip to content

Move op dispatching logic into an Environment class; and use Mode to capture dispatcher instead of tensor. #113

Move op dispatching logic into an Environment class; and use Mode to capture dispatcher instead of tensor.

Move op dispatching logic into an Environment class; and use Mode to capture dispatcher instead of tensor. #113

Workflow file for this run

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
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: '3.10'
- name: Install
shell: bash
working-directory: experimental/torch_xla2
run: |
pip install -r test-requirements.txt
pip install -e .[cpu]
- name: Run tests
working-directory: experimental/torch_xla2
shell: bash
run: |
pytest test/