gador/nixpkgs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Simplify TORCH_CUDA_ARCH_LIST - function call already returns a string

Browse Source
This commit is contained in:
Break Yang
2025-05-20 14:33:32 -07:00
parent e511952b61
commit 4875dd88d8
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkgs/development/python-modules/flashinfer/default.nix
+1 -1
View File
@@ -77,7 +77,7 @@ buildPythonPackage {
'';
CUDA_HOME = "${cudaPackages.cudatoolkit}";
TORCH_CUDA_ARCH_LIST = "${lib.concatStringsSep ";" torch.cudaCapabilities}";
TORCH_CUDA_ARCH_LIST = lib.concatStringsSep ";" torch.cudaCapabilities;
dependencies = [
numpy