microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-13 09:46:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

C++ API changes

Browse Source
This commit is contained in:
Olli Saarikivi
2023-04-19 22:02:23 +00:00
parent 83c7ba1afb
commit 9fbb0debdd
6 changed files with 160 additions and 136 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
src/proxy_cpp.cc
View File

@@ -62,10 +62,14 @@ MSCCLPP_API_CPP void Proxy::start() {
// Flush the tail to device memory. This is either triggered every ProxyFlushPeriod to make sure
// that the fifo can make progress even if there is no request mscclppSync. However, mscclppSync type is for flush
// request.
if ((++flushCnt % ProxyFlushPeriod) == 0 || result == ProxyHandlerResult::FlushAndContinue) {
if ((++flushCnt % ProxyFlushPeriod) == 0 || result == ProxyHandlerResult::FlushFifoTailAndContinue) {
// TODO: relocate this check: || (trigger.fields.type & mscclppSync)
fifo.flushTail();
}
if (result == ProxyHandlerResult::Stop) {
break;
}
}
// make sure the tail is flushed before we shut the proxy