| //===- Synchronization.h - OpenMP synchronization utilities ------- C++ -*-===// |
| // |
| // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. |
| // See https://llvm.org/LICENSE.txt for license information. |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| // |
| //===----------------------------------------------------------------------===// |
| // |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #ifndef OMPTARGET_DEVICERTL_SYNCHRONIZATION_H |
| #define OMPTARGET_DEVICERTL_SYNCHRONIZATION_H |
| |
| #include "Types.h" |
| |
| namespace _OMP { |
| |
| namespace synchronize { |
| |
| /// Initialize the synchronization machinery. Must be called by all threads. |
| void init(bool IsSPMD); |
| |
| /// Synchronize all threads in a warp identified by \p Mask. |
| void warp(LaneMaskTy Mask); |
| |
| /// Synchronize all threads in a block. |
| void threads(); |
| |
| } // namespace synchronize |
| |
| namespace fence { |
| |
| /// Memory fence with \p Ordering semantics for the team. |
| void team(int Ordering); |
| |
| /// Memory fence with \p Ordering semantics for the contention group. |
| void kernel(int Ordering); |
| |
| /// Memory fence with \p Ordering semantics for the system. |
| void system(int Ordering); |
| |
| } // namespace fence |
| |
| namespace atomic { |
| |
| /// Atomically read \p Addr with \p Ordering semantics. |
| uint32_t read(uint32_t *Addr, int Ordering); |
| |
| /// Atomically store \p V to \p Addr with \p Ordering semantics. |
| uint32_t store(uint32_t *Addr, uint32_t V, int Ordering); |
| |
| /// Atomically store \p V to \p Addr with \p Ordering semantics. |
| uint64_t store(uint64_t *Addr, uint64_t V, int Ordering); |
| |
| /// Atomically increment \p *Addr and wrap at \p V with \p Ordering semantics. |
| uint32_t inc(uint32_t *Addr, uint32_t V, int Ordering); |
| |
| /// Atomically add \p V to \p *Addr with \p Ordering semantics. |
| uint32_t add(uint32_t *Addr, uint32_t V, int Ordering); |
| |
| /// Atomically add \p V to \p *Addr with \p Ordering semantics. |
| uint64_t add(uint64_t *Addr, uint64_t V, int Ordering); |
| |
| } // namespace atomic |
| |
| } // namespace _OMP |
| |
| #endif |