Add op paddle.device.cuda.get_device_name and paddle.device.cuda.get_device_capability.

python/paddle/device/cuda/__init__.py

@@ -20,13 +20,8 @@
 from .streams import Event  # noqa: F401
 
 __all__ = [
-    'Stream',
-    'Event',
-    'current_stream',
-    'synchronize',
-    'device_count',
-    'empty_cache',
-    'stream_guard',
+    'Stream', 'Event', 'current_stream', 'synchronize', 'device_count',
+    'empty_cache', 'stream_guard', 'get_device_name', 'get_device_capability'
 ]
 
 
@@ -204,3 +199,63 @@ def stream_guard(stream):
             yield
         finally:
             stream = _set_current_stream(pre_stream)
+
+
+def get_device_name(device=None):
+    '''
+    Return the name of the device which is get from CUDA function `cudaDeviceProp<https://docs.nvidia.com/cuda/cuda-runtime-api/group__CUDART__DEVICE.html#group__CUDART__DEVICE_1g1bf9d625a931d657e08db2b4391170f0>`_..
+
+    Parameters:
+        device(paddle.CUDAPlace()|int, optional): The device or the ID of the device.
+        If device is None, the device is the current device. Default: None.
+
+    Returns:
+        str: the name of the device.
+
+    Examples:
+        .. code-block:: python
+
+            # required: gpu
+
+            import paddle
+
+            paddle.device.cuda.get_device_name()
+
+            paddle.device.cuda.get_device_name(0)
+
+            paddle.device.cuda.get_device_name(paddle.CUDAPlace(0))
+
+    '''
+
+    return get_device_properties(device).name
+
+
+def get_device_capability(device=None):
+    '''
+    Return the major and minor revision numbers defining the device's compute capability which are get from CUDA function `cudaDeviceProp<https://docs.nvidia.com/cuda/cuda-runtime-api/group__CUDART__DEVICE.html#group__CUDART__DEVICE_1g1bf9d625a931d657e08db2b4391170f0>`_..
+
+    Parameters:
+        device(paddle.CUDAPlace()|int, optional): The device or the ID of the device.
+        If device is None, the device is the current device. Default: None.
+
+    Returns:
+        tuple(int,int): the major and minor revision numbers defining the device's compute capability.
+
+    Examples:
+
+        .. code-block:: python
+
+            # required: gpu
+
+            import paddle
+
+            paddle.device.cuda.get_device_capability()
+
+            paddle.device.cuda.get_device_capability(0)
+
+            paddle.device.cuda.get_device_capability(paddle.CUDAPlace(0))
+
+    '''
+
+    return get_device_properties(device).major, get_device_properties(
+        device).minor

python/paddle/fluid/tests/unittests/test_cuda_device_name_capability.py

@@ -0,0 +1,55 @@
+# Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
+# 
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+# 
+#     http://www.apache.org/licenses/LICENSE-2.0
+# 
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import paddle
+import unittest
+
+
+class TestDeviceName(unittest.TestCase):
+    def test_device_name_default(self):
+        if paddle.is_compiled_with_cuda():
+            name = paddle.device.cuda.get_device_name()
+            self.assertIsNotNone(name)
+
+    def test_device_name_int(self):
+        if paddle.is_compiled_with_cuda():
+            name = paddle.device.cuda.get_device_name(0)
+            self.assertIsNotNone(name)
+
+    def test_device_name_CUDAPlace(self):
+        if paddle.is_compiled_with_cuda():
+            name = paddle.device.cuda.get_device_name(paddle.CUDAPlace(0))
+            self.assertIsNotNone(name)
+
+
+class TestDeviceCapability(unittest.TestCase):
+    def test_device_capability_default(self):
+        if paddle.is_compiled_with_cuda():
+            capability = paddle.device.cuda.get_device_capability()
+            self.assertIsNotNone(capability)
+
+    def test_device_capability_int(self):
+        if paddle.is_compiled_with_cuda():
+            capability = paddle.device.cuda.get_device_capability(0)
+            self.assertIsNotNone(capability)
+
+    def test_device_capability_CUDAPlace(self):
+        if paddle.is_compiled_with_cuda():
+            capability = paddle.device.cuda.get_device_capability(
+                paddle.CUDAPlace(0))
+            self.assertIsNotNone(capability)
+
+
+if __name__ == "__main__":
+    unittest.main()