Implement rocshmem_pe_quiet() (#282)

Co-authored-by: Aurelien Bouteiller <aurelien.bouteiller@amd.com>
This commit is contained in:
Yiltan
2025-10-20 11:42:39 -04:00
zatwierdzone przez GitHub
rodzic 6f74cdfd75
commit c3eeae473b
14 zmienionych plików z 79 dodań i 2 usunięć
+15
Wyświetl plik
@@ -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.