Implement rocshmem_ptr in IPC conduit (#197)
* Implement `rocshmem_ptr` in IPC conduit * tests: add functional test for `rocshmem_ptr` - Add safety check for pointer access and condition check before printing results for `rocshmem_ptr` test - Use `rocshmem_put` to store `rocshmem_ptr` availability for data validation
Tá an tiomantas seo le fáil i:
tiomanta ag
GitHub
tuismitheoir
87f99e7ec6
tiomantas
526105d315
@@ -106,6 +106,10 @@ __device__ void IPCContext::quiet() {
|
||||
|
||||
__device__ void *IPCContext::shmem_ptr(const void *dest, int pe) {
|
||||
void *ret = nullptr;
|
||||
void *dst = const_cast<void *>(dest);
|
||||
uint64_t L_offset =
|
||||
reinterpret_cast<char *>(dst) - ipcImpl_.ipc_bases[my_pe];
|
||||
ret = ipcImpl_.ipc_bases[pe] + L_offset;
|
||||
return ret;
|
||||
}
|
||||
|
||||
|
||||
Tagairt in Eagrán Nua
Cuir bac ar úsáideoir