AI/rocm-systems
2
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

updated atomic_fetch() parameters

Browse Source 
[ROCm/rocshmem commit: 8b3854b252]
This commit is contained in:
Yiltan Hassan Temucin committed 2024-10-11 13:34:28 -07:00
1 parent 3d0fca0387
commit 31fe937259
5 files changed
+25 -25
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

projects/rocshmem/include/roc_shmem/roc_shmem.hpp
+4 -4
View File
@@ -1021,12 +1021,12 @@ __device__ ATTR_NO_INLINE void roc_shmem_threadfence_system();
*/
#define ATOMIC_FETCH_API_GEN(T, TNAME) \
__device__ ATTR_NO_INLINE T roc_shmem_ctx_##TNAME##_atomic_fetch( \
roc_shmem_ctx_t ctx, T *dest, int pe); \
__device__ ATTR_NO_INLINE T roc_shmem_##TNAME##_atomic_fetch(T *dest, \
roc_shmem_ctx_t ctx, T *source, int pe); \
__device__ ATTR_NO_INLINE T roc_shmem_##TNAME##_atomic_fetch(T *source, \
int pe); \
__host__ T roc_shmem_ctx_##TNAME##_atomic_fetch(roc_shmem_ctx_t ctx, \
T *dest, int pe); \
__host__ T roc_shmem_##TNAME##_atomic_fetch(T *dest, int pe);
T *source, int pe); \
__host__ T roc_shmem_##TNAME##_atomic_fetch(T *source, int pe);
/*
* MACRO DECLARE SHMEM_ATOMIC_ADD APIs