From 79bb7d59d65fac3b489430373d6a40ab42c7dc19 Mon Sep 17 00:00:00 2001 From: "Yanan Cao (PyTorch)" Date: Thu, 19 Dec 2024 10:51:39 -0800 Subject: [PATCH] executorch/backends/cadence/aot (#7378) Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/7378 Reviewed By: avikchaudhuri, ydwu4 Differential Revision: D67381398 --- backends/cadence/aot/compiler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backends/cadence/aot/compiler.py b/backends/cadence/aot/compiler.py index 6b3a023181c..df1f42601b0 100644 --- a/backends/cadence/aot/compiler.py +++ b/backends/cadence/aot/compiler.py @@ -176,7 +176,7 @@ def export_program( torch._C._set_mkldnn_enabled(False) # else: capture the model and return it. - expo_program = export(model, inputs) + expo_program = export(model, inputs, strict=True) if dump_graphs: logging.info("Exported graph:")