MCPcopy Create free account
hub / github.com/apache/tvm / compute_version

Method compute_version

python/tvm/runtime/device.py:103–119  ·  view source on GitHub ↗

Get compute version number as string. Returns maximum API version (e.g. CUDA/OpenCL/Vulkan) supported by the device. Returns device value for CUDA, ROCm, OpenCL, and Vulkan. Returns remote device value for RPC devices. Returns None for all other devices.

(self)

Source from the content-addressed store, hash-verified

101
102 @property
103 def compute_version(self):
104 """Get compute version number as string.
105
106 Returns maximum API version (e.g. CUDA/OpenCL/Vulkan)
107 supported by the device.
108
109 Returns device value for CUDA, ROCm, OpenCL, and
110 Vulkan. Returns remote device value for RPC devices. Returns
111 None for all other devices.
112
113 Returns
114 -------
115 version : str or None
116 The version string in `major.minor` format.
117
118 """
119 return self._GetDeviceAttr(self.dlpack_device_type(), self.index, 4)
120
121 @property
122 def device_name(self):

Callers

nothing calls this directly

Calls 1

_GetDeviceAttrMethod · 0.95

Tested by

no test coverage detected