It's gpu not cpu. (apache#4832)

neo-ai · Mar 2, 2020 · 14e6b0b · 14e6b0b
1 parent 58ea0ad
commit 14e6b0b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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
     ----------