triton.language.atomic_max

triton.language.atomic_max(pointer, val, mask=None, sem=None, scope=None, _semantic=None)

在 pointer 指定的内存位置执行原子最大值操作。

返回原子操作之前存储在 pointer 的数据。

参数:

• pointer (Block 的 dtype=triton.PointerDType) – 要操作的内存位置

• val (Block 的 dtype=pointer.dtype.element_ty) – 用于执行原子操作的值

• sem (字符串, 可选) – 指定操作的内存语义。可接受的值为“acquire”、“release”、“acq_rel”（代表“ACQUIRE_RELEASE”）和“relaxed”。如果未提供，函数默认使用“acq_rel”语义。

• scope (字符串, 可选) – 定义观察原子操作同步效果的线程范围。可接受的值为“gpu”（默认）、“cta”（合作线程数组，线程块）或“sys”（代表“SYSTEM”）。默认值为“gpu”。

此函数也可以作为成员函数在 tensor 上调用，如 x.atomic_max(...) 而非 atomic_max(x, ...)。