Warn if wrong type is given for Llama export for XNNPACK

Followup to #7775
pytorch · Feb 4, 2025 · 6d60669 · 6d60669
1 parent a3455d9
commit 6d60669
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/examples/models/llama/export_llama_lib.py b/examples/models/llama/export_llama_lib.py
@@ -674,6 +674,12 @@ def _validate_args(args):
                 "If you need this feature, please file an issue."
             )
 
+    if args.xnnpack:
+        if args.dtype_override not in ["fp32", "fp16"]:
+            raise ValueError(
+                f"XNNPACK supports either fp32 or fp16 dtypes only for now. Given {args.dtype_override}."
+            )
+
 
 def _export_llama(args) -> LLMEdgeManager:  # noqa: C901
     _validate_args(args)