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

Remove device-side active-set-based broadcast API interface from rocSHMEM

Browse Source
This commit is contained in:
avinashkethineedi
2024-10-29 19:04:49 +00:00
parent 9a524046fe
commit e1ff06913c
2 changed files with 2 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/roc_shmem/roc_shmem.hpp
+2 -6
View File
@@ -870,16 +870,12 @@ __device__ ATTR_NO_INLINE void roc_shmem_threadfence_system();
*/
#define BROADCAST_API_GEN(T, TNAME) \
__device__ ATTR_NO_INLINE void roc_shmem_ctx_##TNAME##_wg_broadcast( \
roc_shmem_ctx_t ctx, T *dest, const T *source, int nelem, int pe_root, \
int pe_start, int log_pe_stride, int pe_size, \
long *p_sync); /* NOLINT */ \
roc_shmem_ctx_t ctx, roc_shmem_team_t team, T *dest, const T *source, \
int nelem, int pe_root); /* NOLINT */ \
__host__ void roc_shmem_ctx_##TNAME##_broadcast( \
roc_shmem_ctx_t ctx, T *dest, const T *source, int nelem, int pe_root, \
int pe_start, int log_pe_stride, int pe_size, \
long *p_sync); /* NOLINT */ \
__device__ ATTR_NO_INLINE void roc_shmem_ctx_##TNAME##_wg_broadcast( \
roc_shmem_ctx_t ctx, roc_shmem_team_t team, T *dest, const T *source, \
int nelem, int pe_root); /* NOLINT */ \
__host__ void roc_shmem_ctx_##TNAME##_broadcast( \
roc_shmem_ctx_t ctx, roc_shmem_team_t team, T *dest, const T *source, \
int nelem, int pe_root); /* NOLINT */