Revert "[jit kernel] Support per_token_group_quant_8bit jit kernel" (#19131)

2026-02-21 15:54:24 -08:00
parent 1f2da824dd
commit 4653939cda
6 changed files with 14 additions and 811 deletions
--- a/python/sglang/jit_kernel/utils.py
+++ b/python/sglang/jit_kernel/utils.py
@@ -10,6 +10,7 @@ import torch
 if TYPE_CHECKING:
    from tvm_ffi import Module

+
 F = TypeVar("F", bound=Callable[..., Any])


@@ -72,9 +73,7 @@ class CPPArgList(list[str]):
 CPP_DTYPE_MAP = {
    torch.float: "fp32_t",
    torch.float16: "fp16_t",
-    torch.float8_e4m3fn: "fp8_e4m3_t",
    torch.bfloat16: "bf16_t",
-    torch.int8: "int8_t",
 }