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)