changes for hugepages backed host buffer for larger allocations (#1841)

[ROCm/rccl commit: 65b69bf318]
2025-09-28 11:10:22 +05:30
parent 722b0cd579
commit 3169352cad
3 changed files with 87 additions and 18 deletions
@@ -95,6 +95,8 @@ NCCL_PARAM(NvlsChannels, "NVLS_NCHANNELS", NCCL_CONFIG_UNDEF_INT);

 struct allocationTracker allocTracker[MAX_ALLOC_TRACK_NGPU] = {};
 static ncclResult_t commReclaim(ncclComm_t comm);
+std::unordered_map<void*, size_t> hugepageAllocs;
+std::mutex hugepageAllocsMutex;

 #ifdef ENABLE_MSCCLPP
 size_t std::hash<ncclUniqueId>::operator ()(const ncclUniqueId& uniqueId) const noexcept {