Skip to content

Commit 25e4e87

Browse files
author
Doug Lehr
committed
remove duplicate aiter_GEMM_check function
1 parent bcc4e69 commit 25e4e87

File tree

1 file changed

+0
-10
lines changed

1 file changed

+0
-10
lines changed

vllm/model_executor/layers/utils.py

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -94,16 +94,6 @@ def default_unquantized_gemm(layer: torch.nn.Module,
9494
bias: Optional[torch.Tensor] = None):
9595
return torch.nn.functional.linear(x, weight, bias)
9696

97-
def aiter_GEMM_check(m, n, k):
98-
if ((n == 5120 and k == 2880)
99-
or (n == 2880 and k == 4096)
100-
or (n == 128 and k == 2880)
101-
or (n == 640 and k == 2880)
102-
or (n == 2880 and k == 512)):
103-
return True
104-
return False
105-
106-
10797
def aiter_GEMM_check(m, n, k):
10898
if ((n == 5120 and k == 2880)
10999
or (n == 2880 and k == 4096)

0 commit comments

Comments
 (0)