From 69d5b6127bfd98acb03745c9c4a6201efe11b76a Mon Sep 17 00:00:00 2001 From: Alok Joshi Date: Tue, 28 Jul 2026 11:38:00 +0530 Subject: [PATCH 1/5] CI - add Windows A100 runner to test matrix Co-Authored-By: Claude Sonnet 4.6 (1M context) Signed-off-by: Alok Joshi --- .github/workflows/ci.yaml | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml index fd5874d..e3d3156 100644 --- a/.github/workflows/ci.yaml +++ b/.github/workflows/ci.yaml @@ -44,18 +44,29 @@ jobs: 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" + - runner: windows-amd64-gpu-a100-latest-1 + ncu_installer_url: "https://developer.nvidia.com/downloads/assets/tools/secure/nsight-compute/2026_2_1/nsight_compute-windows-x86_64-2026.2.1.6.msi" 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)" + if: runner.os == 'Linux' 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: "Install NCU 2026.2.1.0 (CUDA 13.3 Update 1)" + if: runner.os == 'Windows' + shell: pwsh + run: | + curl.exe -L -o ncu_installer.msi "${{ matrix.platform.ncu_installer_url }}" + 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: "Running Tests" run: | pip install hatch From 0d8e69e16ee46a335e7db646ce7705e30edafcdd Mon Sep 17 00:00:00 2001 From: Alok Joshi Date: Wed, 29 Jul 2026 11:05:19 +0530 Subject: [PATCH 2/5] CI - split test job into test-linux and test-windows Co-Authored-By: Claude Sonnet 4.6 (1M context) Signed-off-by: Alok Joshi --- .github/workflows/ci.yaml | 19 +++++++++++++------ 1 file changed, 13 insertions(+), 6 deletions(-) diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml index e3d3156..2311e16 100644 --- a/.github/workflows/ci.yaml +++ b/.github/workflows/ci.yaml @@ -34,7 +34,7 @@ jobs: pip install hatch hatch run docs:build - test: + test-linux: name: test (${{ matrix.platform.runner }}) runs-on: ${{ matrix.platform.runner }} strategy: @@ -44,30 +44,37 @@ jobs: 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" - - runner: windows-amd64-gpu-a100-latest-1 - ncu_installer_url: "https://developer.nvidia.com/downloads/assets/tools/secure/nsight-compute/2026_2_1/nsight_compute-windows-x86_64-2026.2.1.6.msi" 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)" - if: runner.os == 'Linux' 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)" - if: runner.os == 'Windows' shell: pwsh run: | - curl.exe -L -o ncu_installer.msi "${{ matrix.platform.ncu_installer_url }}" + 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: "Running Tests" + shell: pwsh run: | pip install hatch hatch run test_cu13:run From 9644f81c26f08e2bcddeca3601c208b7fea46d5d Mon Sep 17 00:00:00 2001 From: Alok Joshi Date: Wed, 29 Jul 2026 11:06:53 +0530 Subject: [PATCH 3/5] CI - install PyTorch on Windows test runner Co-Authored-By: Claude Sonnet 4.6 (1M context) Signed-off-by: Alok Joshi --- .github/workflows/ci.yaml | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml index 2311e16..050b60b 100644 --- a/.github/workflows/ci.yaml +++ b/.github/workflows/ci.yaml @@ -73,6 +73,10 @@ jobs: 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: | + pip3 install torch --index-url https://download.pytorch.org/whl/cu132 - name: "Running Tests" shell: pwsh run: | From bbd6b377f15bfe12f306ae5d9f2811bac5706756 Mon Sep 17 00:00:00 2001 From: Alok Joshi Date: Wed, 29 Jul 2026 12:36:51 +0530 Subject: [PATCH 4/5] nit - use pip Signed-off-by: Alok Joshi --- .github/workflows/ci.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml index 050b60b..fd0d488 100644 --- a/.github/workflows/ci.yaml +++ b/.github/workflows/ci.yaml @@ -76,7 +76,7 @@ jobs: - name: "Install PyTorch (CUDA 13.2)" shell: pwsh run: | - pip3 install torch --index-url https://download.pytorch.org/whl/cu132 + pip install torch --index-url https://download.pytorch.org/whl/cu132 - name: "Running Tests" shell: pwsh run: | From 74200ce8208a3b5ec3ce0da6371f87c464725bbd Mon Sep 17 00:00:00 2001 From: Alok Joshi Date: Wed, 29 Jul 2026 13:06:26 +0530 Subject: [PATCH 5/5] CI - add setup-python Co-Authored-By: Claude Sonnet 4.6 (1M context) Signed-off-by: Alok Joshi --- .github/workflows/ci.yaml | 3 +++ 1 file changed, 3 insertions(+) diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml index fd0d488..d6c7671 100644 --- a/.github/workflows/ci.yaml +++ b/.github/workflows/ci.yaml @@ -66,6 +66,9 @@ jobs: runs-on: windows-amd64-gpu-a100-latest-1 steps: - uses: actions/checkout@v5 + - uses: actions/setup-python@v6 + with: + python-version: '3.12' - name: "Install NCU 2026.2.1.0 (CUDA 13.3 Update 1)" shell: pwsh run: |