[Misc] Upgrade triton to 2.2.0 (#4061)

This commit is contained in:
Roy 2024-04-15 08:43:54 +08:00 committed by GitHub
parent aceb17cf2d
commit 8db1bf32f8
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 1 additions and 2 deletions

View File

@ -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.
triton >= 2.2.0 # FIXME(woosuk): This is a hack to avoid import error.

View File

@ -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