movefasta/nixpkgs
1
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2025-06-09 19:13:26 +03:00
Code Issues Projects Releases Packages Wiki Activity

tree-wide: cudaPackages.flags updates

Browse source 
Signed-off-by: Connor Baker <ConnorBaker01@gmail.com>
This commit is contained in:
Connor Baker 2025-05-15 20:49:17 +00:00
parent 765529dfff
commit ead6581362
7 changed files with 8 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

2
pkgs/applications/science/misc/colmap/default.nix
View file

@ -46,7 +46,7 @@ stdenv'.mkDerivation rec {
cmakeFlags = lib.optionals cudaSupport [
(lib.cmakeBool "CUDA_ENABLED" true)
(lib.cmakeFeature "CMAKE_CUDA_ARCHITECTURES" (
lib.strings.concatStringsSep ";" (map cudaPackages.flags.dropDot cudaCapabilities)
lib.strings.concatStringsSep ";" (map cudaPackages.flags.dropDots cudaCapabilities)
))
];

2
pkgs/by-name/dl/dlib/package.nix
View file

@ -42,7 +42,7 @@
]
++ lib.optionals cudaSupport [
(lib.cmakeFeature "DLIB_USE_CUDA_COMPUTE_CAPABILITIES" (
builtins.concatStringsSep "," (with cudaPackages.flags; map dropDot cudaCapabilities)
builtins.concatStringsSep "," (with cudaPackages.flags; map dropDots cudaCapabilities)
))
];

4
pkgs/by-name/gp/gpu-burn/package.nix
View file

@ -17,7 +17,7 @@ let
cuda_nvcc
libcublas
;
inherit (cudaPackages.flags) cudaCapabilities dropDot isJetsonBuild;
inherit (cudaPackages.flags) cudaCapabilities dropDots isJetsonBuild;
in
backendStdenv.mkDerivation {
pname = "gpu-burn";
@ -53,7 +53,7 @@ backendStdenv.mkDerivation {
makeFlags = [
"CUDAPATH=${getBin cuda_nvcc}"
"COMPUTE=${last (map dropDot cudaCapabilities)}"
"COMPUTE=${last (map dropDots cudaCapabilities)}"
"IS_JETSON=${boolToString isJetsonBuild}"
];

2
pkgs/by-name/mi/mistral-rs/package.nix
View file

@ -60,7 +60,7 @@ let
))
else
cudaCapability;
cudaCapability' = lib.toInt (cudaPackages.flags.dropDot cudaCapabilityString);
cudaCapability' = lib.toInt (cudaPackages.flags.dropDots cudaCapabilityString);
mklSupport =
assert accelIsValid;

2
pkgs/by-name/mo/moshi/package.nix
View file

@ -38,7 +38,7 @@ let
))
else
cudaCapability;
cudaCapability' = lib.toInt (cudaPackages.flags.dropDot cudaCapabilityString);
cudaCapability' = lib.toInt (cudaPackages.flags.dropDots cudaCapabilityString);
in
rustPlatform.buildRustPackage (finalAttrs: {
pname = "moshi";

2
pkgs/by-name/ti/tiny-cuda-nn/package.nix
View file

@ -43,7 +43,7 @@ let
cudaCapabilities = lists.subtractLists unsupportedCudaCapabilities flags.cudaCapabilities;
cudaArchitecturesString = strings.concatMapStringsSep ";" flags.dropDot cudaCapabilities;
cudaArchitecturesString = strings.concatMapStringsSep ";" flags.dropDots cudaCapabilities;
in
stdenv.mkDerivation (finalAttrs: {
pname = "tiny-cuda-nn";

2
pkgs/development/libraries/science/math/magma/generic.nix
View file

@ -94,7 +94,7 @@ let
minArch =
let
# E.g. [ "80" "86" "90" ]
cudaArchitectures = (builtins.map flags.dropDot flags.cudaCapabilities);
cudaArchitectures = (builtins.map flags.dropDots flags.cudaCapabilities);
minArch' = builtins.head (builtins.sort strings.versionOlder cudaArchitectures);
in
# "75" -> "750" Cf. https://github.com/icl-utk-edu/magma/blob/v2.9.0/CMakeLists.txt#L200-L201