microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-24 14:54:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

removing extra stream and destroying created ones

Browse Source
This commit is contained in:
Saeed Maleki
2023-04-02 02:07:41 +00:00
parent 4c6616e7b9
commit 1731911d00
2 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/init.cc
View File

@@ -163,6 +163,10 @@ mscclppResult_t mscclppCommDestroy(mscclppComm_t comm)
MSCCLPPCHECK(mscclppCudaHostFree(proxyState->triggerFifo));
MSCCLPPCHECK(mscclppCudaFree(proxyState->fifoHead));
MSCCLPPCHECK(mscclppCudaFree(proxyState->fifoTailDev));
if (proxyState->p2pStream)
CUDACHECK(cudaStreamDestroy(proxyState->p2pStream));
CUDACHECK(cudaStreamDestroy(proxyState->fifoStream));
free(proxyState);
}
}

1
src/proxy.cc
View File

@@ -39,7 +39,6 @@ struct proxyArgs
{
struct mscclppComm* comm;
struct mscclppProxyState* proxyState;
cudaStream_t stream;
};
static void readTrigger(mscclppTrigger* dst, mscclppTrigger* src)