From 18ee95b655f32977dac23f4786786eb4daac61d2 Mon Sep 17 00:00:00 2001 From: Luke Hutton Date: Wed, 1 Dec 2021 16:50:52 +0000 Subject: [PATCH] fix codegen tests after migration to target hooks Change-Id: Idcab5f71f3c5a971831b96fe8f9a5c8c73b267da --- tests/python/contrib/test_ethosu/test_codegen.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/tests/python/contrib/test_ethosu/test_codegen.py b/tests/python/contrib/test_ethosu/test_codegen.py index d3a965ac22f7..afd635d96cc6 100644 --- a/tests/python/contrib/test_ethosu/test_codegen.py +++ b/tests/python/contrib/test_ethosu/test_codegen.py @@ -593,15 +593,12 @@ def create_mod_from_relay(): ) # Assumes only two runtime.Modules are created -- i.e. single offload module - imported_modules = compiled_models[0].executor_factory.lib.imported_modules - assert len(imported_modules) == 2 - ethosu_module = imported_modules[0] + ethosu_module = compiled_models[0].executor_factory.lib.imported_modules[0].imported_modules[0] # Verify generated C source - get_cs = tvm._ffi.get_global_func("runtime.module.ethos-u.getcs") - cmms = get_cs(ethosu_module) - cmms = bytes.fromhex(cmms) - + get_artifacts = tvm._ffi.get_global_func("runtime.module.ethos-u.get_artifacts") + compilation_artifacts = get_artifacts(ethosu_module) + cmms = bytes.fromhex(compilation_artifacts[0].command_stream) infra.print_payload(cmms) infra.verify_source(compiled_models, accel_type)