File size: 879 Bytes
c67ae40 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 | 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
|