Skip to content

Run Tests with GPU

Run Tests with GPU #25

name: Run Tests with GPU
on:
workflow_dispatch: # Allows for manual triggering
concurrency:
group: run-tests-gpu # Fixed group name to ensure only one instance runs
cancel-in-progress: false
jobs:
test:
runs-on: ubuntu-latest
steps:
- name: Set SSH permissions
run: |
mkdir -p ~/.ssh
chmod 700 ~/.ssh
sudo chown $USER:$USER ~/.ssh
- name: Checkout code
uses: actions/checkout@v3
- name: Set up Python 3.10
uses: actions/setup-python@v5
with:
python-version: "3.10"
- name: Install and configure dstack
run: |
pip install dstack
dstack config --url https://sky.dstack.ai --project ${{ secrets.DSTACK_PROJECT }} --token ${{ secrets.DSTACK_TOKEN }}
dstack init
- name: Run tests with GPU
run: |
HF_TOKEN=${{ secrets.HF_TOKEN }} dstack apply -f tests.dstack.yml --force -y
- name: Extract pytest logs
if: ${{ always() }}
run: |
dstack logs aana-tests | sed -n '/============================= test session starts ==============================/,$p'