接口列表
- LLM-DataDist:大模型分布式集群和数据加速组件,提供了集群KV数据管理能力,以支持全量图和增量图分离部署。
- LLM-DataDist相关接口存放在:"${INSTALL_DIR}/include/llm_datadist/llm_datadist.h"。${INSTALL_DIR}请替换为CANN软件安装后文件存储路径。若安装的Ascend-cann-toolkit软件包,以root安装举例,则安装后文件存储路径为:/usr/local/Ascend/ascend-toolkit/latest。
- LLM-DataDist接口对应的库文件是:libllm_datadist.so。
- ADXL:Ascend Device Transfer Library,提供高性能、零拷贝的点对点数据传输的能力,并通过简易API开放给用户。
- ADXL相关接口存放在:"${INSTALL_DIR}/include/adxl/adxl_engine.h"。${INSTALL_DIR}请替换为CANN软件安装后文件存储路径。若安装的Ascend-cann-toolkit软件包,以root安装举例,则安装后文件存储路径为:/usr/local/Ascend/ascend-toolkit/latest。
- ADXL接口对应的库文件是:libllm_datadist.so。
支持的产品形态如下:
Atlas 800I A2 推理产品 /A200I A2 Box 异构组件Atlas A3 训练系列产品 /Atlas A3 推理系列产品 ,该场景下采用HCCS传输协议时,不支持Host内存作为远端Cache。
接口列表如下。
LLM-DataDist
接口名称 |
简介 |
---|---|
构造LLM-DataDist。 |
|
LLM-DataDist对象析构函数。 |
|
初始化LLM-DataDist。 |
|
释放LLM-DataDist。 |
|
设置当前LLM-DataDist的角色。 |
|
建链。 |
|
断链。 |
|
以连续内存方式拉取KV Cache。 |
|
以block列表的方式拉取KV Cache。 |
|
推送Cache到远端节点。 |
|
通过block列表的方式,推送Cache到远端节点。 |
|
注册本地KV Cache内存。 |
|
解除注册本地KV Cache内存。 |
ADXL
接口名称 |
简介 |
---|---|
创建AdxlEngine对象。 |
|
AdxlEngine对象析构函数。 |
|
初始化AdxlEngine,在调用其他接口前需要先调用该接口。 |
|
AdxlEngine资源清理函数。 |
|
注册内存地址。用于TransferSync调用指定本地内存地址和远端内存地址,TransferSync指定的地址可以为注册的地址的子集,其中本地内存地址需在当前AdxlEngine进行注册,远端内存地址需要在远端AdxlEngine进行注册。 |
|
解注册内存。 |
|
与远端AdxlEngine进行建链。 |
|
与远端AdxlEngine进行断链。 |
|
与远端AdxlEngine进行内存传输。 |