Function
get_dst_align
(tile: TileDescription, out_layout: LayoutType)
Source from the content-addressed store, hash-verified
| 753 | return flt_align |
| 754 | |
| 755 | def get_dst_align(tile: TileDescription, out_layout: LayoutType) -> int: |
| 756 | nonlocal dst_align |
| 757 | if ( |
| 758 | tile.math_instruction.opcode_class == OpcodeClass.TensorOp |
| 759 | and dst_layout == LayoutType.TensorNC4HW4 |
| 760 | ): |
| 761 | dst_align = 32 |
| 762 | return dst_align |
| 763 | |
| 764 | def filter_epilogue_with_conv_kind( |
| 765 | epilogue: EpilogueFunctor, conv_kind: ConvKind |
Tested by
no test coverage detected