diff options
author | Connor Baker <connor.baker@tweag.io> | 2024-01-18 15:43:03 +0000 |
---|---|---|
committer | Connor Baker <connor.baker@tweag.io> | 2024-01-18 15:43:06 +0000 |
commit | c9dca80340c944308a687261138d19d71998ff8d (patch) | |
tree | 4d867e0a8d7b1273da756e876d31328e73bd1cfd /pkgs/development/cuda-modules/generic-builders | |
parent | 3498ca649e1b360e8e13545f689c5c29b201cacd (diff) |
cudaPackages: fix multiplex builder isSupported check
Co-Authored-By: Yann Hamdaoui <yann.hamdaoui@tweag.io>
Diffstat (limited to 'pkgs/development/cuda-modules/generic-builders')
-rw-r--r-- | pkgs/development/cuda-modules/generic-builders/multiplex.nix | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/pkgs/development/cuda-modules/generic-builders/multiplex.nix b/pkgs/development/cuda-modules/generic-builders/multiplex.nix index 6353b07545a4a..b8ac84bda9133 100644 --- a/pkgs/development/cuda-modules/generic-builders/multiplex.nix +++ b/pkgs/development/cuda-modules/generic-builders/multiplex.nix @@ -63,11 +63,12 @@ let # computeName :: Package -> String computeName = {version, ...}: mkVersionedPackageName pname version; - # Check whether a package supports our CUDA version + # Check whether a package supports our CUDA version and platform. # isSupported :: Package -> Bool isSupported = package: - !(strings.hasPrefix "unsupported" package.platform) + # The `platform` attribute of the package is NVIDIA's name for a redistributable architecture. + redistArch == package.platform && strings.versionAtLeast cudaVersion package.minCudaVersion && strings.versionAtLeast package.maxCudaVersion cudaVersion; |