We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c0946e1 commit 4b183daCopy full SHA for 4b183da
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
152
mod: tvm.runtime.Module = tvm.runtime.load_module("compiled_artifact.so")
153
arr: tvm.runtime.NDArray = tvm.nd.array([1, 2, 3], device=tvm.cuda(0))
154
fun: tvm.runtime.PackedFunc = mod["addone"]
155
- fun(a)
156
- print(a.numpy())
+ fun(arr)
+ print(arr.numpy())
157
158
159
:py:class:`tvm.runtime.Module` encapsulates the result of compilation. A runtime.Module contains a GetFunction method to obtain PackedFuncs by name.
0 commit comments