(
self,
problem_shape: HyperCube, # m, n, k, l
tile_shape: HyperCube,
cluster_shape: HyperCube,
hw_info: KernelHardwareInfo,
max_swizzle_size: int,
raster_order_option: RasterOrderOptions,
)
| 27 | self.divmod_cluster_blk_major = None |
| 28 | |
| 29 | def initialize( |
| 30 | self, |
| 31 | problem_shape: HyperCube, # m, n, k, l |
| 32 | tile_shape: HyperCube, |
| 33 | cluster_shape: HyperCube, |
| 34 | hw_info: KernelHardwareInfo, |
| 35 | max_swizzle_size: int, |
| 36 | raster_order_option: RasterOrderOptions, |
| 37 | ): |
| 38 | problem_blocks: dim3 = TileParams.get_tiled_cta_shape_mnl( |
| 39 | problem_shape, tile_shape, cluster_shape |
| 40 | ) |
| 41 | self._initialize( |
| 42 | problem_blocks, |
| 43 | cluster_shape, |
| 44 | hw_info, |
| 45 | max_swizzle_size, |
| 46 | raster_order_option, |
| 47 | ) |
| 48 | |
| 49 | def _initialize( |
| 50 | self, |
no test coverage detected