nit - use pip #137
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
| # SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved. | |
| # SPDX-License-Identifier: Apache-2.0 | |
| name: "CI" | |
| on: | |
| push: | |
| branches: | |
| - "main" | |
| - "pull-request/[0-9]+" | |
| jobs: | |
| lint: | |
| runs-on: ubuntu-latest | |
| steps: | |
| - uses: actions/checkout@v5 | |
| - uses: actions/setup-python@v6 | |
| with: | |
| python-version: '3.10' | |
| - name: "Running Lint" | |
| run: | | |
| pip install hatch | |
| hatch run lint:check | |
| build-docs: | |
| runs-on: ubuntu-latest | |
| steps: | |
| - uses: actions/checkout@v5 | |
| - uses: actions/setup-python@v6 | |
| with: | |
| python-version: '3.10' | |
| - name: "Building Docs" | |
| run: | | |
| pip install hatch | |
| hatch run docs:build | |
| test-linux: | |
| name: test (${{ matrix.platform.runner }}) | |
| runs-on: ${{ matrix.platform.runner }} | |
| strategy: | |
| matrix: | |
| platform: | |
| - runner: linux-amd64-gpu-h100-latest-1 | |
| ncu_installer_url: "https://developer.nvidia.com/downloads/assets/tools/secure/nsight-compute/2026_2_1/nsight_compute-linux-x86_64-2026.2.1.6.run" | |
| - runner: linux-arm64-gpu-l4-latest-1 | |
| ncu_installer_url: "https://developer.nvidia.com/downloads/assets/tools/secure/nsight-compute/2026_2_1/nsight_compute-linux-sbsa-2026.2.1.6.run" | |
| container: | |
| image: nvcr.io/nvidia/pytorch:26.06-py3 | |
| options: "--cap-add=SYS_ADMIN --gpus all" | |
| steps: | |
| - uses: actions/checkout@v5 | |
| - name: "Install NCU 2026.2.1.0 (CUDA 13.3 Update 1)" | |
| run: | | |
| curl -L -o ncu_installer.run "${{ matrix.platform.ncu_installer_url }}" | |
| chmod +x ncu_installer.run | |
| ./ncu_installer.run -- -noprompt | |
| rm ncu_installer.run | |
| echo "/usr/local/NVIDIA-Nsight-Compute-2026.2" >> $GITHUB_PATH | |
| - name: "Running Tests" | |
| run: | | |
| pip install hatch | |
| hatch run test_cu13:run | |
| test-windows: | |
| name: test (windows-amd64-gpu-a100-latest-1) | |
| runs-on: windows-amd64-gpu-a100-latest-1 | |
| steps: | |
| - uses: actions/checkout@v5 | |
| - name: "Install NCU 2026.2.1.0 (CUDA 13.3 Update 1)" | |
| shell: pwsh | |
| run: | | |
| curl.exe -L -o ncu_installer.msi "https://developer.nvidia.com/downloads/assets/tools/secure/nsight-compute/2026_2_1/nsight_compute-windows-x86_64-2026.2.1.6.msi" | |
| Start-Process msiexec.exe -Wait -ArgumentList '/i ncu_installer.msi /quiet /norestart' | |
| Remove-Item ncu_installer.msi | |
| "C:\Program Files\NVIDIA Corporation\Nsight Compute 2026.2.1" | Out-File -FilePath $env:GITHUB_PATH -Encoding utf8 -Append | |
| - name: "Install PyTorch (CUDA 13.2)" | |
| shell: pwsh | |
| run: | | |
| pip install torch --index-url https://download.pytorch.org/whl/cu132 | |
| - name: "Running Tests" | |
| shell: pwsh | |
| run: | | |
| pip install hatch | |
| hatch run test_cu13:run |