快速入门【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge整体开发流程简介为了方便后续的描述我们对于不同的推理框架抽象为几个模块资源初始化模块。KV Cache管理模块。包括KV Cache的内存的创建分配PA场景以及销毁。模型推理模块。资源释放模块。本章节主要是介绍开发者如何在推理框架中使能LLM-DataDist的能力。开发流程找到推理框架中的资源初始化模块在该阶段中调用LLM-DataDist的初始化接口和建链接口。找到推理框架中的KV Cache管理模块在该阶段中调用LLM-DataDist的KV Cache申请接口将申请好的KV Cache转换为不同框架的KV Cache类型进行推理。推理框架需要拆分出Prefill阶段和Decode阶段对推理脚本进行分离部署部署到不同的集群节点上。在Decode阶段执行前需要接收来自Prefill阶段的输出作为输入同时调用LLM-DataDist提供的KV Cache传输接口拉取或推送Prefill侧缓存的KV Cache。分别执行Prefill推理脚本和Decode推理脚本。在框架资源释放模块释放LLM-DataDist相关资源。环境准备支持的产品形态如下Atlas A2 训练系列产品/Atlas A2 推理系列产品针对Atlas A2 训练系列产品/Atlas A2 推理系列产品仅支持Atlas 800I A2 推理服务器、A200I A2 Box 异构组件。Atlas A3 训练系列产品/Atlas A3 推理系列产品请参考《CANN 软件安装》安装好驱动固件以及CANN软件。使用hccn_tool查询Device IP并且进行卡间网络检测要求各个集群上的卡间有RDMA链路连接否则无法使能LLM-DataDist能力。hccn_tool详细介绍请参考《HCCN Tool 接口参考》。以下是常用命令参考。|命令|使用场景| |--|--| |hccn_tool [-i %d] -link -g|获取指定Device网口Link状态。-i指定Device。样例hccn_tool -i 0 -link -g| |hccn_tool [-i %d] -ip -g|获取IP地址和子网掩码。-i指定Device。样例hccn_tool -i 0 -ip -g| |hccn_tool [-i %d] -ping -g [address %s ]|获取指定设备到目的地址的ping结果。-i指定当前server的某个Device address指定ping的目的地址。样例hccn_tool -i 0 -ping -g address 192.168.2.1|使用LLM-DataDist过程中还涉及到如下环境变量具体请参见《环境变量参考》。|名称|使用场景| |--|--| |HCCL_RDMA_TC、HCCL_RDMA_SL|当客户对参数面网络做了自己的规划时对各种业务流量规定了类型优先级。通过这两个环境变量设置参数面集合通信流量在网络上的流量类型和优先级以适配客户网络流量规划的要求。| |HCCL_RDMA_RETRY_CNT、HCCL_RDMA_TIMEOUT|分别对应RDMA网卡的重试次数和重传超时时间的系数timeout。设置太大导致对网络异常反应不敏感不能感知到网络故障。设置太小则容易造成网络闪断直接造成业务中断不能被网卡硬件屏蔽。用户可根据自身网络情况来设置合适的值。例如可以根据大部分闪断的时间范围进行配置。推荐按照如下公式进行配置以减少网络抖动带来的影响。HCCL_RDMA_TIMEOUTlog2(pull kv超时时间 * 10^6 / (HCCL_RDMA_RETRY_CNT 1) / 4.096)向上取整。当pull kv超时时间和HCCL_RDMA_RETRY_CNT都等于默认值时HCCL_RDMA_TIMEOUT建议配置成15。| |HCCL_INTRA_ROCE_ENABLE|用于配置Server内是否使用RoCE环路进行多卡间的通信。|完整样例参考本示例通过LLM-DataDist接口实现分离部署场景下KV Cache管理功能。为不同框架使用LLM-DataDist的功能提供一些参考和改造思路。快速体验请单击gitcode选择对应分支在“examples/dflow/cpp”目录中获取样例。【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考
