From fb38e977ccd9b8d693b603bcc1768f83fef648bf Mon Sep 17 00:00:00 2001 From: Tristan Konolige Date: Thu, 6 May 2021 13:20:20 -0700 Subject: [PATCH] [FIX,DOCS] Prevent segfault when building docs when VTA is not available --- vta/tutorials/frontend/deploy_classification.py | 2 ++ 1 file changed, 2 insertions(+) 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