MCPcopy Create free account
hub / github.com/KnowingNothing/MatmulTutorial / get_log_swizzle_size

Method get_log_swizzle_size

cutlass.py/tile_scheduler.py:183–192  ·  view source on GitHub ↗
(problem_ctas_m, problem_ctas_n, max_swizzle_size)

Source from the content-addressed store, hash-verified

181
182 @staticmethod
183 def get_log_swizzle_size(problem_ctas_m, problem_ctas_n, max_swizzle_size):
184 min_cta_dim = min(problem_ctas_m, problem_ctas_n)
185 if max_swizzle_size >= 8 and min_cta_dim >= 6:
186 return 3
187 elif max_swizzle_size >= 4 and min_cta_dim >= 3:
188 return 2
189 elif max_swizzle_size >= 2 and min_cta_dim >= 2:
190 return 1
191 else:
192 return 0
193
194 @staticmethod
195 def get_rasterization_order(tiles_m, tiles_n, raster_order_option):

Callers 2

_initializeMethod · 0.95
_get_grid_shapeMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected