pypto.Tensor.get_cache_policy#

产品支持情况#

  • Ascend 950PR/Ascend 950DT：支持

  • Atlas A3 训练系列产品/Atlas A3 推理系列产品：支持

  • Atlas A2 训练系列产品/Atlas A2 推理系列产品：支持

功能说明#

获取某个缓存策略是否被启用。

函数原型#

get_cache_policy(self, policy: CachePolicy) -> bool

参数说明#

参数名

输入/输出

说明

policy

输入

缓存策略类型。

返回值说明#

Cache策略是否被启用。

约束说明#

无。

调用示例#

t = pypto.tensor((16, 16), pypto.DT_FP32)
out = t.get_cache_policy(pypto.CachePolicy.PREFETCH)

结果示例如下：

输出数据out: False