Skip to content
版 本

asc_storeunalign_post

产 品 支 持 情 况

产 品是 否 支 持
Ascend 950PR/Ascend 950DT

功 能 说 明

reg计 算 数 据 搬 运 接 口,适 用 于 从 矢 量 数 据 寄 存 器 连 续 非32B对 齐 的 起 始 地 址 连 续 搬 出 到UB的 尾 块 场 景。

需 要 先 调 用 下 列 接 口 之 一 后,再 调 用 本 接 口。

搬 运 原 理 如 下: 记 目 的 操 作 数 的 起 始 地 址 为 dst_start,结 束 地 址 为 dst_end,尾 块 元 素 个 数 为 unalign_count = (dst_end - dst_end / 32 * 32) / sizeof(T)。 则 搬 运 的 数 据 会 分 为 两 部 分,分 别 是:

  • 32B对 齐 的 主 块 部 分UB[dst_start, dst_end - unalign_count * sizeof(T)]。
  • 非32B对 齐 的 尾 块 部 分UB[dst_end - unalign_count * sizeof(T), dst_end]。 store_unalign或store_unalign_postupdate接 口 执 行 时,会 将 主 块 搬 出 至UB,尾 块 暂 存 至 非 对 齐 寄 存 器[0, unalign_count]。 本 接 口 执 行 时,会 将 尾 块 从 非 对 齐 寄 存 器 搬 出 至UB。

函 数 原 型

  • 不 指 定 偏 移 量,配 合asc_storeunalign_postupdate使 用 不 指 定 存 储 偏 移 量 的 接 口。

    C++
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int8_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ uint8_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp4x2_e2m1_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp4x2_e1m2_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int4b_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp8_e8m0_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp8_e5m2_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp8_e4m3fn_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int16_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ uint16_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ half* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ bfloat16_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int32_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ uint32_t* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ float* dst, vector_store_unalign src)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int64_t* dst, vector_store_unalign src)
    
  • 使 用int32_t作 为 偏 移 量,配 合asc_storeunalign接 口。

    C++
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int8_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ uint8_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp4x2_e2m1_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp4x2_e1m2_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int4b_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp8_e8m0_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp8_e5m2_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp8_e4m3fn_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ hifloat8_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int16_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ uint16_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ half* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ bfloat16_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int32_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ uint32_t* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ float* dst, vector_store_unalign src, int32_t offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int64_t* dst, vector_store_unalign src, int32_t offset)
    
  • 使 用iter_reg作 为 偏 移 量,配 合asc_storeunalign_postupdate使 用iter_reg作 为 存 储 偏 移 量 的 接 口。

    C++
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int8_t* dst, vector_store_unalign src, iter_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ uint8_t* dst, vector_store_unalign src, iter_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp4x2_e2m1_t* dst, vector_store_unalign src, iter_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp4x2_e1m2_t* dst, vector_store_unalign src, iter_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int4b_t* dst, vector_store_unalign src, iter_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp8_e8m0_t* dst, vector_store_unalign src, iter_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp8_e5m2_t* dst, vector_store_unalign src, iter_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ fp8_e4m3fn_t* dst, vector_store_unalign src, iter_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int16_t* dst, vector_store_unalign src, iter_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ uint16_t* dst, vector_store_unalign src, iter_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ half* dst, vector_store_unalign src, iter_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ bfloat16_t* dst, vector_store_unalign src, iter_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int32_t* dst, vector_store_unalign src, iter_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ uint32_t* dst, vector_store_unalign src, iter_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ float* dst, vector_store_unalign src, iter_reg offset)
    __simd_callee__ inline void asc_storeunalign_post(__ubuf__ int64_t* dst, vector_store_unalign src, iter_reg offset)
    

参 数 说 明

  • 不 指 定 偏 移 量

    参 数 名输 入/输 出描 述
    dst输 出目 的 操 作 数(矢 量)的 起 始 地 址。
    src输 入非 对 齐 寄 存 器,用 于 保 存 非 对 齐 数 据,长 度32B。
  • 使 用int32_t作 为 存 储 偏 移 量

    参 数 名输 入/输 出描 述
    dst输 出目 的 操 作 数(矢 量)的 起 始 地 址。
    src输 入非 对 齐 寄 存 器,用 于 保 存 非 对 齐 数 据,长 度32B。
    offset输 入目 的 操 作 数 结 束 地 址 的 偏 移,类 型 为int32_t,单 位 为 元 素 个 数。
  • 使 用iter_reg作 为 存 储 偏 移 量

    参 数 名输 入/输 出描 述
    dst输 出目 的 操 作 数(矢 量)的 起 始 地 址。
    src输 入非 对 齐 寄 存 器,用 于 保 存 非 对 齐 数 据,长 度32B。
    offset输 入目 的 操 作 数 结 束 地 址 的 偏 移,类 型 为 地 址 寄 存 器,单 位 为 元 素 个 数。

