Repository navigation
Consider splitting results into outputs and token in InfeedOp/RecvOp #670
Description
def StableHLO_RecvOp : StableHLO_Op<"recv", []> {
let arguments = (ins
HLO_Token:$token,
StableHLO_ChannelHandle:$channel_handle,
DefaultValuedOptionalAttr<BoolAttr, "false">:$is_host_transfer
);
let results = (outs Variadic<HLO_TensorOrToken>);
}
The last result is known to be a token, so we should split results into two parts: 1) all results except for the last one will go to output, 2) the last one will go to result.
Activity
[-]Split results into outputs and token in RecvOp[/-][+]Split results into outputs and token in InfeedOp/RecvOp[/+][-]Split results into outputs and token in InfeedOp/RecvOp[/-][+]Consider splitting results into outputs and token in InfeedOp/RecvOp[/+]