From 91ea9bce8bffda702698cbceb8d66b529cbb8812 Mon Sep 17 00:00:00 2001 From: Matthew LeMay Date: Thu, 20 Aug 2020 20:45:17 -0400 Subject: [PATCH] Changed TVMCTVMContext to TVMContext (#6306) --- python/tvm/contrib/graph_runtime.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/contrib/graph_runtime.py b/python/tvm/contrib/graph_runtime.py index 22d0a8b8bc1d..17c331d49955 100644 --- a/python/tvm/contrib/graph_runtime.py +++ b/python/tvm/contrib/graph_runtime.py @@ -81,7 +81,7 @@ def get_device_ctx(libmod, ctx): ctx = [ctx] elif not isinstance(ctx, (list, tuple)): raise ValueError("ctx has to be the type of TVMContext or a list of " - "TVMCTVMContext") + "TVMContext") for cur_ctx in ctx: if not isinstance(cur_ctx, TVMContext): raise ValueError("ctx has to be the type of TVMContext or a list "