about summary refs log tree commit diff
path: root/pkgs/development/python-modules/torch
diff options
context:
space:
mode:
authorgithub-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>2023-11-14 18:01:11 +0000
committerGitHub <noreply@github.com>2023-11-14 18:01:11 +0000
commit6cc772a6fa213f2624f490735f4cf49426bcc0e0 (patch)
tree398d8785581708be3eb823fcffdc18b0f8726e65 /pkgs/development/python-modules/torch
parent0523b2e6c2ea7730352fa32e8a921ec331569571 (diff)
parentcd014bdb858891bc7a3f0c1267ef8ad4fd11f19c (diff)
Merge master into staging-next
Diffstat (limited to 'pkgs/development/python-modules/torch')
-rw-r--r--pkgs/development/python-modules/torch/default.nix5
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 c5d591c3f27a1..c9bf7c45766f0 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