Implement rocshmem_pe_quiet() (#282)

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
vanhempi 6bc1cc63ae
commit 92a7904656
14 muutettua tiedostoa jossa 79 lisäystä ja 2 poistoa
@@ -177,6 +177,10 @@ __device__ void GDAContext::quiet() {
}
}
__device__ void GDAContext::pe_quiet(size_t pe) {
qps[pe].quiet();
}
__device__ void *GDAContext::shmem_ptr(const void *dest, int pe) {
void *ret = nullptr;
int local_pe{-1};
@@ -60,6 +60,8 @@ class GDAContext : public Context {
__device__ void quiet();
__device__ void pe_quiet(size_t pe);
__device__ void *shmem_ptr(const void *dest, int pe);
__device__ void barrier_all();