Skip to content

asc_get_scalar_load_cache_mode

产品支持情况

  • Ascend 950PR/Ascend 950DT:支持
  • Atlas A3 训练系列产品/Atlas A3 推理系列产品:不支持
  • Atlas A2 训练系列产品/Atlas A2 推理系列产品:不支持
  • Atlas 200I/500 A2 推理产品:不支持
  • Atlas 推理系列产品AI Core:不支持
  • Atlas 推理系列产品Vector Core:不支持
  • Atlas 训练系列产品:不支持

功能说明

头文件路径为:"c_api/cache_ctrl/cache_ctrl.h"

读取当前标量读(load)路径的L2 Cache管理策略,并返回asc_load_l2_cache_mode枚举值。

常用于临时切换策略前保存原配置,或在设置后做回读校验。

函数原型

C
__aicore__ inline asc_load_l2_cache_mode asc_get_scalar_load_cache_mode()

参数说明

返回值说明

当前标量load路径的L2 Cache管理策略,类型为asc_load_l2_cache_mode

流水类型

PIPE_S

约束说明

  • 本接口仅反映标量load侧策略,与标量store侧以及向量搬运接口参数中的l2_cache_mode无关。

调用示例

场景:GM tiling表在tile循环内被多次标量读取(每tile读lengthoffset);进入循环前查询并保存当前load策略,临时切换为NORMAL_FIRST_VICTIM,循环结束后恢复。若tiling只读一次,可改用NOTALLOC_KEEP或无需切换。

C++
__gm__ int32_t* tiling_gm;   // [tile_count, len[0..n-1], off[0..n-1]]
__gm__ half* src_gm;
__ubuf__ half* dst_ub;

// 查询并保存当前标量load策略
asc_load_l2_cache_mode old_mode = asc_get_scalar_load_cache_mode();
asc_set_scalar_cache_mode(asc_load_l2_cache_mode::NORMAL_FIRST_VICTIM);

int32_t tile_count = tiling_gm[0];
for (int32_t tile = 0; tile < tile_count; ++tile) {
    int32_t tile_len = tiling_gm[1 + tile];              // 每tile标量读GM
    int32_t tile_off = tiling_gm[1 + tile_count + tile]; // 每tile标量读GM
    asc_copy_gm2ub_align(dst_ub, src_gm + tile_off, tile_len * sizeof(half));
    // ... 向量计算 ...
}

// 恢复进入本段逻辑前的标量load策略
asc_set_scalar_cache_mode(old_mode);

免责声明:本站内容由 asc-devkit 仓 master 分支自动编译生成,属于持续开发版本,可能存在缺陷,仅供预览与参考。如需稳定及商用资料,请查阅官方 昇腾社区