github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-04-17 03:00:54 -04:00
Code Issues Packages Projects Releases Wiki Activity

fix(compiler): handle splice streams (streams originating on device, used both on device and on host) in SDFG for GPU scheduling.

Browse Source
This commit is contained in:
Antoniu Pop
2023-07-21 13:56:51 +00:00
committed by Antoniu Pop
parent 6b5baca515
commit acf5780ee3
4 changed files with 24 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
compilers/concrete-compiler/compiler/lib/Conversion/SDFGToStreamEmulator/SDFGToStreamEmulator.cpp
View File

@@ -337,6 +337,9 @@ struct LowerSDFGMakeStream
case SDFG::StreamKind::device_to_host:
t = TS_STREAM_TYPE_TOPO_TO_X86_LSAP;
break;
case SDFG::StreamKind::device_to_both:
t = TS_STREAM_TYPE_TOPO_TO_BOTH;
break;
}
auto sType = msOp->getResultTypes()[0].dyn_cast_or_null<SDFG::StreamType>();
assert(sType && "SDFG MakeStream operation should return a stream type");