From 62543d49f4da4a46221a61e417401fe774720270 Mon Sep 17 00:00:00 2001 From: Just do it <1467221872@qq.com> Date: Fri, 7 Feb 2020 01:25:06 +0800 Subject: [PATCH] It's gpu not cpu. (#4832) --- python/tvm/runtime/ndarray.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/runtime/ndarray.py b/python/tvm/runtime/ndarray.py index cce73e09fba7..7fbad7c943a5 100644 --- a/python/tvm/runtime/ndarray.py +++ b/python/tvm/runtime/ndarray.py @@ -315,7 +315,7 @@ def cpu(dev_id=0): def gpu(dev_id=0): - """Construct a CPU device + """Construct a GPU device Parameters ----------