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(
448
448
],
449
449
)
450
450
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
+
451
467
python_library(
452
468
name = "memory_planning",
453
469
srcs = [
@@ -456,6 +472,7 @@ python_library(
456
472
deps = [
457
473
"fbsource//third-party/pypi/tabulate:tabulate",
458
474
":memory_constraints",
475
+ ":memory_planning_algo",
459
476
":pass_utils",
460
477
"//caffe2:torch",
461
478
"//executorch/exir:lib",
You can’t perform that action at this time.
0 commit comments