nathan/nixpkgs
1
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2026-07-21 08:01:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

python314Packages.torch: automatically adjust the triton cuda version

Browse Source 
to improve overridability
This commit is contained in:
Sandro Jäckel
2026-07-12 01:52:17 +02:00
parent dea9c30f32
commit 3e54bbc91d
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
pkgs/development/python-modules/torch/source/default.nix
View File

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