treewide: use config.rocmSupport

Makes use of the per nixpkgs config flag to enable rocmSupport in
packages that support it.
This commit is contained in:
Martin Weinelt 2023-11-14 01:51:57 +01:00
parent e720baecf0
commit ea1cd4ef8e
No known key found for this signature in database
GPG Key ID: 87C1E9888F856759
5 changed files with 8 additions and 6 deletions

View File

@ -10,7 +10,7 @@
, cudaSupport ? config.cudaSupport
, cudaPackages ? { }
, rocmSupport ? false
, rocmSupport ? config.rocmSupport
, rocmPackages ? { }
, openclSupport ? false

View File

@ -8,8 +8,9 @@
, libxml2
, libffi
, makeWrapper
, config
, rocmPackages
, rocmSupport ? false
, rocmSupport ? config.rocmSupport
}:
let
inherit (llvmPackages_15) stdenv;

View File

@ -24,7 +24,8 @@
, libpthreadstubs
, magmaRelease
, ninja
, rocmSupport ? false
, config
, rocmSupport ? config.rocmSupport
, static ? false
, stdenv
, symlinkJoin

View File

@ -42,8 +42,9 @@
pythonOlder,
# ROCm dependencies
rocmSupport ? false,
gpuTargets ? [ ], rocmPackages
rocmSupport ? config.rocmSupport,
rocmPackages,
gpuTargets ? [ ]
}:
let

View File

@ -14070,7 +14070,6 @@ self: super: with self; {
toposort = callPackage ../development/python-modules/toposort { };
torch = callPackage ../development/python-modules/torch {
inherit (pkgs.config) cudaSupport;
magma =
if pkgs.config.cudaSupport
then pkgs.magma-cuda-static