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

Implement rocshmem_pe_quiet() (#282)

Browse Source 
Co-authored-by: Aurelien Bouteiller <aurelien.bouteiller@amd.com>

[ROCm/rocshmem commit: c3eeae473b]
This commit is contained in:
Yiltan
2025-10-20 11:42:39 -04:00
committed by GitHub
parent 6bc1cc63ae
commit 92a7904656
14 changed files with 79 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocshmem/docs/api/memory_ordering.rst
+15
View File
@@ -34,3 +34,18 @@ ROCSHMEM_QUIET
**Description:**
This routine completes all previous operations posted to this context.
ROCSHMEM_PE_QUIET
--------------
.. cpp:function:: __device__ void rocshmem_ctx_pe_quiet(shmem_ctx_t ctx, const int *target_pes, size_t npes)
.. cpp:function:: __device__ void rocshmem_pe_quiet(const int *target_pes, size_t npes)
:param ctx: Context with which to perform this operation.
:param target_pes: Address of target PE array where the operations need to be completed
:param npes: The number of PEs in the target PE array
:returns: None.
**Description:**
This routine completes all previous operations posted to this context
for the PEs in the `target_pes` array.