Implement rocshmem_pe_quiet() (#282)

Co-authored-by: Aurelien Bouteiller <aurelien.bouteiller@amd.com>
Αυτή η υποβολή περιλαμβάνεται σε:
Yiltan
2025-10-20 11:42:39 -04:00
υποβλήθηκε από GitHub
γονέας 6f74cdfd75
υποβολή c3eeae473b
14 αρχεία άλλαξαν με 79 προσθήκες και 2 διαγραφές
@@ -161,6 +161,11 @@ __device__ void ROContext::quiet() {
true, get_status_flag(), is_default_ctx);
}
__device__ void ROContext::pe_quiet(size_t pe) {
// TODO: Optimize
quiet();
}
__device__ void *ROContext::shmem_ptr(const void *dest, int pe) {
void *ret = nullptr;
int local_pe{-1};
@@ -61,6 +61,8 @@ class ROContext : 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();