Maximum number of threads on each block. Returns device value for CUDA, Metal, ROCm, OpenCL, and Vulkan devices. Returns remote device value for RPC devices. Returns None for all other devices. Returns ------- max_threads_per_block : int or None
(self)
| 52 | |
| 53 | @property |
| 54 | def max_threads_per_block(self): |
| 55 | """Maximum number of threads on each block. |
| 56 | |
| 57 | Returns device value for CUDA, Metal, ROCm, OpenCL, and Vulkan |
| 58 | devices. Returns remote device value for RPC devices. |
| 59 | Returns None for all other devices. |
| 60 | |
| 61 | Returns |
| 62 | ------- |
| 63 | max_threads_per_block : int or None |
| 64 | The number of threads on each block |
| 65 | |
| 66 | """ |
| 67 | return self._GetDeviceAttr(self.dlpack_device_type(), self.index, 1) |
| 68 | |
| 69 | @property |
| 70 | def warp_size(self): |