Skip to content

Commit b5ea0f3

Browse files
committed
triton: updating to matching version
1 parent 8579d71 commit b5ea0f3

2 files changed

Lines changed: 16 additions & 2 deletions

File tree

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
diff --git a/requirements/cuda.txt b/requirements/cuda.txt
2+
index 2dd380d14..22477dc82 100644
3+
--- a/requirements/cuda.txt
4+
+++ b/requirements/cuda.txt
5+
@@ -11,9 +11,6 @@ torchaudio==2.10.0
6+
torchvision==0.25.0 # Required for phi3v processor. See https://github.com/pytorch/vision?tab=readme-ov-file#installation for corresponding version
7+
# FlashInfer should be updated together with the Dockerfile
8+
flashinfer-python==0.6.4
9+
-# Cap nvidia-cudnn-frontend (transitive dep of flashinfer) due to
10+
-# breaking changes in 1.19.0
11+
-nvidia-cudnn-frontend>=1.13.0,<1.19.0
12+
13+
# QuACK and Cutlass DSL for FA4 (cute-DSL implementation)
14+
nvidia-cutlass-dsl>=4.4.0.dev1

pkgs/development/python-modules/vllm/default.nix

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -179,8 +179,8 @@ let
179179
triton-kernels = fetchFromGitHub {
180180
owner = "triton-lang";
181181
repo = "triton";
182-
tag = "v3.5.0";
183-
hash = "sha256-F6T0n37Lbs+B7UHNYzoIQHjNNv3TcMtoXjNrT8ZUlxY=";
182+
tag = "v3.6.0";
183+
hash = "sha256-JFSpQn+WsNnh7CAPlcpOcUp0nyKXNbJEANdXqmkt4Tc=";
184184
};
185185

186186
# grep for GIT_TAG in the following file

0 commit comments

Comments
 (0)