Implement rocshmem_pe_quiet() (#282)
Co-authored-by: Aurelien Bouteiller <aurelien.bouteiller@amd.com>
Αυτή η υποβολή περιλαμβάνεται σε:
@@ -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();
|
||||
|
||||
Αναφορά σε νέο ζήτημα
Block a user