From 06c9b5104a99a296dca2fbddd87b20b37abe2d52 Mon Sep 17 00:00:00 2001 From: Mehrdad Hessar Date: Tue, 14 Jun 2022 14:56:56 -0700 Subject: [PATCH] revert to tvm.runtime.Module --- python/tvm/micro/model_library_format.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/tvm/micro/model_library_format.py b/python/tvm/micro/model_library_format.py index d13947527e81f..e220fa1ca5430 100644 --- a/python/tvm/micro/model_library_format.py +++ b/python/tvm/micro/model_library_format.py @@ -71,7 +71,7 @@ def generate_c_interface_header( def _populate_codegen_dir( mods: typing.Union[ typing.List[executor_factory.ExecutorFactoryModule], - typing.List[build_module.OperatorModule], + typing.List[tvm.runtime.Module], ], codegen_dir: str, ): @@ -92,7 +92,7 @@ def _populate_codegen_dir( for mod in mods: if isinstance(mod, executor_factory.ExecutorFactoryModule): lib = mod.lib - elif isinstance(mod, build_module.OperatorModule): + elif isinstance(mod, tvm.runtime.Module): lib = mod else: raise RuntimeError(f"Not supported module type: {type(mod)}")