| try: | |
| from .._ops import ops | |
| def get_tma_aligned_size(x, element_size): | |
| return ops.get_tma_aligned_size(x, element_size) | |
| def get_mn_major_tma_aligned_tensor(sf): | |
| return ops.get_mn_major_tma_aligned_tensor(sf) | |
| def get_mn_major_tma_aligned_packed_ue8m0_tensor(sf): | |
| return ops.get_mn_major_tma_aligned_packed_ue8m0_tensor(sf) | |
| def get_k_grouped_mn_major_tma_aligned_packed_ue8m0_tensor(sf, ks_tensor, ks): | |
| return ops.get_k_grouped_mn_major_tma_aligned_packed_ue8m0_tensor( | |
| sf, ks_tensor, ks) | |
| except ImportError: | |
| pass | |
| from .._ops import ops as _ops | |
| def get_mk_alignment_for_contiguous_layout(): | |
| return _ops.get_mk_alignment_for_contiguous_layout() | |
| get_m_alignment_for_contiguous_layout = get_mk_alignment_for_contiguous_layout | |
| get_k_alignment_for_contiguous_layout = get_mk_alignment_for_contiguous_layout | |