AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update(DeviceProxy): Dynamically Determine Memory Allocation Size & Remove Compile-Time size Calculations (#48)

Browse Source 
* Update(DeviceProxy): Dynamically Determine Memory Allocation Size & Remove Compile-Time size Calculations

- Modified the Device proxy class to determine memory allocation size at runtime.
- Updated all classes that include the Device proxy to use dynamic memory allocation.
- Removed compile-time memory size calculations.
- Ensured the allocated number of backend queue data structures matches the number of RO device contexts.

[ROCm/rocshmem commit: eb5a38e806]
This commit is contained in:
Avinash Kethineedi
2025-02-24 15:11:46 -06:00
committed by GitHub
parent 95c4c0d428
commit 1831a1b33c
23 changed files with 302 additions and 118 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocshmem/src/reverse_offload/context_ro_device.cpp
+1 -1
View File
@@ -50,7 +50,7 @@ __host__ ROContext::ROContext(Backend *b, size_t block_id)
auto block_base{backend->block_handle_proxy_.get()};
block_handle = &block_base[block_id];
}
ro_net_win_id = block_id % backend->ro_window_proxy_->MAX_NUM_WINDOWS;
ro_net_win_id = block_id % backend->ro_window_proxy_->get_num_MPI_windows();
ipcImpl_.ipc_bases = b->ipcImpl.ipc_bases;
ipcImpl_.shm_size = b->ipcImpl.shm_size;