Implement rocshmem_pe_quiet() (#282)
Co-authored-by: Aurelien Bouteiller <aurelien.bouteiller@amd.com>
[ROCm/rocshmem commit: c3eeae473b]
This commit is contained in:
@@ -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();
|
||||
|
||||
Viittaa uudesa ongelmassa
Block a user