Skip to content

Commit 891b4cc

Browse files
committed
fix(quant): add layer_type parameter to torchair w8a8 dynamic
Signed-off-by: Anionex <[email protected]>
1 parent dc5c02f commit 891b4cc

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

vllm_ascend/torchair/quantization/torchair_w8a8_dynamic.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -778,7 +778,8 @@ def get_perchannel_param(
778778
return params_dict
779779

780780
def get_pergroup_param(self, input_size: int, output_size: int,
781-
params_dtype: torch.dtype) -> Dict[str, Any]:
781+
params_dtype: torch.dtype,
782+
layer_type: Optional[str] = None) -> Dict[str, Any]:
782783
return {}
783784

784785
@staticmethod

0 commit comments

Comments
 (0)