Updating MSCCLLang Examples (#462)

Co-authored-by: Caio Rocha <aiorocha@microsoft.com>
This commit is contained in:
Caio Rocha
2025-02-19 09:48:31 -08:00
committed by GitHub
parent 55789bc551
commit 8a564977e5
3 changed files with 11 additions and 11 deletions

View File

@@ -31,16 +31,16 @@ def send_recv(instances):
c = chunk(r, Buffer.input, 0)
c.put(
nghr,
"scratch",
Buffer.scratch,
1,
sendtb=0,
chan_type=ChannelType.port,
)
c.signal(nghr, "scratch", 1, sendtb=0, chan_type=ChannelType.port)
c.flush(nghr, "scratch", 1, sendtb=0, chan_type=ChannelType.port)
c.signal(nghr, Buffer.scratch, 1, sendtb=0, chan_type=ChannelType.port)
c.flush(nghr, Buffer.scratch, 1, sendtb=0, chan_type=ChannelType.port)
for r in range(size):
c = chunk(r, "scratch", 1)
c = chunk(r, Buffer.scratch, 1)
c.wait(1 - r, Buffer.input, 0, recvtb=0, chan_type=ChannelType.port)
c.copy(r, Buffer.output, 0, sendtb=0)