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

Method max_thread_dimensions

python/tvm/runtime/device.py:170–183  ·  view source on GitHub ↗

Return the maximum size of each thread axis Returns device value for CUDA, ROCm, OpenCL, and Vulkan. Returns remote device value for RPC devices. Returns None for all other devices. Returns ------- dims: List of int, or None The maximum

(self)

Source from the content-addressed store, hash-verified

168
169 @property
170 def max_thread_dimensions(self):
171 """Return the maximum size of each thread axis
172
173 Returns device value for CUDA, ROCm, OpenCL, and Vulkan.
174 Returns remote device value for RPC devices. Returns None for
175 all other devices.
176
177 Returns
178 -------
179 dims: List of int, or None
180 The maximum length of threadIdx.x, threadIdx.y, threadIdx.z
181
182 """
183 return json.loads(self._GetDeviceAttr(self.dlpack_device_type(), self.index, 8))
184
185 @property
186 def api_version(self):

Callers

nothing calls this directly

Calls 1

_GetDeviceAttrMethod · 0.95

Tested by

no test coverage detected