Qualcomm AI Engine Direct - Fix aihub path failing due to memory planning pass (pytorch#5482)

winskuo-quic · facebook-github-bot · commit 90dcea5488a5 · 2024-09-24T11:50:52.000-07:00
Summary: Update MemoryPlanningPass to use default greedy algo. Pull Request resolved: pytorch#5482 Reviewed By: kirklandsign Differential Revision: D63270305 Pulled By: cccclai fbshipit-source-id: 7fb03365e3cadb53e4879feb7f105aadc9e9497c
diff --git a/examples/qualcomm/qaihub_scripts/utils/utils.py b/examples/qualcomm/qaihub_scripts/utils/utils.py
@@ -68,7 +68,6 @@ def gen_pte_from_ctx_bin(
     for pte_name in pte_names:
         print(f"{pte_name} generating...")
         memory_planning_pass = MemoryPlanningPass(
-            memory_planning_algo="greedy",
             alloc_graph_input=False,
             alloc_graph_output=False,
         )

Original file line number	Diff line number	Diff line change
`@@ -68,7 +68,6 @@ def gen_pte_from_ctx_bin(`
`68`	`68`	`for pte_name in pte_names:`
`69`	`69`	`print(f"{pte_name} generating...")`
`70`	`70`	`memory_planning_pass = MemoryPlanningPass(`
`71`		`- memory_planning_algo="greedy",`
`72`	`71`	`alloc_graph_input=False,`
`73`	`72`	`alloc_graph_output=False,`
`74`	`73`	`)`