From af814c73900a3e4796a0f2288cecdfa3d9490a4e Mon Sep 17 00:00:00 2001
From: doctorpangloss <@hiddenswitch.com>
Date: Wed, 22 May 2024 22:59:04 -0700
Subject: [PATCH] pytorch-triton-rocm now installs successfully

---
 setup.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 2e70ea4ed..c767fd12b 100644
--- a/setup.py
+++ b/setup.py
@@ -136,7 +136,8 @@ def dependencies(force_nightly: bool = False) -> List[str]:
                 _dependencies[i] = f"{stripped}=={existing_torch}"
                 break
         return _dependencies
-    _alternative_indices = [amd_torch_index, nvidia_torch_index]
+    # some torch packages redirect to https://download.pytorch.org/whl/
+    _alternative_indices = [amd_torch_index, nvidia_torch_index, ("https://download.pytorch.org/whl/", "https://download.pytorch.org/whl/")]
     session = PipSession()
 
     # (stable, nightly) tuple