diff --git a/src/relay/backend/build_module.cc b/src/relay/backend/build_module.cc index 78978e192f0e..4dd12ad1d106 100644 --- a/src/relay/backend/build_module.cc +++ b/src/relay/backend/build_module.cc @@ -501,7 +501,7 @@ class RelayBuildModule : public runtime::ModuleNode { ret_.mod = tvm::codegen::CSourceModuleCreate(";", "", Array{}); } } else { - ret_.mod = tvm::build(lowered_funcs, target_host_); + ret_.mod = tvm::build(lowered_funcs, target_host); } auto ext_mods = executor_codegen_->GetExternalModules(); diff --git a/tests/python/driver/tvmc/test_compiler.py b/tests/python/driver/tvmc/test_compiler.py index 7dd65c507df3..2ef84d7f1a6f 100644 --- a/tests/python/driver/tvmc/test_compiler.py +++ b/tests/python/driver/tvmc/test_compiler.py @@ -416,7 +416,7 @@ def test_compile_tflite_module_with_external_codegen_cmsisnn( for name in mlf_package.getnames() if re.match(r"\./codegen/host/src/\D+\d+\.c", name) ] - assert len(c_source_files) == 4 + assert len(c_source_files) == 3 @pytest.mark.skipif(