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:
parent
e720baecf0
commit
ea1cd4ef8e
5 changed files with 8 additions and 6 deletions
|
@ -10,7 +10,7 @@
|
|||
, cudaSupport ? config.cudaSupport
|
||||
, cudaPackages ? { }
|
||||
|
||||
, rocmSupport ? false
|
||||
, rocmSupport ? config.rocmSupport
|
||||
, rocmPackages ? { }
|
||||
|
||||
, openclSupport ? false
|
||||
|
|
|
@ -8,8 +8,9 @@
|
|||
, libxml2
|
||||
, libffi
|
||||
, makeWrapper
|
||||
, config
|
||||
, rocmPackages
|
||||
, rocmSupport ? false
|
||||
, rocmSupport ? config.rocmSupport
|
||||
}:
|
||||
let
|
||||
inherit (llvmPackages_15) stdenv;
|
||||
|
|
|
@ -24,7 +24,8 @@
|
|||
, libpthreadstubs
|
||||
, magmaRelease
|
||||
, ninja
|
||||
, rocmSupport ? false
|
||||
, config
|
||||
, rocmSupport ? config.rocmSupport
|
||||
, static ? false
|
||||
, stdenv
|
||||
, symlinkJoin
|
||||
|
|
|
@ -42,8 +42,9 @@
|
|||
pythonOlder,
|
||||
|
||||
# ROCm dependencies
|
||||
rocmSupport ? false,
|
||||
gpuTargets ? [ ], rocmPackages
|
||||
rocmSupport ? config.rocmSupport,
|
||||
rocmPackages,
|
||||
gpuTargets ? [ ]
|
||||
}:
|
||||
|
||||
let
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in a new issue