[Misc] Upgrade triton to 2.2.0 (#4061)

8db1bf32 · Roy · GitHub · aceb17cf · 8db1bf32 · 8db1bf32
Unverified Commit 8db1bf32 authored Apr 15, 2024 by Roy Committed by GitHub Apr 14, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

requirements-cpu.txt requirements-cpu.txt +1 -1

requirements-cuda.txt requirements-cuda.txt +0 -1

No files found.
--- a/requirements-cpu.txt
+++ b/requirements-cpu.txt
@@ -3,4 +3,4 @@

 # Dependencies for x86_64 CPUs
 torch == 2.2.1+cpu
-triton >= 2.1.0  # FIXME(woosuk): This is a hack to avoid import error.
\ No newline at end of file
+triton >= 2.2.0  # FIXME(woosuk): This is a hack to avoid import error.
\ No newline at end of file
--- a/requirements-cuda.txt
+++ b/requirements-cuda.txt
@@ -7,4 +7,3 @@ pynvml == 11.5.0
 vllm-nccl-cu12>=2.18,<2.19  # for downloading nccl library
 torch == 2.2.1
 xformers == 0.0.25  # Requires PyTorch 2.2.1
-triton >= 2.1.0