Skip to main content

triton_language.atomic_and

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

在由 pointer 指定的内存位置执行原子逻辑与操作。

在原子操作前返回保存在 pointer 处的数据。

参数**:**

  • pointer (Block of dtype=triton.PointerDType) - 要操作的内存位置
  • val (pointer.dtype.element_ty 的块) - 执行原子操作所需的值
  • sem (str, optional) - 指定操作的内存语义。可接受的值包括「acquire」(获取)、「release」(释放)、「acq_rel」(获取释放)和「relaxed」(放松)。如果未提供,则函数默认使用「acq_rel」语义。
  • scope (str, optional) - 定义观察原子操作同步效果的线程范围。可接受的值包括「gpu」(默认值)、「cta」(协作线程数组,即线程块)或「sys」(代表 “SYSTEM”)。默认值为 「gpu」。

这个函数也可作为 tensor 的成员函数调用,使用 x.atomic_and(...) 而不是 atomic_and(x, ...)