-
Notifications
You must be signed in to change notification settings - Fork 258
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
add fp8 example and document (#1639)
Signed-off-by: xinhe3 <[email protected]>
- Loading branch information
Showing
12 changed files
with
775 additions
and
53 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,113 @@ | ||
| FP8 Quantization | ||
| ======= | ||
|
|
||
| 1. [Introduction](#introduction) | ||
| 2. [Supported Parameters](#supported-parameters) | ||
| 3. [Get Start with FP8 Quantization](#get-start-with-fp8-quantization) | ||
| 4. [Examples](#examples) | ||
|
|
||
| ## Introduction | ||
|
|
||
| Float point 8 (FP8) is a promising data type for low precision quantization which provides a data distribution that is completely different from INT8 and it's shown as below. | ||
|
|
||
| <div align="center"> | ||
| <img src="./imgs/fp8_dtype.png" height="250"/> | ||
| </div> | ||
|
|
||
| Intel Gaudi2, also known as HPU, provides this data type capability for low precision quantization, which includes `E4M3` and `E5M2`. For more information about these two data type, please refer to [link](https://arxiv.org/abs/2209.05433). | ||
|
|
||
| Intel Neural Compressor provides general quantization APIs to leverage HPU FP8 capability. with simple with lower memory usage and lower compute cost, 8 bit model | ||
|
|
||
| ## Supported Parameters | ||
|
|
||
| <style type="text/css"> | ||
| .tg {border-collapse:collapse;border-spacing:0;} | ||
| .tg td{border-color:black;border-style:solid;border-width:1px;font-family:Arial, sans-serif;font-size:14px; | ||
| overflow:hidden;padding:10px 5px;word-break:normal;} | ||
| .tg th{border-color:black;border-style:solid;border-width:1px;font-family:Arial, sans-serif;font-size:14px; | ||
| font-weight:normal;overflow:hidden;padding:10px 5px;word-break:normal;} | ||
| .tg .tg-fymr{border-color:inherit;font-weight:bold;text-align:left;vertical-align:top} | ||
| .tg .tg-0pky{border-color:inherit;text-align:left;vertical-align:top} | ||
| </style> | ||
| <table class="tg"><thead> | ||
| <tr> | ||
| <th class="tg-fymr">Attribute</th> | ||
| <th class="tg-fymr">Description</th> | ||
| <th class="tg-fymr">Values</th> | ||
| </tr></thead> | ||
| <tbody> | ||
| <tr> | ||
| <td class="tg-0pky">fp8_config</td> | ||
| <td class="tg-0pky">The target data type of FP8 quantization.</td> | ||
| <td class="tg-0pky">E4M3 (default) - As Fig. 2<br>E5M2 - As Fig. 1.</td> | ||
| </tr> | ||
| <tr> | ||
| <td class="tg-0pky">hp_dtype</td> | ||
| <td class="tg-0pky">The high precision data type of non-FP8 operators.</td> | ||
| <td class="tg-0pky">bf16 (default) - torch.bfloat16<br>fp16 - torch.float16.<br>fp32 - torch.float32.</td> | ||
| </tr> | ||
| <tr> | ||
| <td class="tg-0pky">observer</td> | ||
| <td class="tg-0pky">The observer to measure the statistics.</td> | ||
| <td class="tg-0pky">maxabs (default), saves all tensors to files.</td> | ||
| </tr> | ||
| <tr> | ||
| <td class="tg-0pky">allowlist</td> | ||
| <td class="tg-0pky">List of nn.Module names or types to quantize. When setting an empty list, all the supported modules will be quantized by default. See Supported Modules. Not setting the list at all is not recommended as it will set the allowlist to these modules only: torch.nn.Linear, torch.nn.Conv2d, and BMM.</td> | ||
| <td class="tg-0pky">Default = {'names': [], 'types': <span title=["Matmul","Linear","FalconLinear","KVCache","Conv2d","LoRACompatibleLinear","LoRACompatibleConv","Softmax","ModuleFusedSDPA","LinearLayer","LinearAllreduce","ScopedLinearAllReduce","LmHeadLinearAllreduce"]>FP8_WHITE_LIST}</span></td> | ||
| </tr> | ||
| <tr> | ||
| <td class="tg-0pky">blocklist</td> | ||
| <td class="tg-0pky">List of nn.Module names or types not to quantize. Defaults to empty list, so you may omit it from the config file.</td> | ||
| <td class="tg-0pky">Default = {'names': [], 'types': ()}</td> | ||
| </tr> | ||
| <tr> | ||
| <td class="tg-0pky">mode</td> | ||
| <td class="tg-0pky">The mode, measure or quantize, to run HQT with.</td> | ||
| <td class="tg-0pky">MEASURE - Measure statistics of all modules and emit the results to dump_stats_path.<br>QUANTIZE - Quantize and run the model according to the provided measurements.<br>AUTO (default) - Select from [MEASURE, QUANTIZE] automatically.</td> | ||
| </tr> | ||
| <tr> | ||
| <td class="tg-0pky">dump_stats_path</td> | ||
| <td class="tg-0pky">The path to save and load the measurements. The path is created up until the level before last "/". The string after the last / will be used as prefix to all the measurement files that will be created.</td> | ||
| <td class="tg-0pky">Default = "./hqt_output/measure"</td> | ||
| </tr> | ||
| <tr> | ||
| <td class="tg-0pky">scale_method</td> | ||
| <td class="tg-0pky">The method for calculating the scale from the measurement.</td> | ||
| <td class="tg-0pky">- without_scale - Convert to/from FP8 without scaling.<br>- unit_scale - Always use scale of 1.<br>- maxabs_hw (default) - Scale is calculated to stretch/compress the maxabs measurement to the full-scale of FP8 and then aligned to the corresponding HW accelerated scale.<br>- maxabs_pow2 - Scale is calculated to stretch/compress the maxabs measurement to the full-scale of FP8 and then rounded to the power of 2.<br>- maxabs_hw_opt_weight - Scale of model params (weights) is chosen as the scale that provides minimal mean-square-error between quantized and non-quantized weights, from all possible HW accelerated scales. Scale of activations is calculated the same as maxabs_hw.<br>- act_maxabs_pow2_weights_pcs_opt_pow2 - Scale of model params (weights) is calculated per-channel of the params tensor. The scale per-channel is calculated the same as maxabs_hw_opt_weight. Scale of activations is calculated the same as maxabs_pow2.<br>- act_maxabs_hw_weights_pcs_maxabs_pow2 - Scale of model params (weights) is calculated per-channel of the params tensor. The scale per-channel is calculated the same as maxabs_pow2. Scale of activations is calculated the same as maxabs_hw.</td> | ||
| </tr> | ||
| <tr> | ||
| <td class="tg-0pky">measure_exclude</td> | ||
| <td class="tg-0pky">If this attribute is not defined, the default is OUTPUT. Since most models do not require measuring output tensors, you can exclude it to speed up the measurement process.</td> | ||
| <td class="tg-0pky">NONE - All tensors are measured.<br>OUTPUT (default) - Excludes measurement of output tensors.</td> | ||
| </tr> | ||
| </tbody></table> | ||
|
|
||
| ## Get Start with FP8 Quantization | ||
|
|
||
| ### Demo Usage | ||
|
|
||
| ```python | ||
| from neural_compressor.torch.quantization import ( | ||
| FP8Config, | ||
| prepare, | ||
| convert, | ||
| ) | ||
| import torchvision.models as models | ||
|
|
||
| model = models.resnet18() | ||
| qconfig = FP8Config(fp8_config="E4M3") | ||
| model = prepare(model, qconfig) | ||
| # customer defined calibration | ||
| calib_func(model) | ||
| model = convert(model) | ||
| ``` | ||
|
|
||
| ## Examples | ||
|
|
||
| | Task | Example | | ||
| |----------------------|---------| | ||
| | Computer Vision (CV) | [Link](../../examples/3.x_api/pytorch/cv/fp8_quant/) | | ||
| | Large Language Model (LLM) | [Link](https://github.com/HabanaAI/optimum-habana-fork/tree/habana-main/examples/text-generation#running-with-fp8) | | ||
|
|
||
| > Note: For LLM, Optimum-habana provides higher performance based on modified modeling files, so here the Link of LLM goes to Optimum-habana, which utilize Intel Neural Compressor for FP8 quantization internally. |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,28 @@ | ||
| # ImageNet FP8 Quantization | ||
|
|
||
| This implements FP8 quantization of popular model architectures, such as ResNet on the ImageNet dataset, which is supported by Intel Gaudi2 AI Accelerator. | ||
|
|
||
| ## Requirements | ||
|
|
||
| To try on Intel Gaudi2, docker image with Gaudi Software Stack is recommended, please refer to following script for environment setup. More details can be found in [Gaudi Guide](https://docs.habana.ai/en/latest/Installation_Guide/Bare_Metal_Fresh_OS.html#launch-docker-image-that-was-built). | ||
| ```bash | ||
| # Run a container with an interactive shell | ||
| docker run -it --runtime=habana -e HABANA_VISIBLE_DEVICES=all -e OMPI_MCA_btl_vader_single_copy_mechanism=none --cap-add=sys_nice --net=host --ipc=host vault.habana.ai/gaudi-docker/1.17.0/ubuntu22.04/habanalabs/pytorch-installer-2.2.0:latest | ||
| ``` | ||
|
|
||
| - Install requirements | ||
| - `pip install -r requirements.txt` | ||
| - Download the ImageNet dataset from http://www.image-net.org/ | ||
| - Then, move and extract the training and validation images to labeled subfolders, using [the following shell script](extract_ILSVRC.sh) | ||
|
|
||
| ## Quantizaiton | ||
|
|
||
| To quant a model and validate accaracy, run `main.py` with the desired model architecture and the path to the ImageNet dataset: | ||
|
|
||
| ```bash | ||
| python main.py --pretrained -t -a resnet50 -b 30 /path/to/imagenet | ||
| ``` | ||
| or | ||
| ```bash | ||
| bash run_quant.sh --input_model=resnet50 --dataset_location=/path/to/imagenet | ||
| ``` |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,80 @@ | ||
| #!/bin/bash | ||
| # | ||
| # script to extract ImageNet dataset | ||
| # ILSVRC2012_img_train.tar (about 138 GB) | ||
| # ILSVRC2012_img_val.tar (about 6.3 GB) | ||
| # make sure ILSVRC2012_img_train.tar & ILSVRC2012_img_val.tar in your current directory | ||
| # | ||
| # Adapted from: | ||
| # https://github.com/facebook/fb.resnet.torch/blob/master/INSTALL.md | ||
| # https://gist.github.com/BIGBALLON/8a71d225eff18d88e469e6ea9b39cef4 | ||
| # | ||
| # imagenet/train/ | ||
| # ├── n01440764 | ||
| # │ ├── n01440764_10026.JPEG | ||
| # │ ├── n01440764_10027.JPEG | ||
| # │ ├── ...... | ||
| # ├── ...... | ||
| # imagenet/val/ | ||
| # ├── n01440764 | ||
| # │ ├── ILSVRC2012_val_00000293.JPEG | ||
| # │ ├── ILSVRC2012_val_00002138.JPEG | ||
| # │ ├── ...... | ||
| # ├── ...... | ||
| # | ||
| # | ||
| # Make imagnet directory | ||
| # | ||
| mkdir imagenet | ||
| # | ||
| # Extract the training data: | ||
| # | ||
| # Create train directory; move .tar file; change directory | ||
| mkdir imagenet/train && mv ILSVRC2012_img_train.tar imagenet/train/ && cd imagenet/train | ||
| # Extract training set; remove compressed file | ||
| tar -xvf ILSVRC2012_img_train.tar && rm -f ILSVRC2012_img_train.tar | ||
| # | ||
| # At this stage imagenet/train will contain 1000 compressed .tar files, one for each category | ||
| # | ||
| # For each .tar file: | ||
| # 1. create directory with same name as .tar file | ||
| # 2. extract and copy contents of .tar file into directory | ||
| # 3. remove .tar file | ||
| find . -name "*.tar" | while read NAME ; do mkdir -p "${NAME%.tar}"; tar -xvf "${NAME}" -C "${NAME%.tar}"; rm -f "${NAME}"; done | ||
| # | ||
| # This results in a training directory like so: | ||
| # | ||
| # imagenet/train/ | ||
| # ├── n01440764 | ||
| # │ ├── n01440764_10026.JPEG | ||
| # │ ├── n01440764_10027.JPEG | ||
| # │ ├── ...... | ||
| # ├── ...... | ||
| # | ||
| # Change back to original directory | ||
| cd ../.. | ||
| # | ||
| # Extract the validation data and move images to subfolders: | ||
| # | ||
| # Create validation directory; move .tar file; change directory; extract validation .tar; remove compressed file | ||
| mkdir imagenet/val && mv ILSVRC2012_img_val.tar imagenet/val/ && cd imagenet/val && tar -xvf ILSVRC2012_img_val.tar && rm -f ILSVRC2012_img_val.tar | ||
| # get script from soumith and run; this script creates all class directories and moves images into corresponding directories | ||
| wget -qO- https://raw.githubusercontent.com/soumith/imagenetloader.torch/master/valprep.sh | bash | ||
| # | ||
| # This results in a validation directory like so: | ||
| # | ||
| # imagenet/val/ | ||
| # ├── n01440764 | ||
| # │ ├── ILSVRC2012_val_00000293.JPEG | ||
| # │ ├── ILSVRC2012_val_00002138.JPEG | ||
| # │ ├── ...... | ||
| # ├── ...... | ||
| # | ||
| # | ||
| # Check total files after extract | ||
| # | ||
| # $ find train/ -name "*.JPEG" | wc -l | ||
| # 1281167 | ||
| # $ find val/ -name "*.JPEG" | wc -l | ||
| # 50000 | ||
| # |
Oops, something went wrong.