Skip to content

GitLab

"projects/PointSup/point_sup/mask_head.py" did not exist on "5b3792fc3ef9ab6a6f8f30634ab2e52fb0941af3"
Unverified Commit e2521926 authored by Ke Bao's avatar Ke Bao Committed by GitHub
Browse files

Fix deepgemm masked grouped gemm jit compile (#8679)

parent 07e46eca
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
python/sglang/srt/layers/quantization/deep_gemm_wrapper/compile_utils.py
View file @ e2521926
...@@ -148,7 +148,7 @@ def _compile_grouped_gemm_nt_f8f8bf16_masked_one( ...@@ -148,7 +148,7 @@ def _compile_grouped_gemm_nt_f8f8bf16_masked_one(
"NUM_MATH_THREADS_PER_GROUP": num_math_threads_per_group, "NUM_MATH_THREADS_PER_GROUP": num_math_threads_per_group,
"N": n, "N": n,
"K": k, "K": k,
"NUM_GROUPS": 1, "NUM_GROUPS": num_groups,
"BLOCK_M": block_m, "BLOCK_M": block_m,
"BLOCK_N": block_n, "BLOCK_N": block_n,
"BLOCK_K": block_k, "BLOCK_K": block_k,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment