Skip to content

feat(profiling): add support for pytorch profiling #36

feat(profiling): add support for pytorch profiling

feat(profiling): add support for pytorch profiling #36

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