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

Method api_version

python/tvm/runtime/device.py:186–202  ·  view source on GitHub ↗

Returns version number of the SDK used to compile TVM. For example, CUDA_VERSION for CUDA or VK_HEADER_VERSION for Vulkan. Returns device value for CUDA, ROCm, OpenCL, and Vulkan. Returns remote device value for RPC devices. Returns None for all other devic

(self)

Source from the content-addressed store, hash-verified

184
185 @property
186 def api_version(self):
187 """Returns version number of the SDK used to compile TVM.
188
189 For example, CUDA_VERSION for CUDA or VK_HEADER_VERSION for
190 Vulkan.
191
192 Returns device value for CUDA, ROCm, OpenCL, and Vulkan.
193 Returns remote device value for RPC devices. Returns None for
194 all other devices.
195
196 Returns
197 -------
198 version : int or None
199 The version of the SDK
200
201 """
202 return self._GetDeviceAttr(self.dlpack_device_type(), self.index, 11)
203
204 @property
205 def driver_version(self):

Callers

nothing calls this directly

Calls 1

_GetDeviceAttrMethod · 0.95

Tested by

no test coverage detected