From 52959e44a83c6edb132785395fcbfc628c076998 Mon Sep 17 00:00:00 2001 From: Devashish Lal Date: Fri, 13 Mar 2026 17:05:21 -0700 Subject: [PATCH] fix: add missing re-exports for rmsnorm quant and fused_add_rmsnorm quant Signed-off-by: Devashish Lal --- flashinfer/__init__.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/flashinfer/__init__.py b/flashinfer/__init__.py index 62cabd0a6e..9ea91d5508 100644 --- a/flashinfer/__init__.py +++ b/flashinfer/__init__.py @@ -107,10 +107,12 @@ from .gemm import tgv_gemm_sm100 as tgv_gemm_sm100 from .mla import BatchMLAPagedAttentionWrapper as BatchMLAPagedAttentionWrapper from .norm import fused_add_rmsnorm as fused_add_rmsnorm +from .norm import fused_add_rmsnorm_quant as fused_add_rmsnorm_quant from .norm import layernorm as layernorm from .norm import gemma_fused_add_rmsnorm as gemma_fused_add_rmsnorm from .norm import gemma_rmsnorm as gemma_rmsnorm from .norm import rmsnorm as rmsnorm +from .norm import rmsnorm_quant as rmsnorm_quant try: from .norm import rmsnorm_fp4quant as rmsnorm_fp4quant