From 4de62ad6cd6b1307d51e726e224bc43a330c209a Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Wed, 15 Sep 2021 17:52:44 +0800 Subject: [PATCH] docs(mge/device): fix docstring of get_cuda_compute_capability GitOrigin-RevId: 4ec04e8b9b77186a3b91f4d2a0c3583539d43bc9 --- imperative/python/megengine/device.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/imperative/python/megengine/device.py b/imperative/python/megengine/device.py index c47af399..dbd54fd2 100644 --- a/imperative/python/megengine/device.py +++ b/imperative/python/megengine/device.py @@ -144,10 +144,13 @@ def get_mem_status_bytes(device: Optional[str] = None): def get_cuda_compute_capability(device: int, device_type=DeviceType.CUDA) -> int: - r""" - Get compute capability of the specified device. + r"""Gets compute capability of the specified device. - It returns a version number, or `SM version`. + Args: + device: device number. + + Returns: + a version number, or `SM version`. """ return _get_cuda_compute_capability(device, device_type)