Skip to content

Commit be07160

Browse files
authored
Create a MemoryPlanningAlgo class.
Differential Revision: D76954785 Pull Request resolved: #11824
1 parent da36d8a commit be07160

File tree

3 files changed

+326
-224
lines changed

3 files changed

+326
-224
lines changed

backends/cadence/aot/TARGETS

Lines changed: 17 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -448,6 +448,22 @@ python_unittest(
448448
],
449449
)
450450

451+
python_library(
452+
name = "memory_planning_algo",
453+
srcs = [
454+
"memory_planning_algo.py",
455+
],
456+
deps = [
457+
":memory_constraints",
458+
":pass_utils",
459+
"//executorch/exir:lib",
460+
"//executorch/exir:memory_planning",
461+
"//executorch/exir:tensor",
462+
"//executorch/exir/passes:lib",
463+
"fbsource//third-party/pypi/tabulate:tabulate",
464+
],
465+
)
466+
451467
python_library(
452468
name = "memory_planning",
453469
srcs = [
@@ -456,6 +472,7 @@ python_library(
456472
deps = [
457473
"fbsource//third-party/pypi/tabulate:tabulate",
458474
":memory_constraints",
475+
":memory_planning_algo",
459476
":pass_utils",
460477
"//caffe2:torch",
461478
"//executorch/exir:lib",

0 commit comments

Comments
 (0)