File tree Expand file tree Collapse file tree 3 files changed +326
-224
lines changed Expand file tree Collapse file tree 3 files changed +326
-224
lines changed Original file line number Diff line number Diff 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+
451467python_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",
You can’t perform that action at this time.
0 commit comments