We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a196ca8 commit 02b2179Copy full SHA for 02b2179
export.py
@@ -48,7 +48,12 @@ def main(args):
48
output_pte_path = args.output_pte_path
49
output_dso_path = args.output_dso_path
50
51
- if output_dso_path and "mps" in builder_args.device:
+ if output_pte_path and builder_args.device != "cpu":
52
+ print(
53
+ f"Warning! ExecuTorch export target is controlled by export recipe, not device setting. Ignoring device={builder_args.device} setting."
54
+ )
55
+ builder_args.device = "cpu"
56
+ elif "mps" in builder_args.device:
57
print("Warning! Device MPS not supported for export. Exporting for device CPU.")
58
builder_args.device = "cpu"
59
0 commit comments