distinct name for low latency gemm module

hypdeb · hypdeb · commit 960a8cd89773 · 2025-10-08T19:09:47.000Z
diff --git a/flashinfer/jit/gemm/core.py b/flashinfer/jit/gemm/core.py
@@ -513,7 +513,7 @@ def gen_trtllm_low_latency_gemm_module() -> JitSpec:
     # make sure "flashinferMetaInfo.h" is downloaded or cached
     assert metainfo, f"{header_name}.h not found"
     return gen_jit_spec(
-        "trtllm_gemm",
+        "trtllm_low_latency_gemm",
         [
             jit_env.FLASHINFER_CSRC_DIR / "trtllm_low_latency_gemm_runner.cu",
         ],

Original file line number	Diff line number	Diff line change
`@@ -513,7 +513,7 @@ def gen_trtllm_low_latency_gemm_module() -> JitSpec:`
`513`	`513`	`# make sure "flashinferMetaInfo.h" is downloaded or cached`
`514`	`514`	`assert metainfo, f"{header_name}.h not found"`
`515`	`515`	`return gen_jit_spec(`
`516`		`- "trtllm_gemm",`
	`516`	`+ "trtllm_low_latency_gemm",`
`517`	`517`	`[`
`518`	`518`	`jit_env.FLASHINFER_CSRC_DIR / "trtllm_low_latency_gemm_runner.cu",`
`519`	`519`	`],`