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
Tento commit je obsažen v:
Avinash Kethineedi
2025-07-28 12:01:02 -05:00
odevzdal GitHub
rodič 87f99e7ec6
revize 526105d315
7 změnil soubory, kde provedl 122 přidání a 23 odebrání
+4
Zobrazit soubor
@@ -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;
}