[1/n jit_kernel restruct] unify cache usage and clean up naming in ngram_embedding (#20244)

2026-03-10 15:53:43 +08:00
parent 7cf0551014
commit c812504b92
4 changed files with 14 additions and 15 deletions
--- a/python/sglang/jit_kernel/add_constant.py
+++ b/python/sglang/jit_kernel/add_constant.py
@@ -12,7 +12,7 @@ if TYPE_CHECKING:

@cache_once
 def _jit_add_constant_module(constant: int) -> Module:
-    args = make_cpp_args(constant)  # pass all the template argument
+    args = make_cpp_args(constant)
    return load_jit(
        "add_constant",
        *args,