Running Mobilenet test mix with vanilla accelerator example results in: schedule_injective not registered for ‘vanilla_accelerator’ exception from topi generic injective module. i tried to register the injective, but it didn’t seem to help.
Code:
def test_mobilenet():
“”“Full network test with Mobilenet”""
use_unpacked_api = True
interface_api = “c”
runner = AOT_DEFAULT_RUNNER
mod, params = testing.mobilenet.get_workload(batch_size=1)
uma_backend = VanillaAcceleratorBackend()
uma_backend.register()
target = tvm.target.Target(“vanilla_accelerator”, host=tvm.target.Target(“c”))
target_c = tvm.target.Target(“c”)
data_shape = [int(x) for x in mod[“main”].checked_type.arg_types[0].shape]
data = np.random.uniform(size=data_shape).astype(“float32”)
input_list = {“data”: data}
output_list = generate_ref_data(mod, input_list, params)
mod = uma_backend.partition(mod)
pass_config = {“tir.usmp.enable”: True}
runner = AOTRunner(
makefile=runner.makefile,
prologue=runner.prologue,
epilogue=runner.epilogue,
includes=runner.includes,
parameters=runner.parameters,
pass_config=pass_config,
)
export_directory = tvm.contrib.utils.tempdir(keep_for_debug=True).path
print(f"Generated files are in {export_directory}")
compile_and_run(
AOTModel(module=mod, inputs=input_list, outputs=output_list),
runner,
interface_api="c",
use_unpacked_api=True,
target=target,
test_dir=str(export_directory),
)