Searched refs:torch_layout (Results 1 – 2 of 2) sorted by relevance
28 def torch_layout_to_ck_layout(torch_layout): argument29 if torch_layout.stride[-1] == 1:31 elif torch_layout.stride[-2] == 1:
558 …def cutlass_layout(torch_layout: ir.Layout) -> "Optional[cutlass_lib.LayoutType]": # type: ignore…573 if torch_layout.stride[-1] == 1:575 elif torch_layout.stride[-2] == 1:596 torch_layout: ir.Layout,600 return CUTLASSGemmTemplate.cutlass_layout(torch_layout) == cutlass_layout603 def set_alignment(torch_layout, op_element) -> bool: argument617 alignment = cutlass_utils.get_max_alignment(torch_layout)