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

Method max_shared_memory_per_block

python/tvm/runtime/device.py:87–100  ·  view source on GitHub ↗

Total amount of shared memory per block in bytes. Returns device value for CUDA, ROCm, OpenCL, and Vulkan. Returns remote device value for RPC devices. Returns None for all other devices. Returns ------- max_shared_memory_per_block : int or None

(self)

Source from the content-addressed store, hash-verified

85
86 @property
87 def max_shared_memory_per_block(self):
88 """Total amount of shared memory per block in bytes.
89
90 Returns device value for CUDA, ROCm, OpenCL, and Vulkan.
91 Returns remote device value for RPC devices. Returns None for
92 all other devices.
93
94 Returns
95 -------
96 max_shared_memory_per_block : int or None
97 Total amount of shared memory per block in bytes
98
99 """
100 return self._GetDeviceAttr(self.dlpack_device_type(), self.index, 3)
101
102 @property
103 def compute_version(self):

Callers

nothing calls this directly

Calls 1

_GetDeviceAttrMethod · 0.95

Tested by

no test coverage detected