Skip to content

GitLab

Unverified Commit 50f28f65 authored by Xiaoyu Zhang's avatar Xiaoyu Zhang Committed by GitHub
Browse files

fix typo in deep gemm benchmarking(#3991)

parent 90a55e25
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
benchmark/kernels/deepseek/benchmark_deepgemm_fp8_gemm.py
View file @ 50f28f65
...@@ -211,7 +211,6 @@ def get_benchmark(tp_size): ...@@ -211,7 +211,6 @@ def get_benchmark(tp_size):
x = torch.randn((m, k), device="cuda", dtype=torch.bfloat16) x = torch.randn((m, k), device="cuda", dtype=torch.bfloat16)
y = torch.randn((n, k), device="cuda", dtype=torch.bfloat16) y = torch.randn((n, k), device="cuda", dtype=torch.bfloat16)
# 预处理数据，在计时之前完成
x_fp8, x_scale = per_token_cast_to_fp8(x) x_fp8, x_scale = per_token_cast_to_fp8(x)
y_fp8, y_scale = per_block_cast_to_fp8(y) y_fp8, y_scale = per_block_cast_to_fp8(y)
x_scale_col_major = get_col_major_tma_aligned_tensor(x_scale.clone()) x_scale_col_major = get_col_major_tma_aligned_tensor(x_scale.clone())
......
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