diff options
Diffstat (limited to 'pkgs/development/python-modules/torch/default.nix')
-rw-r--r-- | pkgs/development/python-modules/torch/default.nix | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/torch/default.nix b/pkgs/development/python-modules/torch/default.nix index c5d591c3f27a..c9bf7c45766f 100644 --- a/pkgs/development/python-modules/torch/default.nix +++ b/pkgs/development/python-modules/torch/default.nix @@ -40,8 +40,9 @@ pythonOlder, # ROCm dependencies - rocmSupport ? false, - gpuTargets ? [ ], rocmPackages + rocmSupport ? config.rocmSupport, + rocmPackages, + gpuTargets ? [ ] }: let |