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 ----------