BlocksCacheKey
函数功能
PagedAttention场景下,构造BlocksCacheKey。
函数原型
BlocksCacheKey(prompt_cluster_id: int, model_id: int = 0)
参数说明
参数名称 |
数据类型 |
取值说明 |
---|---|---|
prompt_cluster_id |
int |
cache所在prompt节点的集群id。 |
model_id |
int |
cache关联的model_id。 |
调用示例
from llm_datadist import BlocksCacheKey blocks_cache_key = BlocksCacheKey(0, 0)
返回值
正常场景下无返回值。
传入数据类型错误情况下会抛出TypeError异常。
约束说明
无