From 86b7ebb4767d2ad3fa3c732289a55ec5769c3340 Mon Sep 17 00:00:00 2001
From: Eugene Indenbom <45334274+eindenbom@users.noreply.github.com>
Date: Mon, 13 Nov 2023 19:43:42 +0200
Subject: [PATCH] Relax triton requirements for compatibility with pytorch 2.1
 and newer (#1802)

---
 setup.py | 9 +--------
 1 file changed, 1 insertion(+), 8 deletions(-)

diff --git a/setup.py b/setup.py
index 35c173cdb..97e43ee68 100644
--- a/setup.py
+++ b/setup.py
@@ -14,7 +14,6 @@ def read_version(fname="whisper/version.py"):
 requirements = []
 whisper_rocm = os.getenv('WHISPER_ROCM',default='0')
 if sys.platform.startswith("linux") and platform.machine() == "x86_64":
-<<<<<<< HEAD
     from check_rocm_platform import is_command, check_amd_gpu_rocminfo, check_amd_gpu_lspci, check_rocm_packages
     ROCM_PLATFORM = False
     if is_command("rocminfo"):
@@ -30,14 +29,8 @@ def read_version(fname="whisper/version.py"):
         print("rocm")
         requirements.append("pytorch-triton-rocm>=2.0.1")
     else :
-        requirements.append("triton==2.0.0")
-=======
-    if whisper_rocm == "1" :
-        requirements.append("pytorch_triton_rocm==2.0.2")
-    else :
-        requirements.append("triton==2.0.0")
+        requirements.append("triton>=2.0.0,<3")
 
->>>>>>> c41de6ac767b30524779e260969ed4a4d6a43523
 setup(
     name="openai-whisper",
     py_modules=["whisper"],