diff --git a/python/tvm/contrib/hexagon/session.py b/python/tvm/contrib/hexagon/session.py index 466103f6e2c9..506f1d968d70 100644 --- a/python/tvm/contrib/hexagon/session.py +++ b/python/tvm/contrib/hexagon/session.py @@ -248,8 +248,9 @@ def get_aot_executor( GraphModule : Runtime graph module that can be used to execute the graph. """ - aot_mod = self.load_module(module_file) - return tvm.runtime.executor.AotModule(aot_mod["default"](self.device)) + # Temporary workaround for https://github.com/apache/tvm/issues/13741 + self.aot_mod = self.load_module(module_file) + return tvm.runtime.executor.AotModule(self.aot_mod["default"](self.device)) def get_graph_debug_executor( self,