BoomMicrophone/nixpkgs-suyu
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

cudaPackages.cudaFlags: drop unused capabilitiesAndForward

Browse source
This commit is contained in:
Someone Serge 2023-03-03 12:29:11 +02:00
parent 0c25f5aa7f
commit ac64f07f9c
No known key found for this signature in database
GPG key ID: 7B0E3B1390D61DA4
1 changed files with 4 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
pkgs/development/compilers/cudatoolkit/flags.nix
View file

@ -19,13 +19,14 @@ let
# passing a configuration based on your specific GPU environment.
#
# config.cudaCapabilities :: List Capability
# List of hardware generations to build
# Last item is considered the optional forward-compatibility arch
# List of hardware generations to build.
# E.g. [ "8.0" ]
# Currently, the last item is considered the optional forward-compatibility arch,
# but this may change in the future.
#
# config.cudaForwardCompat :: Bool
# Whether to include the forward compatibility gencode (+PTX)
# to support future GPU generations:
# to support future GPU generations.
# E.g. true
#
# Please see the accompanying documentation or https://github.com/NixOS/nixpkgs/pull/205351
@ -91,17 +92,6 @@ let
formatCapabilities = { cudaCapabilities, enableForwardCompat ? true }: rec {
inherit cudaCapabilities enableForwardCompat;
# forwardCapability :: String
# Forward "compute" capability, a.k.a PTX
# E.g. "8.6+PTX"
forwardCapability = (lists.last cudaCapabilities) + "+PTX";
# capabilitiesAndForward :: List String
# The list of supported CUDA architectures, including the forward compatibility architecture.
# If forward compatibility is disabled, this will be the same as cudaCapabilities.
# E.g. [ "7.5" "8.6" "8.6+PTX" ]
capabilitiesAndForward = cudaCapabilities ++ lists.optionals enableForwardCompat [ forwardCapability ];
# archNames :: List String
# E.g. [ "Turing" "Ampere" ]
archNames = lists.unique (builtins.map (cap: cudaComputeCapabilityToName.${cap}) cudaCapabilities);
@ -141,9 +131,6 @@ assert (formatCapabilities { cudaCapabilities = [ "7.5" "8.6" ]; }) == {
cudaCapabilities = [ "7.5" "8.6" ];
enableForwardCompat = true;
capabilitiesAndForward = [ "7.5" "8.6" "8.6+PTX" ];
forwardCapability = "8.6+PTX";
archNames = [ "Turing" "Ampere" ];
realArches = [ "sm_75" "sm_86" ];
virtualArches = [ "compute_75" "compute_86" ];