Implemented workgroup puts (#238)

[ROCm/rocshmem commit: 58f96af7ec]
This commit is contained in:
Yiltan
2025-09-08 10:57:39 -04:00
gecommit door GitHub
bovenliggende fb7d89c7e6
commit 9b8404693c
2 gewijzigde bestanden met toevoegingen van 11 en 10 verwijderingen
@@ -130,9 +130,10 @@ __device__ void *GDAContext::shmem_ptr(const void *dest, int pe) {
__device__ void GDAContext::putmem_wg(void *dest, const void *source,
size_t nelems, int pe) {
uint64_t L_offset = reinterpret_cast<char*>(dest) - base_heap[my_pe];
if (is_thread_zero_in_block()) {
printf("rocshmem::gda:putmem_wg not implemented\n");
abort();
qps[pe].put_nbi(base_heap[pe] + L_offset, source, nelems, pe);
qps[pe].quiet();
}
}
@@ -146,9 +147,9 @@ __device__ void GDAContext::getmem_wg(void *dest, const void *source,
__device__ void GDAContext::putmem_nbi_wg(void *dest, const void *source,
size_t nelems, int pe) {
uint64_t L_offset = reinterpret_cast<char*>(dest) - base_heap[my_pe];
if (is_thread_zero_in_block()) {
printf("rocshmem::gda:putmem_nbi_wg not implemented\n");
abort();
qps[pe].put_nbi(base_heap[pe] + L_offset, source, nelems, pe);
}
}