From ff2e33b8d422073b119a95069839493dc5d0331f Mon Sep 17 00:00:00 2001 From: Martin Yuan Date: Mon, 16 Dec 2024 21:37:36 -0800 Subject: [PATCH] Fix pyre issue for diff train (#7336) Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/7336 There's pyre issue introduced in D67262774. Fix it to unblock the diff train to land. Reviewed By: JacobSzwejbka Differential Revision: D67315650 --- extension/llm/export/partitioner_lib.py | 1 + 1 file changed, 1 insertion(+) diff --git a/extension/llm/export/partitioner_lib.py b/extension/llm/export/partitioner_lib.py index ff8d8f93284..19c89596765 100644 --- a/extension/llm/export/partitioner_lib.py +++ b/extension/llm/export/partitioner_lib.py @@ -126,6 +126,7 @@ def _validate_ios_version() -> None: # On iPhone 15 Pro, CPU decode model is over 8x faster than GPU for stories110M, # so default to CPU_ONLY coreml_compute_units = "cpu_only" + # pyre-ignore coreml_compute_units = { "cpu_only": ct.ComputeUnit.CPU_ONLY, "cpu_and_ne": ct.ComputeUnit.CPU_AND_NE,