Ds-inference Int8 support through ZeroQuant technology#2217
Conversation
|
hey @RezaYazdaniAminabadi I have been waiting for this PR for a long time and recently tried out your branch for ZeroQuant (with GPT-J). I found a couple of issues: |
Hey @sdpmas |
|
@sdpmas, btw, can you please paste the whole trace here to help me debug this better? Thanks |
|
yea, sure, here's what the trace looks like: and my code looks something like this: |
|
@RezaYazdaniAminabadi any estimates on when you will release int8 kernels? that would be really helpful. I've been trying to speed up generation for GPT-J and ZeroQuant seems to be the way to go! |
|
@pai4451 this has been fixed. |
@mayank31398 I did install the latest master branch of DeepSpeed (you can see my output of pip freeze). But I initialize DeepSpeed by specifying the keyword |
|
After upgrade to I can mitigate the repetitiveness by increasing the value of |
Hi @pai4451 |
|
I have tested this with Here is the text I see when using fp16 version of checkpoint: By passing Here is the INT8 version of the text: |
|
Thanks @RezaYazdaniAminabadi for your explanation. I will keep monitoring the output from FP16 and Int8. I guess both of them will return repetitive result in some cases. Thanks a lot! |
|
Is there any guide to running inference on compressed models(especially ZeroQuant)? |
|
Please see the demo scripts for BLOOM Inference here: |
|
@RezaYazdaniAminabadi |
|
When trying to generate int8 shards (from Full TracebackTraceback (most recent call last):
File "t.py", line 402, in
model = deepspeed.init_inference(model,
File "/home/ubuntu/venv/lib/python3.8/site-packages/deepspeed/__init__.py", line 305, in init_inference
engine = InferenceEngine(model,
File "/home/ubuntu/venv/lib/python3.8/site-packages/deepspeed/inference/engine.py", line 145, in __init__
self._apply_injection_policy(
File "/home/ubuntu/venv/lib/python3.8/site-packages/deepspeed/inference/engine.py", line 342, in _apply_injection_policy
replace_transformer_layer(client_module,
File "/home/ubuntu/venv/lib/python3.8/site-packages/deepspeed/module_inject/replace_module.py", line 865, in replace_transformer_layer
load_model_with_checkpoint(
File "/home/ubuntu/venv/lib/python3.8/site-packages/deepspeed/module_inject/load_checkpoint.py", line 229, in load_model_with_checkpoint
load_module_recursive(r_module)
File "/home/ubuntu/venv/lib/python3.8/site-packages/deepspeed/module_inject/load_checkpoint.py", line 224, in load_module_recursive
load_module_recursive(
File "/home/ubuntu/venv/lib/python3.8/site-packages/deepspeed/module_inject/load_checkpoint.py", line 224, in load_module_recursive
load_module_recursive(
File "/home/ubuntu/venv/lib/python3.8/site-packages/deepspeed/module_inject/load_checkpoint.py", line 222, in load_module_recursive
layer_policies[child.__class__](child, prefix + name + '.')
File "/home/ubuntu/venv/lib/python3.8/site-packages/deepspeed/module_inject/load_checkpoint.py", line 142, in load_transformer_layer
module.attention.attn_qkvw = mp_replace.copy(module.attention.attn_qkvw,
File "/home/ubuntu/venv/lib/python3.8/site-packages/deepspeed/module_inject/replace_module.py", line 109, in copy
dst.data.copy_(weight_split.contiguous())
RuntimeError: The size of tensor a (6144) must match the size of tensor b (14336) at non-singleton dimension 1
My code looks like this ( Is it enough to simply change the dtype parameter on EDIT: I am synced to the tip of master |
|
@zcrypt0 you can't quantize just be specifying int8 as dtype Not sure how these weights are quantized. But ZeroQuant also does layer distillation for quantization (we don't know what data Microsoft used for this). -> is this documentation available @RezaYazdaniAminabadi ? This is an example to quantize gpt2 |
|
Much appreciated @mayank31398 I was hoping it was just magic, but I will read the paper. When I try to use the quantized weights ( The error (note this doesn't occur on an 8xA6000 node): Full TracebackTraceback (most recent call last):
File "t.py", line 422, in
model = deepspeed.init_inference(model,
File "/home/ubuntu/venv/lib/python3.8/site-packages/deepspeed/__init__.py", line 305, in init_inference
engine = InferenceEngine(model,
File "/home/ubuntu/venv/lib/python3.8/site-packages/deepspeed/inference/engine.py", line 145, in __init__
self._apply_injection_policy(
File "/home/ubuntu/venv/lib/python3.8/site-packages/deepspeed/inference/engine.py", line 342, in _apply_injection_policy
replace_transformer_layer(client_module,
File "/home/ubuntu/venv/lib/python3.8/site-packages/deepspeed/module_inject/replace_module.py", line 895, in replace_transformer_layer
load_model_with_checkpoint(replaced_module,
File "/home/ubuntu/venv/lib/python3.8/site-packages/deepspeed/module_inject/load_checkpoint.py", line 229, in load_model_with_checkpoint
load_module_recursive(r_module)
File "/home/ubuntu/venv/lib/python3.8/site-packages/deepspeed/module_inject/load_checkpoint.py", line 224, in load_module_recursive
load_module_recursive(
File "/home/ubuntu/venv/lib/python3.8/site-packages/deepspeed/module_inject/load_checkpoint.py", line 224, in load_module_recursive
load_module_recursive(
File "/home/ubuntu/venv/lib/python3.8/site-packages/deepspeed/module_inject/load_checkpoint.py", line 222, in load_module_recursive
layer_policies[child.__class__](child, prefix + name + '.')
File "/home/ubuntu/venv/lib/python3.8/site-packages/deepspeed/module_inject/load_checkpoint.py", line 138, in load_transformer_layer
load_parameters(child, prefix + n + '.')
File "/home/ubuntu/venv/lib/python3.8/site-packages/deepspeed/module_inject/load_checkpoint.py", line 128, in load_parameters
p.data.copy_(bias_split)
RuntimeError: The size of tensor a (6144) must match the size of tensor b (4608) at non-singleton dimension 0
|
|
I haven't tried on 7 GPUs. I can give it a shot. |
* Fix the layer-past for GPT based models (deepspeedai#2196) * Add gradient_average flag support for sparse grads (deepspeedai#2188) * Add gradient_average flag support for sparse grads * formatting fixes * Add tests Co-authored-by: Olatunji Ruwase <[email protected]> * Adding additional instructiosn in the compression tutorial on pre-training distillation and quantization for GPT (deepspeedai#2197) Co-authored-by: Olatunji Ruwase <[email protected]> Co-authored-by: Jeff Rasley <[email protected]> * Log user config exactly (deepspeedai#2201) * Fix the tensor-slicing copy for qkv parameters (deepspeedai#2198) Co-authored-by: Olatunji Ruwase <[email protected]> * Refactor Distributed Tests (deepspeedai#2180) Refactor Distributed unit tests * fix table syntax (deepspeedai#2204) Co-authored-by: Conglong Li <[email protected]> Co-authored-by: Olatunji Ruwase <[email protected]> * Correctly detect offload configuration (deepspeedai#2208) Co-authored-by: Jeff Rasley <[email protected]> * add cuda 11.7 (deepspeedai#2211) * add cuda 11.7 * formatting * use torch 1.9 (deepspeedai#2215) * [zero-3] print warning once and support torch parameter (deepspeedai#2127) * print warning only once. * add support for torch param and only warn on gpu 0 * remove type checking. will be done on a new PR with more tests. Co-authored-by: Jeff Rasley <[email protected]> Co-authored-by: Olatunji Ruwase <[email protected]> * Add support of OPT models (deepspeedai#2205) * add opt replace policy * simplify inf. api * fix opt replace policy * fix use-cash & add relu * Add support of custom MLP act. function * Revert "simplify inf. api" This reverts commit 9e910fc. * fix the inference API (temp. solution) * fix code formatting * add unit tests for OPT models. * refactor pre-attention layer norm configuration * add support of opt-350m model * refactor the HF model config initialization * fix hf model config issue Co-authored-by: Reza Yazdani <[email protected]> Co-authored-by: Jeff Rasley <[email protected]> Co-authored-by: Reza Yazdani <[email protected]> * fix typos in readme. (deepspeedai#2218) Co-authored-by: Olatunji Ruwase <[email protected]> * [device abstraction] add device abstraction to allow other device than CUDA be used * Fix regression w. dist_init_required (deepspeedai#2225) * add doc for new bert example (deepspeedai#2224) * Remove the random-generator from context during inference (deepspeedai#2228) * Fix the tensor-slicing copy for qkv parameters * remove the random-generator from context during inference * formatting Co-authored-by: Jeff Rasley <[email protected]> * allow saving ckpt w/o ckpt json + bloom copy fix (deepspeedai#2237) * Correctly detect zero_offload (deepspeedai#2213) * Correctly detect offload configuration * Correctly detect offload configuration * Handle deprecated cpu offload setting * Correcly detect zero_offload setting * Minor tweak Co-authored-by: Jeff Rasley <[email protected]> Co-authored-by: Ammar Ahmad Awan <[email protected]> * update videos (deepspeedai#2249) * Refactor dist tests: Checkpointing (deepspeedai#2202) Refactor distributed tests: checkpointing Co-authored-by: Michael Wyatt <[email protected]> * Make OPT policy backward compatible with pre-OPT transformers versions (deepspeedai#2254) * fix ds-inference without policy (deepspeedai#2247) Co-authored-by: Jeff Rasley <[email protected]> * bump to 0.7.2 * Enable contiguous gradients with Z1+MoE (deepspeedai#2250) MoE training with zero stage 1 only works with `contiguous gradients=True`. * [rebase-202208] additional changes needed when rebase to 202208 * [rebase] cleanup direct cuda usage after merge * Correctly detect CPU optimizer usage (deepspeedai#2257) * Correctly detect CPU optimizer usage * Update nv-transformers-v100.yml (deepspeedai#2259) Co-authored-by: Jeff Rasley <[email protected]> * [precommit] fix pre-commit issues * Update half precision header guards (deepspeedai#2261) * fix deepspeedai#2240: wrong time unit in flops_profiler (deepspeedai#2241) Co-authored-by: Olatunji Ruwase <[email protected]> Co-authored-by: Jeff Rasley <[email protected]> * bump to 0.7.3 * Add blob storage to CI runners (deepspeedai#2260) Add blob storage to CI runners and enable for transformers cache on inference tests * Update replace_module.py, test-gptj.py related fix (deepspeedai#2269) Fix RuntimeError: Boolean value of Tensor with more than one value is ambiguous when running test-gptj.py * Fix OrderedDict import for python3.6 (deepspeedai#2267) Co-authored-by: Olatunji Ruwase <[email protected]> * Ds inference/fix mp2 (deepspeedai#2270) * Trajepl: nebula load fix (deepspeedai#2182) Co-authored-by: Olatunji Ruwase <[email protected]> Co-authored-by: chenguo <[email protected]> * prevent torch ext folder mkdir at tmp (deepspeedai#2274) * Ds-inference Int8 support through ZeroQuant technology (deepspeedai#2217) Co-authored-by: Jeff Rasley <[email protected]> * add a new unit test for cuda ops (deepspeedai#2278) Co-authored-by: cmikeh2 <[email protected]> * Add to codeowners file (deepspeedai#2279) * [pin_memory] make pin_memory select device type * Memory Access Utility (deepspeedai#2276) Co-authored-by: Ammar Ahmad Awan <[email protected]> * Fp32 accuracy bug fix (deepspeedai#2285) Co-authored-by: Arash Bakhtiari <[email protected]> Co-authored-by: Arash Bakhtiari <[email protected]> * Refactor universal checkpointing and tensor fragments (deepspeedai#2253) * Refactor universal checkpointing and tensor fragments * Formatting * [ds-inference] fix progress bar (deepspeedai#2286) when loading the non-sharded checkpoint update the progress bar (fix by @RezaYazdaniAminabadi) - I've just tested it to work. Co-authored-by: Olatunji Ruwase <[email protected]> * Offload all gradients to nvme (deepspeedai#2282) * fused bias relu unittest (deepspeedai#2297) * fix for pytest picking up local deepspeed dir instead of installed deepspeed (deepspeedai#2299) * Fix for Zero3 when MP>1 and at least one batch param undefined (deepspeedai#2289) Co-authored-by: anthony.301 <[email protected]> Co-authored-by: Jeff Rasley <[email protected]> * [downstream] merge from xpu support downstream * Unit test for bias add kernel (deepspeedai#2298) * added unit test * Update pt_binding.cpp * formatting * Update test_bias_add.py * Update relu.cu with mem_access_utils (deepspeedai#2306) * Add tensor parallel inference unit tests (deepspeedai#2232) Co-authored-by: Reza Yazdani <[email protected]> Co-authored-by: Jeff Rasley <[email protected]> Co-authored-by: Sam Ade Jacobs <[email protected]> * Fix the residual add mp scaling for GPTNeoX (deepspeedai#2310) * Add unit tests for residual_add kernels (deepspeedai#2307) * add inference eval scripts (deepspeedai#2303) * Upgrade P40 tests to torch 1.8 (deepspeedai#2316) Co-authored-by: Jeff Rasley <[email protected]> * ZeRO-Inference blog (deepspeedai#2271) * ZeRO-Inference blog * ZeRO-Inference blog * Format fixes * Apply feedback * Feedback * Update docs/_posts/2022-08-27-zero-inference.md Co-authored-by: Stas Bekman <[email protected]> * Update docs/_posts/2022-08-27-zero-inference.md Co-authored-by: Stas Bekman <[email protected]> * Address feedback * Format fixes * More tweaks * long sequence, nvme offload * Add image Co-authored-by: Stas Bekman <[email protected]> Co-authored-by: Jeff Rasley <[email protected]> * ZeRO-Inference blog - wrap up (deepspeedai#2321) * ZeRO-Inference blog - Update README (deepspeedai#2322) * refactor to use mem_access (deepspeedai#2317) * add quant unit test (deepspeedai#2315) * add quant unit test * add codeowner * format fix * fix undefined symbol: curandSetPseudoRandomGeneratorSeed * modify ref fn name and add comment * add comments * add 4bit quant 16groups * fix * modify groups in ref code * parameterize tensor shape * single param * detach tensor * remove -lcurand flag * add back -lcurand flag Co-authored-by: Ammar Ahmad Awan <[email protected]> * only override forward if using cuda-graph (deepspeedai#2291) * Add more options to inference benchmark (deepspeedai#2325) * bump to 0.7.4 * MOE residual matmult unit test (deepspeedai#2323) MOE residual matmul unit tests Co-authored-by: Michael Wyatt <[email protected]> Co-authored-by: Ammar Ahmad Awan <[email protected]> * [device] port cuda device to literal_device() in new tests * MOE matmult with memaccess (deepspeedai#2336) * Fix formatting * Remove redundant variable * Refactor residual add kernels (deepspeedai#2333) Co-authored-by: Ammar Ahmad Awan <[email protected]> * [accel_runtime] add pin_memory to accelerator runtime interface. * mem access for quantize kernel (deepspeedai#2331) * mem access for quantize kernel * format * format fp32 * modify quant kernel * modify quant kernel2 * modify format * format * fix comments in pytest * fix comments in pytest * format * rerun Co-authored-by: Ammar Ahmad Awan <[email protected]> Co-authored-by: Connor Holmes <[email protected]> * increase min pre-commit versions (deepspeedai#2346) * Extend scratch buffer for long prompts (deepspeedai#2212) Co-authored-by: Reza Yazdani <[email protected]> Co-authored-by: Reza Yazdani <[email protected]> Co-authored-by: Jeff Rasley <[email protected]> * fix zero docs (deepspeedai#2350) * Inference profiling updates/fixes (deepspeedai#2348) (deepspeedai#2349) Co-authored-by: Jeff Rasley <[email protected]> Co-authored-by: Michael Wyatt <[email protected]> * Kernel Data Conversion Utility (deepspeedai#2327) * Unify macro definitions and constants in a single file * Conversion utility implementation. * Fix reversion from formatting * Bugfixes after testing with correct DeepSpeed * Inline markers are available on both HIP + CUDA * Add Onebit Optimzers in __init__ (deepspeedai#2340) Co-authored-by: Saeyeol Lee <[email protected]> Co-authored-by: Olatunji Ruwase <[email protected]> * [accelerator abstraction] merge from deepspeedai#2320 * docs(mixture-of-experts-inference): fix typo in tuto (deepspeedai#2345) Co-authored-by: Olatunji Ruwase <[email protected]> * download cifar to blob storage (deepspeedai#2342) Co-authored-by: Olatunji Ruwase <[email protected]> * Refactor gptj_residual_add kernels for better readability (deepspeedai#2358) Co-authored-by: Reza Yazdani <[email protected]> * Updated issue templates (deepspeedai#2363) * Update issue templates * fix cuda invalid config error in dequant kernel (deepspeedai#2362) * format * remove round fn * Add missing pytest fixture scope (deepspeedai#2353) Co-authored-by: Olatunji Ruwase <[email protected]> Co-authored-by: Michael Wyatt <[email protected]> * Extend residual_add kernel tests to conver pre_attn_norm (deepspeedai#2354) Co-authored-by: Jeff Rasley <[email protected]> * Refactor fused_bias_residual kernels for better readability (deepspeedai#2356) Co-authored-by: Olatunji Ruwase <[email protected]> * Capture error message during sweep tests (deepspeedai#2351) * Collect error messages in results.csv Co-authored-by: Michael Wyatt <[email protected]> Co-authored-by: Olatunji Ruwase <[email protected]> * fix an exception when recursively casting dicts to fp16 (deepspeedai#2370) * Refactor remaining distributed tests (deepspeedai#2216) * batch of refactored tests * more test refactoring * fp16 test refactor * more refactors * added DistributedFixture class * applied DistributedFixture to first batch of tests as a trial * added DistributedFixture test and documentation * last tests * fixes for refactored tests * remove subdirs in workflow files * fix pytest syntax error * fix another syntax error * update imports * use DistFixture with elastic checkpoint test * missing import * update to shared class tmpdir for elastic test * moved test files * avoid duplicate test file name * last refactor and moving test files * formatting * fix broken import * testing forked AMD tests * update abstract method * use blob storage for accelerate and transformers tests * upgrade torch for acclerate CI Co-authored-by: Olatunji Ruwase <[email protected]> * Fix the MLP output tensor's shape (deepspeedai#2380) * allow building with latest CUDA (11.8), it is backwards compatible (deepspeedai#2390) * pin transformers version for unit tests (deepspeedai#2402) * Change type to tuple in replace_wo_policy isinstance check (deepspeedai#2387) Update the isinstance check inside the `replace_wo_policy` function to `tuple` and `str` instead of `dict`, since the layers are provided as a `tuple` type. Co-authored-by: Lev Kurilenko <[email protected]> Co-authored-by: Molly Smith <[email protected]> Co-authored-by: Lok Chand Koppaka <[email protected]> Co-authored-by: Ammar Ahmad Awan <[email protected]> * Checkpoint backwards-compatbility workaround (deepspeedai#2384) * Add predicated global load (deepspeedai#2373) Co-authored-by: Reza Yazdani <[email protected]> * change call site of literal_device, on_accel_device and accel_runtime to get_accelerator() call * add new interface definition from olruwase/accelerator_abstraction * MII blog post (deepspeedai#2418) Co-authored-by: Samyam Rajbhandari <[email protected]> Co-authored-by: Ammar Ahmad Awan <[email protected]> * Fix figure reference (deepspeedai#2419) * [docs] update news items * [docs] add mii repo link * Add SLURM Multinode Runner (deepspeedai#2404) Signed-off-by: Dashiell Stander <[email protected]> Co-authored-by: Dashiell Stander <[email protected]> Co-authored-by: Jeff Rasley <[email protected]> * Fix issue with corrupted output on long generation for GPT (deepspeedai#2359) Co-authored-by: Reza Yazdani <[email protected]> Co-authored-by: Olatunji Ruwase <[email protected]> * MII blog title update on Readme * DeepSpeed-MII title change in website * Fix GPT Neo-X multi-gpu inference (deepspeedai#2401) Co-authored-by: Reza Yazdani <[email protected]> Co-authored-by: Jeff Rasley <[email protected]> * MII-Public and MII-Azure subheading in mii post * CI fixes related to triton (deepspeedai#2422) * [docs] update mii blog title (deepspeedai#2423) * add SD injection policy (deepspeedai#2381) Co-authored-by: Reza Yazdani <[email protected]> Co-authored-by: Reza Yazdani <[email protected]> * [accelerator abstraction] remove name() from interface, device_name() should be used. * merge with master (ec13da6) * fix checkpoint loading when it is a dictionary (deepspeedai#2425) * Make error regex more generic in collect_results.py (deepspeedai#2415) Co-authored-by: Jeff Rasley <[email protected]> * fixes deepspeedai#2389 (deepspeedai#2411) truncating expert param storage for checkpointing Co-authored-by: Alexander Jipa <[email protected]> Co-authored-by: Michael Wyatt <[email protected]> * Fix for inference gpt-j test (deepspeedai#2430) * fix for gpt-j failing due to tokenizer error * limit number of gpt-j tokens generated due to low memory * Fixing bug 2361 (deepspeedai#2410) * fixing bug 2361 * adding pytest for config initialization * chaning expected output to FusedAdam * remove print statement * running yapf on modified files * running pre-commit formatting Co-authored-by: Olatunji Ruwase <[email protected]> * Universal checkpoint for zero stage 1 (deepspeedai#2284) * Refactor universal checkpointing and tensor fragments * Formatting * Support zero stage1; Expand TP dim * Remove debug prints * Detect sharded optimizer state * Format fixes * Encode reshaping guide * More symbolic constants Co-authored-by: Michael Wyatt <[email protected]> * only add deps if extra is explictly called (deepspeedai#2432) * Add TestInjectionPolicy inference unittest class for testing custom injection policies (deepspeedai#2426) This PR adds a TestInjectionPolicy inference unittest class for testing custom injection policies. This test differs from the existing tests in that the injection_policy dictionary is explicitly specified when calling the DeepSpeed init_inference API. The google/t5-v1_1-small text2text-generation model and the roberta-large fill-mask model are added as tests with the injection policy explicitly specified. This is done to expand our unittest coverage to test the path where the replace_wo_policy function is invoked (see deepspeedaiGH-2387). Co-authored-by: Lev Kurilenko <[email protected]> Co-authored-by: Michael Wyatt <[email protected]> * [memory estimators] new config args sync (deepspeedai#2431) Co-authored-by: Olatunji Ruwase <[email protected]> Co-authored-by: Jeff Rasley <[email protected]> * parallelize writing of layer checkpoint files across data parallel instances (deepspeedai#1419) * parallelize layer checkpoints across data parallel groups * use partition_uniform to determine start/end index values * formatting fix * config: add option for parallel write of layer checkpoints in pipeline stage * yapf fixes * enable parallel layer write according to config param * avoid extraneous makedir when rank 0 writes all layers Co-authored-by: Olatunji Ruwase <[email protected]> * Fix broken link to DeepSpeed Megatron fork (deepspeedai#2440) Co-authored-by: Lev Kurilenko <[email protected]> * bump to 0.7.5 * [OpBuilder] Add op builder abstraction * convert op builder usage in merged code * merge diff files from upstream * [OpBuilder] add create_op_builder interface in abstract_accelerator.py * remove files that is deleted from upstream * [OpBuilder] add left over op builder usage in tests * [OpBuilder] fix op builder usage in tests * [OpBuilder] fix <op builder>.NAME usage in tests to follow op builder abstraction design * import get_accelerator from deepspeed.accelerator directly * [OpBuilder] remove unused function and sync with main * add missing import * revert changes in device.py to avoid conflict with main * fix alexnet_model to use /tmp instead of /blob * Mingzhi/solve pr108 b (deepspeedai#115) * move ALL_OPs from __init__.py to all_Op.py to solve circular import * delete deepspeedexamples * fix import * fix regression (deepspeedai#117) * fix pin_memory * fix regression * fix error Signed-off-by: Dashiell Stander <[email protected]> Co-authored-by: Reza Yazdani <[email protected]> Co-authored-by: Mikhail Druzhinin <[email protected]> Co-authored-by: Olatunji Ruwase <[email protected]> Co-authored-by: Minjia Zhang <[email protected]> Co-authored-by: Jeff Rasley <[email protected]> Co-authored-by: Michael Wyatt <[email protected]> Co-authored-by: Kamal Raj <[email protected]> Co-authored-by: Conglong Li <[email protected]> Co-authored-by: Ammar Ahmad Awan <[email protected]> Co-authored-by: Arash Bakhtiari <[email protected]> Co-authored-by: Reza Yazdani <[email protected]> Co-authored-by: Zhihong Chen <[email protected]> Co-authored-by: Siddharth Singh <[email protected]> Co-authored-by: Connor Holmes <[email protected]> Co-authored-by: 叶志晟 <[email protected]> Co-authored-by: Molly Smith <[email protected]> Co-authored-by: trajep <[email protected]> Co-authored-by: chenguo <[email protected]> Co-authored-by: Arash Bakhtiari <[email protected]> Co-authored-by: Stas Bekman <[email protected]> Co-authored-by: Quentin Anthony <[email protected]> Co-authored-by: anthony.301 <[email protected]> Co-authored-by: Sam Ade Jacobs <[email protected]> Co-authored-by: Guanhua Wang <[email protected]> Co-authored-by: Saeyeol Lee <[email protected]> Co-authored-by: Saeyeol Lee <[email protected]> Co-authored-by: Jean-Louis Queguiner <[email protected]> Co-authored-by: Matt Smith <[email protected]> Co-authored-by: Thomas-MMJ <[email protected]> Co-authored-by: lekurile <[email protected]> Co-authored-by: Lev Kurilenko <[email protected]> Co-authored-by: Molly Smith <[email protected]> Co-authored-by: Lok Chand Koppaka <[email protected]> Co-authored-by: Samyam Rajbhandari <[email protected]> Co-authored-by: Dashiell Stander <[email protected]> Co-authored-by: Dashiell Stander <[email protected]> Co-authored-by: Andrey Chernykh <[email protected]> Co-authored-by: Alexander Jipa <[email protected]> Co-authored-by: Alexander Jipa <[email protected]> Co-authored-by: Joe Mayer <[email protected]> Co-authored-by: Adam Moody <[email protected]> Co-authored-by: AGUL <[email protected]>
|
I tried to run 8-bit quantized inference of BLOOM-176B on 8 40G A100 GPUs, but encountered the error: "AttributeError: 'GroupQuantizer' object has no attribute 'num_groups' ". I think it's because'GroupQuantizer' doe not initialize 'num_groups' attribute. Could you please help fix it? |
|
@JingfengYang, I've reproduced this and will update this thread once we have a fix. Sorry for the inconvenience and thanks for reporting this to us as well. |
|
Thanks! Also, when I'm using a prior committed version of this repo, there is not such error but a new error "AttributeError: 'Parameter' object has no attribute 'scale'" occurs. FYI, I'm using this script: https://github.com/huggingface/transformers-bloom-inference/blob/main/bloom-inference-scripts/bloom-ds-inference.py recommended by Huggingface official website to run 8-bit quantized inference of BLOOM-176B on 8 40G A100 GPUs. |
|
@JingfengYang, I think we've come up with a fix for the num_groups issue. I've pushed a PR #2645 that should fix this. I need to consult w. @RezaYazdaniAminabadi after the winter break to ensure I am not missing anything here, but feel free to give it a try on your side. |
|
Is this problem solved? I also encountered the same problem here.
|
|
@RezaYazdaniAminabadi After some simple modification of code, I ran my model in INT8 with low cuda memory usage, very appreciated for this. I wonder when will the full ZeroQuant, I mean INT8 calculation of Gemm will be released? |
|
Is there any guide I can use ZeroQuant? And is next example just about quantizing(ZeroQuant) gpt2? Then how can I inference to see how much latency is improved? |
|
Sorry, I leave this comment cuz there' no specific guide for inferencing ZeroQuant model. https://github.com/microsoft/DeepSpeedExamples/blob/master/compression/gpt2/bash_script/run_zero_quant.sh Setting Thanks. import time
from transformers import pipeline
from deepspeed.accelerator import get_accelerator
def print_latency(latency_set, title, warmup=3):
# trim warmup queries
latency_set = list(latency_set)
latency_set = latency_set[warmup:]
count = len(latency_set)
if count > 0:
latency_set.sort()
n50 = (count - 1) * 0.5 + 1
n90 = (count - 1) * 0.9 + 1
n95 = (count - 1) * 0.95 + 1
n99 = (count - 1) * 0.99 + 1
n999 = (count - 1) * 0.999 + 1
avg = sum(latency_set) / count
p50 = latency_set[int(n50) - 1]
p90 = latency_set[int(n90) - 1]
p95 = latency_set[int(n95) - 1]
p99 = latency_set[int(n99) - 1]
p999 = latency_set[int(n999) - 1]
print(f"====== latency stats {title} ======")
print("\tAvg Latency: {0:8.2f} ms".format(avg * 1000))
print("\tP50 Latency: {0:8.2f} ms".format(p50 * 1000))
print("\tP90 Latency: {0:8.2f} ms".format(p90 * 1000))
print("\tP95 Latency: {0:8.2f} ms".format(p95 * 1000))
print("\tP99 Latency: {0:8.2f} ms".format(p99 * 1000))
print("\t999 Latency: {0:8.2f} ms".format(p999 * 1000))
deepspeed.init_distributed()
dtype = torch.float16
pipe = pipeline("text-generation", model=model, framework="pt", tokenizer=tokenizer)
if True: # using deepspeed
pipe.model = deepspeed.init_inference(
pipe.model,
dtype=dtype,
tensor_parallel={"tp_size": 1},
replace_with_kernel_inject=True,
)
pipe.model.profile_model_time()
responses = []
times = []
mtimes = []
for i in range(30):
get_accelerator().synchronize()
start = time.time()
r = pipe("DeepSpeed is", do_sample=False, max_new_tokens=50)
get_accelerator().synchronize()
end = time.time()
responses.append(r)
times.append(end - start) # / (args.max_tokens - 3))
if True: # using deepspeed
mtimes.append(sum(pipe.model.model_times()))
if args.local_rank == 0:
print_latency(times, "(e2e) latency")
if True: # using deepspeed
print_latency(mtimes, "(model-only) latency")
print_latency(map(lambda t: t / (50 - 3), times), "(e2e) per token latency")
print(f"RESPONSE 0:")
print("-" * 30)
print(responses[0][0]["generated_text"])
print("-" * 30) |
This PR adds the Int8 support using the ZeroQuant technology introduced here . Note that the kernels added in this PR is just doing a dequantization of the int8-weight matrices and the GeMM operation is still done in the FP16 format using FP16 tensor-core ops. However, the kernels mentioned in this paper can directly operate on the int8 data, by using the int8 tensor-cores if available (such as on A100), which can improve the GeMM throughput by 2x. There will be another PR which uses these kernels to give a much faster inference through MII system.
Here is some performance evaluation:
FP16 on 8 A100-80G:
INT8 on 4 A100-80G:
INT8 on 8 A100-80G:
cc: @stas00 @jeffra