Fix pyre issue for diff train

Differential Revision: D67315650 Pull Request resolved: pytorch#7336
kirklandsign · Dec 17, 2024 · 9d1a310 · 9d1a310
1 parent 0fb2ccf
commit 9d1a310
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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,