下载
中文
注册

LayerSynchronizer

抽象类,需要由用户继承该抽象类实现相关接口。当前该类下只有一个接口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调用。

返回值

是否同步成功

约束说明