-
Notifications
You must be signed in to change notification settings - Fork 646
Closed
Labels
module: quantizationIssues related to quantizationIssues related to quantizationmodule: xnnpackIssues related to xnnpack delegation and the code under backends/xnnpack/Issues related to xnnpack delegation and the code under backends/xnnpack/rfcRequest for comment and feedback on a post, proposal, etc.Request for comment and feedback on a post, proposal, etc.
Description
hi.
could some kinder helper tell me what do"128" in "Groupwise 4-bit (128)" indicatedin the "https://github.com/pytorch/executorch/tree/main/examples/models/llama2" ?
Thank u
Metadata
Metadata
Assignees
Labels
module: quantizationIssues related to quantizationIssues related to quantizationmodule: xnnpackIssues related to xnnpack delegation and the code under backends/xnnpack/Issues related to xnnpack delegation and the code under backends/xnnpack/rfcRequest for comment and feedback on a post, proposal, etc.Request for comment and feedback on a post, proposal, etc.
Type
Projects
Milestone
Relationships
Development
Select code repository
Activity
digantdesai commentedon May 9, 2024
In the case of the LLama2 Linear operation, the weights are quantized. There are various methods to perform quantization. In this instance, we utilized "Symmetric, per channel groupwise" quantization to convert and represent the original fp32 weight elements as int4.
The term "groupwise" refers to the number of weight elements in the same output channels that are quantized together and share the same quantization scale. For this particular case, we empirically chose a group size of 128. But support other 'standard' values like 32 or 256.
kimishpatel commentedon May 9, 2024
Groupwise in "Groupwise 4-bit" refers to the how many elements are in a group that share the quantization parameters. See this and this for the details on what are the quantization parametesr, names scale and zero point.
So for example weight tensor of linear layer might of shape (NxK) [4096, 1024]. 4096=N=# of output channels, 1024=K=# of input channels.
If you we quantize entire tensor with one set of quantization parameters then we have per tensor quantization.
If we quantize each input channel with one set of quantization parameters then we have per channel quantization. In this case each channel is one group and groupsize is 1024. Thus quantization parameter has size [4096, 1] corresponding to each output channel.
If we quantize each input channel with more than one set of quantization parameters then we get groupwise per channel quantization. For example groupwise = 128 means, each channel has is divided into group of 128 elements. In our examples you have 1024/128= 8 groups. Thus quantizaton parametesr are of size [4096, 8]
l2002924700 commentedon May 10, 2024
@kimishpatel @digantdesai Thank u. After I read your answer, I have issues as follow:
kimishpatel commentedon May 10, 2024
Ansering some
Yes
Naming is bit confusing but 8da4w represents 4bit weight quantization. 8da refers to the need of quantizing activation, during inference, to 8 bits. This is known as dynamic quantization. https://pytorch.org/tutorials/recipes/recipes/dynamic_quantization.html
--quantization_mode 8da4w. Please see https://github.com/pytorch/executorch/tree/main/examples/models/llama2#option-c-download-and-export-llama3-8b-model. (You can use llama2 7b or llama2 8b model. Just highlighting this as the readme contains the repro instructions)