Raptor sync wait

This commit is contained in:
ilgeco
2026-08-06 14:32:46 +02:00
parent a963009855
commit a39fdba366
48 changed files with 3357 additions and 96 deletions
+41 -2
View File
@@ -550,7 +550,8 @@ def SpatChannelSendOp : SpatOp<"channel_send", []> {
);
let assemblyFormat = [{
$input `channel` $channelId `from` $sourceCoreId `to` $targetCoreId attr-dict `:` type($input)
$input `channel` $channelId `from` $sourceCoreId `to` $targetCoreId
attr-dict `:` type($input)
}];
}
@@ -568,7 +569,45 @@ def SpatChannelReceiveOp : SpatOp<"channel_receive", []> {
);
let assemblyFormat = [{
`channel` $channelId `from` $sourceCoreId `to` $targetCoreId attr-dict `:` type($output)
`channel` $channelId `from` $sourceCoreId `to` $targetCoreId
attr-dict `:` type($output)
}];
}
def SpatHostStoreSyncOp : SpatOp<"host_store_sync", []> {
let summary = "Store a tensor to host memory and signal its consumer";
let arguments = (ins
Index:$sourceCoreId,
Index:$targetCoreId,
Index:$hostOffset,
Index:$eventRegister,
SpatTensor:$input
);
let assemblyFormat = [{
$input `from` $sourceCoreId `to` $targetCoreId
`host_offset` $hostOffset `event` $eventRegister attr-dict `:` type($input)
}];
}
def SpatHostWaitLoadOp : SpatOp<"host_wait_load", []> {
let summary = "Wait for a producer and load its tensor from host memory";
let arguments = (ins
Index:$sourceCoreId,
Index:$targetCoreId,
Index:$hostOffset,
Index:$eventRegister
);
let results = (outs
SpatTensor:$output
);
let assemblyFormat = [{
`from` $sourceCoreId `to` $targetCoreId
`host_offset` $hostOffset `event` $eventRegister attr-dict `:` type($output)
}];
}