[Neuron] Support quantization on neuron (#18283)

Signed-off-by: Satyajith Chilappagari <satchill@amazon.com>
This commit is contained in:
Satyajith Chilappagari
2025-05-27 15:10:33 -07:00
committed by GitHub
parent b48d5cca16
commit e0cbad4e30
3 changed files with 20 additions and 2 deletions

View File

@@ -13,6 +13,12 @@ from vllm.model_executor.layers.quantization.base_config import (
SUPPORTED_QUANT_DTYPE_LIST = ['s8', 'f8e4m3fn']
class AlwaysSupportedDtypes(list):
def __contains__(self, item):
return True
class NeuronQuantConfig(QuantizationConfig):
"""Int8 Quantization Config class for Neuron Backend."""
@@ -35,7 +41,8 @@ class NeuronQuantConfig(QuantizationConfig):
return "neuron_quant"
def get_supported_act_dtypes(self) -> list[str]:
return SUPPORTED_QUANT_DTYPE_LIST
# Neuron implements custom handling logic for quantization support
return AlwaysSupportedDtypes()
@classmethod
def get_min_capability(cls) -> int: