From 7165cf2a0a1913fcd9b05384ed476f78428dc1a6 Mon Sep 17 00:00:00 2001 From: Philipp van Kempen Date: Thu, 11 Jan 2024 13:12:26 +0100 Subject: [PATCH] fix refresh_model_info feature for tvm backends --- mlonmcu/flow/tvm/backend/backend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mlonmcu/flow/tvm/backend/backend.py b/mlonmcu/flow/tvm/backend/backend.py index d881a1957..11fd85a61 100644 --- a/mlonmcu/flow/tvm/backend/backend.py +++ b/mlonmcu/flow/tvm/backend/backend.py @@ -457,7 +457,7 @@ def generate(self) -> Tuple[dict, dict]: artifacts = [] assert self.model is not None dump = self.dump - if self.generate_wrapper and not self.model_info and "relay" not in dump: + if self.refresh_model_info or (self.generate_wrapper and not self.model_info) and "relay" not in dump: dump.append("relay") with tempfile.TemporaryDirectory() as temp_dir: out_path = Path(temp_dir) / f"{self.prefix}.tar"