下载
EN
注册

synchronize_layer

函数功能

等待模型指定层执行完成,用户需要继承LayerSynchronizer并实现该接口。

该接口会在执行KvCacheManager.transfer_cache_async时被调用,当该接口返回成功,则开始当前层cache的传输。

函数原型

synchronize_layer(layer_index: int, timeout_in_millis: Optional[int]) -> bool

参数说明

参数名称

数据类型

取值说明

layer_index

int

层的下标

timeout_in_millis

Optional[int]

超时时间,暂不支持。

调用示例

该接口不由用户直接调用,而是作为回调由KvCacheManager调用。

返回值

是否同步成功

约束说明