From 06c911e01d5e968983230e47f88b4a280cbe32c3 Mon Sep 17 00:00:00 2001 From: Eric Lunderberg Date: Fri, 13 Aug 2021 08:31:24 -0500 Subject: [PATCH] Renamed graph_runtime references to graph_executor. Following TVM's rename in https://github.com/apache/tvm/pull/7653, update to avoid errors when tvm.contrib.graph_runtime is removed entirely. --- apps/deploy/python_deploy.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/apps/deploy/python_deploy.py b/apps/deploy/python_deploy.py index 25020467..4f447bbc 100644 --- a/apps/deploy/python_deploy.py +++ b/apps/deploy/python_deploy.py @@ -26,7 +26,7 @@ import numpy as np import tvm -from tvm.contrib import graph_runtime, download +from tvm.contrib import graph_executor, download CTX = tvm.ext_dev(0) @@ -50,7 +50,7 @@ def load_model(): lib = tvm.runtime.load_module("./build/model/lib.so") - model = graph_runtime.create(graph, lib, CTX) + model = graph_executor.create(graph, lib, CTX) with open("./build/model/params.params", "rb") as paramfile: param_bytes = paramfile.read()