python3Packages.torchWithRocm: ignore config.cudaSupport

2023-04-08 02:46:54 +03:00
parent a9faf1b9ef
commit 0f76efb481
1 changed files with 2 additions and 0 deletions
@@ -11856,6 +11856,7 @@ self: super: with self; {
  torchWithCuda = self.torch.override {
    magma = pkgs.magma-cuda;
    cudaSupport = true;
+    rocmSupport = false;
  };

  torchWithoutCuda = self.torch.override {
@@ -11865,6 +11866,7 @@ self: super: with self; {
  torchWithRocm = self.torch.override {
    magma = pkgs.magma-hip;
    rocmSupport = true;
+    cudaSupport = false;
  };

  torchWithoutRocm = self.torch.override {