add device name to context attribute (#1090)

* add device name to context attribute

* update for other backends

include/tvm/runtime/device_api.h

@@ -21,6 +21,7 @@ enum DeviceAttrKind : int {
   kWarpSize = 2,
   kMaxSharedMemoryPerBlock = 3,
   kComputeVersion = 4,
+  kDeviceName = 5
 };
 
 /*! \brief Number of bytes each allocation must align to */

python/tvm/_ffi/runtime_ctypes.py

@@ -142,7 +142,7 @@ class TVMContext(ctypes.Structure):
 
     @property
     def max_shared_memory_per_block(self):
-        """Total amount of shared memory per block in bytes"""
+        """Total amount of shared memory per block in bytes."""
         return _api_internal._GetDeviceAttr(
             self.device_type, self.device_id, 3)
 
@@ -160,6 +160,12 @@ class TVMContext(ctypes.Structure):
         return _api_internal._GetDeviceAttr(
             self.device_type, self.device_id, 4)
 
+    @property
+    def device_name(self):
+        """Return the string name of device."""
+        return _api_internal._GetDeviceAttr(
+            self.device_type, self.device_id, 5)
+
     def sync(self):
         """Synchronize until jobs finished at the context."""
         check_call(_LIB.TVMSynchronize(self.device_type, self.device_id, None))

src/runtime/cuda/cuda_device_api.cc

@@ -30,7 +30,7 @@ class CUDADeviceAPI final : public DeviceAPI {
                 &value, cudaDevAttrMaxThreadsPerBlock, ctx.device_id)
             == cudaSuccess);
         break;
-    case kMaxThreadsPerBlock: {
+      case kMaxThreadsPerBlock: {
         CUDA_CALL(cudaDeviceGetAttribute(
             &value, cudaDevAttrMaxThreadsPerBlock, ctx.device_id));
         break;
@@ -56,6 +56,12 @@ class CUDADeviceAPI final : public DeviceAPI {
         *rv = os.str();
         return;
       }
+      case kDeviceName: {
+        cudaDeviceProp props;
+        CUDA_CALL(cudaGetDeviceProperties(&props, ctx.device_id));
+        *rv = std::string(props.name);
+        return;
+      }
     }
     *rv = value;
   }

src/runtime/metal/metal_device_api.mm

@@ -41,6 +41,7 @@ void MetalWorkspace::GetAttr(
     }
     case kMaxSharedMemoryPerBlock: return;
     case kComputeVersion: return;
+    case kDeviceName: return;
     case kExist: break;
   }
 }

src/runtime/opencl/opencl_device_api.cc

@@ -54,6 +54,14 @@ void OpenCLWorkspace::GetAttr(
       break;
     }
     case kComputeVersion: return;
+    case kDeviceName: {
+      char value[128] = {0};
+      OPENCL_CALL(clGetDeviceInfo(
+          devices[index], CL_DEVICE_NAME,
+          sizeof(value) - 1, value, nullptr));
+      *rv = std::string(value);
+      break;
+    }
     case kExist: break;
   }
 }

src/runtime/opengl/opengl_device_api.cc

@@ -93,9 +93,11 @@ void OpenGLWorkspace::GetAttr(
       *rv = 1;
       break;
     }
+    case kMaxSharedMemoryPerBlock: return;
     case kComputeVersion: {
       break;
     }
+    case kDeviceName: return;
   }
 }
 

src/runtime/rocm/rocm_device_api.cc

@@ -51,6 +51,7 @@ class ROCMDeviceAPI final : public DeviceAPI {
         *rv = prop.gcnArch;
         return;
       }
+      case kDeviceName: return;
     }
     *rv = value;
   }

src/runtime/vulkan/vulkan_device_api.cc

@@ -73,6 +73,7 @@ void VulkanWorkspace::GetAttr(
       *rv = os.str();
       break;
     }
+    case kDeviceName: return;
     case kExist: break;
   }
 }