AI/rocm-systems
2
0
Garpu 0
Kode Masalah Tarik Permintaan Actions Packages Projects Rilis Wiki Kegiatan

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

Telusuri Sumber 
[ROCm/rocshmem commit: e1ff06913c]
This commit is contained in:
avinashkethineedi
2024-10-29 19:04:49 +00:00
orang tua daae6f4d60
melakukan 058aa3ccf3
2 mengubah file dengan 2 tambahan dan 26 penghapusan
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocshmem/include/roc_shmem/roc_shmem.hpp
+2 -6
Melihat 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 */