下载
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助
昇腾小AI

create_cpu_cache

函数功能

创建cpu cache。

函数原型

create_cpu_cache(cache_desc: CacheDesc, addrs: List[int])

参数说明

参数名称

数据类型

取值说明

cache_desc

CacheDesc

cache的描述。

addrs

List[int]

cpu cache的地址。

调用示例

from llm_datadist import KvCache
cpu_cache = KvCache.create_cpu_cache(cpu_cache_desc, cpu_addrs) # cpu_addrs来自创建的cpu tensors

返回值

正常情况返回类型为KvCache的cpu_cache。

传入数据类型错误情况下会抛出TypeError异常。

传入参数为None,会抛出AttributeError异常。

约束说明

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词