e96973742c
Co-authored-by: wunhuang <wunhuang@amd.com> Co-authored-by: HAI <hixiao@gmail.com> Co-authored-by: Hubert Lu <55214931+hubertlu-tw@users.noreply.github.com>
14 lines
327 B
Python
14 lines
327 B
Python
from aiter.ops.triton.batched_gemm_afp4wfp4_pre_quant import (
|
|
batched_gemm_afp4wfp4_pre_quant,
|
|
)
|
|
from aiter.ops.triton.fused_mxfp4_quant import (
|
|
fused_flatten_mxfp4_quant,
|
|
fused_rms_mxfp4_quant,
|
|
)
|
|
|
|
__all__ = [
|
|
"fused_rms_mxfp4_quant",
|
|
"fused_flatten_mxfp4_quant",
|
|
"batched_gemm_afp4wfp4_pre_quant",
|
|
]
|