grid: grid config (how many blocks in grid) cluster: cluster config (how many blocks in cluster) block: block config (how many threads in block)
(self, grid: Dim3, cluster: Dim3, block: Dim3)
| 157 | |
| 158 | class GPUKernel: |
| 159 | def __init__(self, grid: Dim3, cluster: Dim3, block: Dim3) -> None: |
| 160 | """ |
| 161 | grid: grid config (how many blocks in grid) |
| 162 | cluster: cluster config (how many blocks in cluster) |
| 163 | block: block config (how many threads in block) |
| 164 | """ |
| 165 | assert type(grid) == Dim3 |
| 166 | assert type(cluster) == Dim3 |
| 167 | assert type(block) == Dim3 |
| 168 | self.grid = grid |
| 169 | self.cluster = cluster |
| 170 | self.block = block |
| 171 | |
| 172 | def blockRange(self): |
| 173 | return Range3( |
nothing calls this directly
no outgoing calls
no test coverage detected