Struct Atomic 

pub struct Atomic<'a, T: ?Sized> { /* private fields */ }

To prevent the user from mixing use of Atomic operation and chunk-based access, we wrap the reference to the data to be modified in an Atomic struct. ensuring that the user cannot access the data without using atomic operations. If user wants to repurpose the data for non-atomic access, they need to drop the Atomic struct first and need a sync across all blocks. For now, sync across all blocks is not supported, we should reject such code in analysis. TODO: Avoid repurposing data for atomic or chunk-based access.

Implementations

impl<'a, T> Atomic<'a, [T]>

pub fn index(&'a self, i: usize) -> Atomic<'a, T>

Get a reference to the data inside the Atomic struct. This is unsafe because the user must ensure that no other thread is accessing the data at the same time.

impl<'a, T: ?Sized> Atomic<'a, T>

Atomic read-modify-write operation with kind defined in MLIR memref::atomic_rmw

pub fn new(data: GpuGlobal<'a, T>) -> Atomic<'a, T>

pub fn atomic_rmw<K: AtomicRMWKind>(&self, val: T) -> T

where T: Num,

impl<'a, T: Num> Atomic<'a, T>

pub fn atomic_addf(&self, val: T) -> T

where T: Float,

Equivalent to: atomic_rmw::<[Atomic<'a, T>::AddF]>

impl<'a, T: Num> Atomic<'a, T>

pub fn atomic_addi(&self, val: T) -> T

where T: PrimInt,

Equivalent to: atomic_rmw::<[Atomic<'a, T>::AddI]>

impl<'a, T: Num> Atomic<'a, T>

pub fn atomic_assign(&self, val: T) -> T

where T: Num,

Equivalent to: atomic_rmw::<[Atomic<'a, T>::Assign]>

impl<'a, T: Num> Atomic<'a, T>

pub fn atomic_maximumf(&self, val: T) -> T

where T: Float,

Equivalent to: atomic_rmw::<[Atomic<'a, T>::MaximumF]>

impl<'a, T: Num> Atomic<'a, T>

pub fn atomic_maxs(&self, val: T) -> T

where T: Signed,

Equivalent to: atomic_rmw::<[Atomic<'a, T>::MaxS]>

impl<'a, T: Num> Atomic<'a, T>

pub fn atomic_maxu(&self, val: T) -> T

where T: Unsigned,

Equivalent to: atomic_rmw::<[Atomic<'a, T>::MaxU]>

impl<'a, T: Num> Atomic<'a, T>

pub fn atomic_minimumf(&self, val: T) -> T

where T: Float,

Equivalent to: atomic_rmw::<[Atomic<'a, T>::MinimumF]>

impl<'a, T: Num> Atomic<'a, T>

pub fn atomic_mins(&self, val: T) -> T

where T: Signed,

Equivalent to: atomic_rmw::<[Atomic<'a, T>::MinS]>

impl<'a, T: Num> Atomic<'a, T>

pub fn atomic_minu(&self, val: T) -> T

where T: Unsigned,

Equivalent to: atomic_rmw::<[Atomic<'a, T>::MinU]>

impl<'a, T: Num> Atomic<'a, T>

pub fn atomic_mulf(&self, val: T) -> T

where T: Float,

Equivalent to: atomic_rmw::<[Atomic<'a, T>::MulF]>

impl<'a, T: Num> Atomic<'a, T>

pub fn atomic_muli(&self, val: T) -> T

where T: PrimInt,

Equivalent to: atomic_rmw::<[Atomic<'a, T>::MulI]>

impl<'a, T: Num> Atomic<'a, T>

pub fn atomic_ori(&self, val: T) -> T

where T: PrimInt,

Equivalent to: atomic_rmw::<[Atomic<'a, T>::OrI]>

impl<'a, T: Num> Atomic<'a, T>

pub fn atomic_andi(&self, val: T) -> T

where T: PrimInt,

Equivalent to: atomic_rmw::<[Atomic<'a, T>::AndI]>

impl<'a, T: Num> Atomic<'a, T>

pub fn atomic_maxnumf(&self, val: T) -> T

where T: Float,

Equivalent to: atomic_rmw::<[Atomic<'a, T>::MaxNumF]>

impl<'a, T: Num> Atomic<'a, T>

pub fn atomic_minnumf(&self, val: T) -> T

where T: Float,

Equivalent to: atomic_rmw::<[Atomic<'a, T>::MinNumF]>