From bb133665a74523c49677090f8263846258f667ad Mon Sep 17 00:00:00 2001 From: Grzegorz Latosinski Date: Fri, 22 Mar 2024 17:28:44 +0100 Subject: [PATCH] [#56762] optimizers: tvm: Made zephyr_template_header optional Signed-off-by: Grzegorz Latosinski --- kenning/optimizers/tvm.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/kenning/optimizers/tvm.py b/kenning/optimizers/tvm.py index 69c24c40..d60e5d3a 100644 --- a/kenning/optimizers/tvm.py +++ b/kenning/optimizers/tvm.py @@ -671,6 +671,9 @@ def generator(): with open(outputpath.with_suffix(".c"), "w") as ops_f: ops_f.write(lib.get_lib().get_source()) + if not self.zephyr_template_header: + return None + # extract TVM functions from source file tvm_funcs = list( set(