Add memory planning tests to Cadence

Differential Revision: D67183235 Pull Request resolved: #7431
pytorch · Dec 27, 2024 · 40e956f · 40e956f
1 parent ed15042
commit 40e956f
Show file tree

Hide file tree

Showing 2 changed files with 623 additions and 0 deletions.
diff --git a/backends/cadence/aot/TARGETS b/backends/cadence/aot/TARGETS
@@ -409,3 +409,22 @@ python_library(
         "//executorch/exir:tensor",
     ],
 )
+
+
+python_unittest(
+    name = "test_memory_passes",
+    srcs = [
+        "tests/test_memory_passes.py",
+    ],
+    typing = True,
+    deps = [
+        ":compiler",
+        ":memory_planning",
+        ":ops_registrations",
+        ":pass_utils",
+        "//caffe2:torch",
+        "//executorch/exir:memory",
+        "//executorch/exir/dialects:lib",
+        "//executorch/exir/tests:models",
+    ],
+)