triton.language.atomic_and¶
- triton.language.atomic_and(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_and(...)
,而非atomic_and(x, ...)
。