From 5e4badc5777ff1371358ec73e0aac91229181fec Mon Sep 17 00:00:00 2001 From: Chenggang Zhao Date: Mon, 10 Mar 2025 13:10:16 +0800 Subject: [PATCH] Fix type lint --- deep_gemm/jit_kernels/gemm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deep_gemm/jit_kernels/gemm.py b/deep_gemm/jit_kernels/gemm.py index 082c8ee..d97a615 100644 --- a/deep_gemm/jit_kernels/gemm.py +++ b/deep_gemm/jit_kernels/gemm.py @@ -56,7 +56,7 @@ def get_smem_size(num_stages: int, k: int, block_m: int, block_n: int, block_k: def get_best_configs(m: int, n: int, k: int, num_groups: int, num_sms: int, - is_grouped_contiguous: bool = False) -> Tuple[int, int, int, int, int]: + is_grouped_contiguous: bool = False) -> Tuple[int, int, int, int, int, int]: if not is_grouped_contiguous: # TODO: for some cases, smaller M block is better, add them into tuning space block_ms = (64 if m <= 64 else 128, )