fix refresh_model_info feature for tvm backends

tum-ei-eda · Jan 11, 2024 · 7165cf2 · 7165cf2
1 parent 9c34132
commit 7165cf2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"