[Profiler plugin] Fix segfault issue with profiler plugin (#1973)
* Fix profiler plugin segfault by correctly setting p2p->func
* Look for librccl-profiler.so instead of libnccl-profiler.so
Signed-off-by: rahulvaidya20 <ravaidya@amd.com>
---------
Signed-off-by: rahulvaidya20 <ravaidya@amd.com>
Co-authored-by: Yongjie Qiu <Yongjie.Qiu@amd.com>
[ROCm/rccl commit: 624f68b2b2]
Ta zmiana jest zawarta w:
1 rodzic
a5256e6219
złożenie
307f787244
4 plików zmienionych
+11
-10
Żadne pliki nie pasują do Twojego wyszukiwania
@@ -2574,6 +2574,7 @@ static ncclResult_t taskAppend(struct ncclComm* comm, struct ncclInfo* info) {
|
||||
// Must be in thread local group before tasks can be alloc'd in `comm->memScoped`.
|
||||
ncclGroupCommJoin(info->comm, ncclGroupTaskTypeCollective);
|
||||
struct ncclTaskP2p* p2p = ncclMemoryPoolAlloc<struct ncclTaskP2p>(&comm->memPool_ncclTaskP2p, &comm->memPermanent);
|
||||
p2p->func = info->coll;
|
||||
p2p->buff = (void*)info->recvbuff;
|
||||
p2p->count = info->count;
|
||||
p2p->datatype = info->datatype;
|
||||
|
||||
Reference in new issue
Zablokuj użytkownika