pytorch_gpu_tests.yml

name: Pytorch Unit Tests (with GPU) 

on:
  push:
    branches:
      - 'main'
      - 'mq-working-branch**'
    paths:
      - 'ddtrace/profiling/collector/pytorch.py'
  pull_request:
    paths:
      - 'ddtrace/profiling/collector/pytorch.py'
  workflow_dispatch: 

jobs:
   unit-tests:
    runs-on: APM-4-CORE-GPU-LINUX
    steps:
      - uses: actions/checkout@v4
        # Include all history and tags
        with:
          persist-credentials: false
          fetch-depth: 0

      - uses: actions/setup-python@v5
        name: Install Python
        with:
          python-version: '3.12'

      - uses: actions-rust-lang/setup-rust-toolchain@v1
      - name: Install latest stable toolchain and rustfmt
        run: rustup update stable && rustup default stable && rustup component add rustfmt clippy

      - name: Install hatch
        uses: pypa/hatch@install
        with:
          version: "1.12.0"
      
      - name: Install PyTorch
        run: pip install torch

      - name: Run tests
        run: hatch run profiling_pytorch:test