diff --git a/vta/tutorials/frontend/deploy_classification.py b/vta/tutorials/frontend/deploy_classification.py index b72301d60c0c..aae994042e5e 100644 --- a/vta/tutorials/frontend/deploy_classification.py +++ b/vta/tutorials/frontend/deploy_classification.py @@ -63,6 +63,8 @@ # Make sure that TVM was compiled with RPC=1 assert tvm.runtime.enabled("rpc") +# Make sure vta is available +assert tvm.get_global_func("tvm.contrib.vta.init", True) is not None ###################################################################### # Define the platform and model targets