From f35f7f1245f042bd851ea413ee958a82c15701b6 Mon Sep 17 00:00:00 2001 From: b8zhong Date: Sat, 15 Nov 2025 19:53:50 -0800 Subject: [PATCH] [Piecewise CUDA Graph] Support W4A8 (#13179) --- python/sglang/srt/layers/quantization/fp8_kernel.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/python/sglang/srt/layers/quantization/fp8_kernel.py b/python/sglang/srt/layers/quantization/fp8_kernel.py index 2be8a8cf5..4a3d1093d 100644 --- a/python/sglang/srt/layers/quantization/fp8_kernel.py +++ b/python/sglang/srt/layers/quantization/fp8_kernel.py @@ -45,6 +45,10 @@ _use_aiter = get_bool_env_var("SGLANG_USE_AITER") and _is_hip if _is_cuda: from sgl_kernel import sgl_per_tensor_quant_fp8, sgl_per_token_quant_fp8 + @torch.library.register_fake("sgl_kernel::sgl_per_tensor_quant_fp8") + def _sgl_per_tensor_quant_fp8(input, output_q, output_s, is_static): + return + # Temporary try: from sgl_kernel import sgl_per_token_group_quant_8bit