非 对 齐 寄 存 器 和 地 址 寄 存 器 的 详 细 说 明 请 参 见reg数 据 类 型 定 义.md

返 回 值 说 明

流 水 类 型

PIPE_V

约 束 说 明

  • 该 接 口 中 的dst不 需 要32B对 齐,但 数 据 类 型 为T的dst需 要sizeof(T) Byte对 齐。
  • 调 用 本 接 口 之 前,需 要 调 用asc_storeunalign接 口,或 者asc_storeunalign_postupdate使 用iter_reg作 为 存 储 偏 移 量 的 接 口,且 本 接 口 与 前 序 接 口 的 非 对 齐 寄 存 器 要 保 持 一 致。
  • 需 要 保 证 目 的 操 作 数 的 地 址 加 上offset对 应 的 偏 移 地 址,结 果 等 于 数 据 搬 运 的 结 束 地 址,具 体 见 调 用 示 例。

调 用 示 例

  • 不 指 定 偏 移 量

    C++
    // dst地 址 为8,非32B对 齐。范 围 为[8:520]。
    __ubuf__ uint32_t* dst = (__ubuf__ uint32_t*)asc_get_phy_buf_addr(8);
    vector_store_unalign ureg;
    vector_uint32_t src;
    asc_storeunalign_postupdate(dst, ureg, src);
    asc_storeunalign_post(dst, ureg);
    
  • 使 用int32_t作 为 存 储 偏 移 量

    C++
    // dst地 址 为8,非32B对 齐,占 用 的 地 址 为UB[8:520]。
    __ubuf__ uint32_t* dst = (__ubuf__ uint32_t*)asc_get_phy_buf_addr(8);
    vector_store_unalign ureg;
    vector_uint32_t src;
    uint32_t count = 64;
    uint32_t repeat = 2;
    for (uint32_t i = 0; i < repeat; i++) {
        // 其 他reg操 作
    
        // 第 一 次:src[0:62] 写 入 dst[0:62](即UB[8:256]);同 时src[62:64] 写 入 ureg[0:2]
        // 第 二 次:ureg[0:2] + src[0:62] 写 入 dst[62:126](即UB[256:512]);同 时src[62:64] 写 入 ureg[0:2]
        asc_storeunalign(dst + i*count, ureg, src, count);
    }
    // dst结 束 地 址 为520,即dst + repeat * count * sizeof(uint32_t),所 以offset要 配 置 为repeat * count
    int32_t offset = repeat * count;
    // ureg[0:2] 写 入 dst[126:128](即UB[512:520])
    asc_storeunalign_post(dst, ureg, offset);
    
  • 使 用iter_reg作 为 存 储 偏 移 量

    C++
    // dst地 址 为8,非32B对 齐。占 用 的 地 址 为[8:520]。
    __ubuf__ uint32_t* dst = (__ubuf__ uint32_t*)asc_get_phy_buf_addr(8);
    vector_store_unalign ureg;
    vector_uint32_t src;
    iter_reg count = asc_create_iter_reg_b32(64);
    uint32_t repeat = 2;
    for (uint32_t i = 0; i < repeat; i++) {
        // 其 他reg操 作
    
        // 第 一 次:src[0:62] 写 入 dst[0:62](即UB[8:256]);同 时src[62:64] 写 入 ureg[0:2];同 时dst自 增 为dst[64](即UB[264])
        // 第 二 次:ureg[0:2] + src[0:62] 写 入 dst[-2:62](即UB[256:512]);同 时src[62:64] 写 入 ureg[0:2];同 时dst自 增 为dst[128](即UB[520])
        asc_storeunalign_postupdate(dst, ureg, src, count);
    }
    // dst结 束 地 址 为520,但 此 时dst已 经 自 增 到 了520,所 以offset要 配 置 为0
    iter_reg offset = asc_create_iter_reg_b32(0);
    // ureg[0:2] 写 入 dst[-2:0](即UB[512:520])
    asc_storeunalign_post(dst, ureg, offset);
    

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