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

Method multi_processor_count

python/tvm/runtime/device.py:154–167  ·  view source on GitHub ↗

Return the number of compute units in the device. Returns device value for CUDA, ROCm, and OpenCL. Returns remote device value for RPC devices. Returns None for all other devices. Returns ------- multi_processor_count : int or None Thee

(self)

Source from the content-addressed store, hash-verified

152
153 @property
154 def multi_processor_count(self):
155 """Return the number of compute units in the device.
156
157 Returns device value for CUDA, ROCm, and OpenCL. Returns
158 remote device value for RPC devices. Returns None for all
159 other devices.
160
161 Returns
162 -------
163 multi_processor_count : int or None
164 Thee number of compute units in the device
165
166 """
167 return self._GetDeviceAttr(self.dlpack_device_type(), self.index, 7)
168
169 @property
170 def max_thread_dimensions(self):

Callers

nothing calls this directly

Calls 1

_GetDeviceAttrMethod · 0.95

Tested by

no test coverage detected