Adding torch accelerator and requirements file to FSDP2 example (#1375) #903
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
name: Run Python Examples | |
on: | |
push: | |
branches: [ main ] | |
pull_request: | |
branches: [ main ] | |
schedule: | |
# Every day at 3:00am | |
- cron: '0 3 * * *' | |
jobs: | |
test: | |
runs-on: ubuntu-latest | |
steps: | |
- uses: actions/checkout@v4 | |
- name: Set up Python 3.10 | |
uses: actions/setup-python@v5 | |
with: | |
python-version: '3.10' | |
- name: Install uv | |
uses: astral-sh/setup-uv@v6 | |
- name: Run Tests | |
env: | |
VIRTUAL_ENV: '.venv' | |
PIP_INSTAL_ARGS: '--pre -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html' | |
run: | | |
./run_python_examples.sh | |
- name: Open issue on failure | |
if: ${{ failure() && github.event_name == 'schedule' }} | |
uses: rishabhgupta/git-action-issue@v2 | |
with: | |
token: ${{ secrets.GITHUB_TOKEN }} | |
title: Daily CI failed | |
body: Commit ${{ github.sha }} daily scheduled [CI run](https://github.com/${{ github.repository }}/actions/runs/${{ github.run_id }}) failed, please check why | |
assignees: '' |