产品
是否支持
Atlas 800I A2 推理产品/A200I A2 Box 异构组件
√
Atlas A2 训练系列产品
x
Atlas 200I/500 A2 推理产品
Atlas 推理系列产品
Atlas 训练系列产品
获取Cache的地址。
1 2
@property tensor_addrs() -> List[int]
无
1 2 3
... kv_cache = cache_manager.allocate_cache(cache_desc, cache_keys) print(kv_cache.tensor_addrs)
正常情况返回类型为Cache的地址。