Magic-Mirror/llvm-capstone
1
0
Fork 0
mirror of https://github.com/capstone-engine/llvm-capstone.git synced 2025-01-26 19:36:28 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[libomptarget] Implement most hip atomic functions in terms of intrinsics

Browse Source 
Summary:
[libomptarget] Implement hip atomic functions in terms of intrinsics

All but atomicInc can be implemented using type generic clang intrinsics.
There is not yet a corresponding intrinsic for atomicInc in clang, only one in
LLVM. This patch leaves atomicInc as an unresolved symbol.

Reviewers: jdoerfert, ABataev, hfinkel, grokos, arsenm

Reviewed By: arsenm

Subscribers: sri, saiislam, wdng, jvesely, mgorny, jfb, openmp-commits

Tags: #openmp

Differential Revision: https://reviews.llvm.org/D73076
This commit is contained in:
Jon Chesterfield 2020-03-04 17:51:38 +00:00 committed by JonChesterfield
parent d09d0bd7a0
commit 133db44996
1 changed files with 21 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
openmp/libomptarget/deviceRTLs/amdgcn/src/hip_atomics.h
View File

@ -11,29 +11,32 @@
#include "target_impl.h"
DEVICE unsigned atomicAdd(unsigned *address, unsigned val);
DEVICE int atomicAdd(int *address, int val);
DEVICE unsigned long long atomicAdd(unsigned long long *address,
unsigned long long val);
// inc requires an amdgcn specific intrinsic which is not yet available
DEVICE unsigned atomicInc(unsigned *address);
DEVICE unsigned atomicInc(unsigned *address, unsigned max);
DEVICE int atomicInc(int *address);
DEVICE int atomicMax(int *address, int val);
DEVICE unsigned atomicMax(unsigned *address, unsigned val);
DEVICE unsigned long long atomicMax(unsigned long long *address,
unsigned long long val);
namespace {
DEVICE int atomicExch(int *address, int val);
DEVICE unsigned atomicExch(unsigned *address, unsigned val);
DEVICE unsigned long long atomicExch(unsigned long long *address,
unsigned long long val);
template <typename T> DEVICE T atomicAdd(T *address, T val) {
return __atomic_fetch_add(address, val, __ATOMIC_SEQ_CST);
}
DEVICE unsigned atomicCAS(unsigned *address, unsigned compare, unsigned val);
DEVICE int atomicCAS(int *address, int compare, int val);
DEVICE unsigned long long atomicCAS(unsigned long long *address,
unsigned long long compare,
unsigned long long val);
template <typename T> DEVICE T atomicMax(T *address, T val) {
return __atomic_fetch_max(address, val, __ATOMIC_SEQ_CST);
}
template <typename T> DEVICE T atomicExch(T *address, T val) {
T r;
__atomic_exchange(address, &val, &r, __ATOMIC_SEQ_CST);
return r;
}
template <typename T> DEVICE T atomicCAS(T *address, T compare, T val) {
(void)__atomic_compare_exchange(address, &compare, &val, false,
__ATOMIC_SEQ_CST, __ATOMIC_RELAXED);
return compare;
}
} // namespace
#endif