From 5bd5079de16f2a50ecc85635495ab4c04e7abf67 Mon Sep 17 00:00:00 2001 From: Surya Periaswamy Date: Thu, 9 Oct 2025 14:08:38 -0500 Subject: [PATCH] MSCCL++ fix split path null deref (#1959) * Add speriaswamy-amd to CODEOWNERS * MSCCL++: fix split path null deref; key maps by parent ncclUniqueId * removed no-op --- src/init.cc | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/init.cc b/src/init.cc index 3828b9b5c1..f684ea8248 100644 --- a/src/init.cc +++ b/src/init.cc @@ -2043,10 +2043,10 @@ static ncclResult_t ncclCommInitRankFunc(struct ncclAsyncJob* job_) { comm->mscclppCompatible = true; comm->mscclpp_threshold = job->parent->mscclpp_threshold; comm->mscclpp_comm = job->parent->mscclpp_comm; - auto& mscclppUniqueId = mscclpp_uniqueIdMap[*job->commId]; - mscclpp_uniqueIdMap[*job->commId] = mscclppUniqueId; - mscclpp_uniqueIdReverseMap[mscclppUniqueId].insert(*job->commId); - ncclCommToUniqueIdMap[comm] = *job->commId; + const ncclUniqueId& parentUniqueId = ncclCommToUniqueIdMap[job->parent]; + auto& mscclppUniqueId = mscclpp_uniqueIdMap[parentUniqueId]; + mscclpp_uniqueIdReverseMap[mscclppUniqueId].insert(parentUniqueId); + ncclCommToUniqueIdMap[comm] = parentUniqueId; } } else