fix benchmark fp8 blockwise group gemm (#8815)
This commit is contained in:
@@ -97,9 +97,7 @@ def bench_deepgemm(
|
|||||||
)
|
)
|
||||||
|
|
||||||
def run_deepgemm():
|
def run_deepgemm():
|
||||||
deep_gemm.m_grouped_gemm_fp8_fp8_bf16_nt_contiguous(
|
deep_gemm.m_grouped_fp8_gemm_nt_contiguous(x_fp8, y_fp8, out, m_indices)
|
||||||
x_fp8, y_fp8, out, m_indices
|
|
||||||
)
|
|
||||||
|
|
||||||
# warmup
|
# warmup
|
||||||
for _ in range(num_warmup):
|
for _ in range(num_warmup):
|
||||||
|
|||||||
Reference in New Issue
Block a user