From a036dfc90b25a793ac3019717279f65d3518dbfc Mon Sep 17 00:00:00 2001 From: "Yanan Cao (PyTorch)" Date: Thu, 19 Dec 2024 10:57:26 -0800 Subject: [PATCH] executorch/profiler/test (#7389) Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/7389 Reviewed By: avikchaudhuri, ydwu4 Differential Revision: D67384075 --- profiler/test/test_profiler_e2e.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/profiler/test/test_profiler_e2e.py b/profiler/test/test_profiler_e2e.py index f5df82176ee..b38644c210c 100644 --- a/profiler/test/test_profiler_e2e.py +++ b/profiler/test/test_profiler_e2e.py @@ -52,7 +52,9 @@ def setUpClass(cls) -> None: # The serialized program file. This must live longer than cls.module, # because the C++ pybindings will have a pointer to it. But none of the # tests should need to touch it. - cls.__buffer: bytes = to_edge(export(model, inputs)).to_executorch().buffer + cls.__buffer: bytes = ( + to_edge(export(model, inputs, strict=True)).to_executorch().buffer + ) cls.module = _load_for_executorch_from_buffer(cls.__buffer)