From 0c0c8776fc7ba920b27e3cb128eb59f4be86e911 Mon Sep 17 00:00:00 2001 From: Gabriel Wu Date: Fri, 14 Jul 2023 00:08:03 +0800 Subject: [PATCH] Fix typo in docs/arch/index.rst --- docs/arch/index.rst | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/arch/index.rst b/docs/arch/index.rst index 3957c6071622..b84afeea2818 100644 --- a/docs/arch/index.rst +++ b/docs/arch/index.rst @@ -152,8 +152,8 @@ The main goal of TVM's runtime is to provide a minimal API for loading and execu mod: tvm.runtime.Module = tvm.runtime.load_module("compiled_artifact.so") arr: tvm.runtime.NDArray = tvm.nd.array([1, 2, 3], device=tvm.cuda(0)) fun: tvm.runtime.PackedFunc = mod["addone"] - fun(a) - print(a.numpy()) + fun(arr) + print(arr.numpy()) :py:class:`tvm.runtime.Module` encapsulates the result of compilation. A runtime.Module contains a GetFunction method to obtain PackedFuncs by name.