asc.language.basic.set_atomic_min

asc.language.basic.set_atomic_min() None

原子操作函数,设置后续从VECOUT传输到GM的数据是否执行原子比较, 将待拷贝的内容和GM已有内容进行比较,将最小值写入GM。 可通过设置模板参数来设定不同的数据类型。

对应的 Ascend C 函数原型

template <typename T>
__aicore__ inline void SetAtomicMin();

参数说明

  • dtype: 原子 min 操作的数据类型,由 Python 前端指定。 - 支持类型:asc.float16、asc.float32、asc.int32、asc.half。

约束说明

使用完后,建议通过set_atomic_none关闭原子累加,以免影响后续相关指令功能。

调用示例

dtype = asc.float16
asc.set_atomic_min(dtype)