mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-05-21 03:36:59 +08:00
correct minimum capability req for channelwise torch
Signed-off-by: vllmellm <vllm.ellm@embeddedllm.com>
This commit is contained in:
parent
a76f7bb90c
commit
f10171cb3d
@ -230,9 +230,6 @@ class RowWiseTorchScaledMMLinearKernel(TorchScaledMMLinearKernel):
|
|||||||
|
|
||||||
|
|
||||||
class ChannelWiseTorchScaledMMLinearKernel(TorchScaledMMLinearKernel):
|
class ChannelWiseTorchScaledMMLinearKernel(TorchScaledMMLinearKernel):
|
||||||
@classmethod
|
|
||||||
def get_min_capability(cls) -> int:
|
|
||||||
return 94
|
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def can_implement(cls, c: FP8ScaledMMLinearLayerConfig) -> tuple[bool, str | None]:
|
def can_implement(cls, c: FP8ScaledMMLinearLayerConfig) -> tuple[bool, str | None]:
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user