From 3e54bbc91de53b06960d9002c25877310d05c2d5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sandro=20J=C3=A4ckel?= Date: Sun, 12 Jul 2026 01:52:17 +0200 Subject: [PATCH] python314Packages.torch: automatically adjust the triton cuda version to improve overridability --- pkgs/development/python-modules/torch/source/default.nix | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pkgs/development/python-modules/torch/source/default.nix b/pkgs/development/python-modules/torch/source/default.nix index b6013d8f2ebd..f8d9d63a873b 100644 --- a/pkgs/development/python-modules/torch/source/default.nix +++ b/pkgs/development/python-modules/torch/source/default.nix @@ -81,7 +81,7 @@ # (dependencies without cuda support). # Instead we should rely on overlays and nixpkgsFun. # (@SomeoneSerge) - _tritonEffective ? if cudaSupport then triton-cuda else triton, + _tritonEffective ? if cudaSupport then triton-cuda.override { inherit cudaPackages; } else triton, triton-cuda, # Disable MKLDNN on aarch64-darwin, it negatively impacts performance,