From 485248b6d584b83adb5ab5443adc3b558c1a85a2 Mon Sep 17 00:00:00 2001 From: winskuo-quic Date: Tue, 24 Sep 2024 11:50:52 -0700 Subject: [PATCH] Qualcomm AI Engine Direct - Fix aihub path failing due to memory planning pass (#5482) Summary: Update MemoryPlanningPass to use default greedy algo. Pull Request resolved: https://github.com/pytorch/executorch/pull/5482 Reviewed By: kirklandsign Differential Revision: D63270305 Pulled By: cccclai fbshipit-source-id: 7fb03365e3cadb53e4879feb7f105aadc9e9497c (cherry picked from commit 90dcea5488a50dffeaa5b30c3d0bfb71c634dd80) --- examples/qualcomm/qaihub_scripts/utils/utils.py | 1 - 1 file changed, 1 deletion(-) diff --git a/examples/qualcomm/qaihub_scripts/utils/utils.py b/examples/qualcomm/qaihub_scripts/utils/utils.py index 67d519a688e..ad55d7fd10b 100644 --- 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, )