Skip to content

GitLab

Unverified Commit 8db1bf32 authored by Roy's avatar Roy Committed by GitHub
Browse files

[Misc] Upgrade triton to 2.2.0 (#4061)

parent aceb17cf
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
requirements-cpu.txt
View file @ 8db1bf32
...@@ -3,4 +3,4 @@ ...@@ -3,4 +3,4 @@
# Dependencies for x86_64 CPUs # Dependencies for x86_64 CPUs
torch == 2.2.1+cpu 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.
\ No newline at end of file \ No newline at end of file
requirements-cuda.txt
View file @ 8db1bf32
...@@ -7,4 +7,3 @@ pynvml == 11.5.0 ...@@ -7,4 +7,3 @@ pynvml == 11.5.0
vllm-nccl-cu12>=2.18,<2.19 # for downloading nccl library vllm-nccl-cu12>=2.18,<2.19 # for downloading nccl library
torch == 2.2.1 torch == 2.2.1
xformers == 0.0.25 # Requires PyTorch 2.2.1 xformers == 0.0.25 # Requires PyTorch 2.2.1
triton >= 2.1.0
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment