ipex-llm/python/llm/example/GPU/Speculative-Decoding
Jean Yu ab476c7fe2
Eagle Speculative Sampling examples (#11104)
* Eagle Speculative Sampling examples

* rm multi-gpu and ray content

* updated README to include Arc A770
2024-05-24 11:13:43 -07:00
..
baichuan2 Upgrade to python 3.11 (#10711) 2024-04-09 17:41:17 +08:00
chatglm3 Upgrade to python 3.11 (#10711) 2024-04-09 17:41:17 +08:00
Eagle Eagle Speculative Sampling examples (#11104) 2024-05-24 11:13:43 -07:00
gpt-j Upgrade to python 3.11 (#10711) 2024-04-09 17:41:17 +08:00
llama2 Add lookahead GPU example (#10785) 2024-04-17 17:41:55 +08:00
mistral Upgrade to python 3.11 (#10711) 2024-04-09 17:41:17 +08:00
qwen Upgrade to python 3.11 (#10711) 2024-04-09 17:41:17 +08:00
README.md Update_document by heyang (#30) 2024-03-25 10:06:02 +08:00

Self-Speculative Decoding for Large Language Model FP16 Inference using IPEX-LLM on Intel GPUs

You can use IPEX-LLM to run FP16 inference for any Huggingface Transformer model with self-speculative decoding on Intel GPUs. This directory contains example scripts to help you quickly get started to run some popular open-source models using self-speculative decoding. Each model has its own dedicated folder, where you can find detailed instructions on how to install and run it.

Verified Hardware Platforms

  • Intel Data Center GPU Max Series

To apply Intel GPU acceleration, therere several steps for tools installation and environment preparation. See the GPU installation guide for mode details.

Step 1, only Linux system is supported now, Ubuntu 22.04 is prefered.

Step 2, please refer to our driver installation for general purpose GPU capabilities.

Note

: IPEX 2.1.10+xpu requires Intel GPU Driver version >= stable_775_20_20231219.

Step 3, you also need to download and install Intel® oneAPI Base Toolkit. OneMKL and DPC++ compiler are needed, others are optional.

Note

: IPEX 2.1.10+xpu requires Intel® oneAPI Base Toolkit's version == 2024.0.

Best Known Configuration on Linux

For optimal performance on Intel Data Center GPU Max Series, it is recommended to set several environment variables.

export LD_PRELOAD=${LD_PRELOAD}:${CONDA_PREFIX}/lib/libtcmalloc.so
export SYCL_PI_LEVEL_ZERO_USE_IMMEDIATE_COMMANDLISTS=1
export ENABLE_SDP_FUSION=1