Files
zhongtianyun 3c51eda5d6 [Embedding] Fix readme and env for contrastive_training (#11108)
modified:   csrc/gpu/unittest/test_get_padding_offset_v2.py
	modified:   paddlenlp/experimental/transformers/llama/modeling.py
	modified:   slm/pipelines/examples/contrastive_training/README.md
	modified:   slm/pipelines/examples/contrastive_training/evaluation/eval_mteb.py
	modified:   slm/pipelines/examples/contrastive_training/evaluation/eval_mteb.sh
	modified:   slm/pipelines/examples/contrastive_training/evaluation/modelling_quant.py
	modified:   slm/pipelines/examples/contrastive_training/shortgpt_prune.py
2025-09-29 10:53:53 +08:00
..
2025-03-20 16:12:18 +08:00
2025-05-14 15:17:37 +08:00
2025-03-05 19:05:51 +08:00
2025-03-07 00:54:58 -08:00

PaddleNLP 大模型高性能自定义推理算子

此文档介绍如何编译安装 PaddleNLP 大模型高性能自定义推理算子的安装教程。

使用这些高性能算子,可以大幅提升大模型推理速度。 大模型推理相关教程详见此处。

安装 C++ 依赖

pip install -r requirements.txt

编译 Cuda 算子

python setup_cuda.py install

FP8 GEMM 自动调优

确保 cutlass 库已经安装,然后执行以下命令进行自动调优。

  • 对于89架构的 GPU,CUDA 版本至少为12.4
  • 对于90架构的 GPU,CUDA 版本至少为12.0
cd tools
sh tune_fp8_gemm.sh