Skip to content

Commit c6e3bba

Browse files
authored
[Feature] Integrate new deepgemm (vllm-project#19820)
Signed-off-by: yewentao256 <zhyanwentao@126.com>
1 parent 91f7d9d commit c6e3bba

File tree

8 files changed

+234
-268
lines changed

8 files changed

+234
-268
lines changed

benchmarks/kernels/benchmark_moe.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -86,6 +86,9 @@ def benchmark_config(
8686
(num_experts, 2 * shard_intermediate_size), dtype=torch.float32
8787
)
8888
w2_scale = torch.randn((hidden_size, num_experts), dtype=torch.float32)
89+
if use_deep_gemm:
90+
# we use the default block shape for deepgemm
91+
block_quant_shape = [128, 128]
8992
if use_fp8_w8a8:
9093
if block_quant_shape:
9194
block_n, block_k = block_quant_shape[0], block_quant_shape[1]

0 commit comments

Comments
 (0)