[nix] downgrade pytorch and accelerate

- accelerate 0.33.0 -> 0.32.0 - pytorch 2.4.0 -> 2.3.1 Signed-off-by: Avimitin <[email protected]>
chipsalliance · Sep 9, 2024 · b18bb10 · b18bb10
1 parent b692adc
commit b18bb10
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/nix/pkgs/buddy-mlir.nix b/nix/pkgs/buddy-mlir.nix
@@ -24,6 +24,31 @@ let
           hash = "sha256-Z78I9S8g9WexoX6HhxwbOD0K0awCTzsqW1ZiWObQNw0=";
         };
       }));
+
+      accelerate = (prev.accelerate.overridePythonAttrs (old: rec {
+        pname = "accelerate";
+        version = "0.32.0";
+
+        src = fetchFromGitHub {
+          owner = "huggingface";
+          repo = "accelerate";
+          rev = "refs/tags/v${version}";
+          hash = "sha256-/Is5aKTYHxvgUJSkF7HxMbEA6dgn/y5F1B3D6qSCSaE=";
+        };
+      }));
+
+      torch = (prev.torch.overridePythonAttrs (old: rec {
+        version = "2.3.1";
+        src = fetchFromGitHub {
+          owner = "pytorch";
+          repo = "pytorch";
+          rev = "refs/tags/v${version}";
+          fetchSubmodules = true;
+          hash = "sha256-vpgtOqzIDKgRuqdT8lB/g6j+oMIH1RPxdbjtlzZFjV8=";
+        };
+        PYTORCH_BUILD_VERSION = version;
+        PYTORCH_BUILD_NUMBER = 0;
+      }));
     };
   };
 

diff --git a/tests/pytorch/lenet/lenet.py b/tests/pytorch/lenet/lenet.py
@@ -5,6 +5,7 @@
 import numpy as np
 import torch
 from torch._inductor.decomposition import decompositions as inductor_decomp
+import torch._inductor.lowering
 
 from buddy.compiler.frontend import DynamoCompiler
 from buddy.compiler.graph import GraphDriver