From d6b3f065303b984e5a0b6fd842587e163008c88a Mon Sep 17 00:00:00 2001
From: Daniel Vega-Myhre <danvm@meta.com>
Date: Fri, 5 Dec 2025 15:35:47 -0800
Subject: [PATCH] update cuda versions for upload-to-pypi in linux wheel build

---
 .github/workflows/build_wheels_linux.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/build_wheels_linux.yml b/.github/workflows/build_wheels_linux.yml
index 2ae04b4461..10a072d89b 100644
--- a/.github/workflows/build_wheels_linux.yml
+++ b/.github/workflows/build_wheels_linux.yml
@@ -57,7 +57,7 @@ jobs:
       package-name: torchao
       trigger-event: ${{ github.event_name }}
       # This is the CUDA version to be uploaded to torchao-nightly pypi
-      upload-to-pypi: cu121
+      upload-to-pypi: cu121,cu126,cu128,cu129,cu130
     secrets:
       PYPI_API_TOKEN: ${{ secrets.PYPI_API_TOKEN }}
   notify: