From a9cb55c19b0907b8e544c72f7399ff5b2d0410fb Mon Sep 17 00:00:00 2001
From: LosCrossos <165311345+loscrossos@users.noreply.github.com>
Date: Fri, 4 Jul 2025 11:53:09 +0200
Subject: [PATCH] adding changes for adding blackwell support for precompiled
 wheels.

Signed-off-by: LosCrossos <165311345+loscrossos@users.noreply.github.com>
---
 .github/workflows/wheels_build.yml | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/.github/workflows/wheels_build.yml b/.github/workflows/wheels_build.yml
index c9ecc25b7d..0f1471f6ab 100644
--- a/.github/workflows/wheels_build.yml
+++ b/.github/workflows/wheels_build.yml
@@ -56,6 +56,9 @@ jobs:
       - if: contains(inputs.toolkit_type, 'cuda') && fromJSON(inputs.toolkit_short_version) >= 120
         run: |
           echo "TORCH_CUDA_ARCH_LIST=$TORCH_CUDA_ARCH_LIST 9.0a" >> ${GITHUB_ENV}
+      - if: contains(inputs.toolkit_type, 'cuda') && fromJSON(inputs.toolkit_short_version) >= 128
+        run: |
+          echo "TORCH_CUDA_ARCH_LIST=$TORCH_CUDA_ARCH_LIST 10.0 12.0" >> ${GITHUB_ENV}
 
       - if: runner.os == 'Windows'
         run: git config --system core.longpaths true