diff --git a/include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td b/include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td index c4e9a84..05fc6bf 100644 --- a/include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td +++ b/include/mlir-hlo/Dialect/mhlo/IR/hlo_ops.td @@ -744,7 +744,7 @@ def HLO_SendOp : HLO_Op<"send", []> { let arguments = (ins HLO_TensorOrTuple:$operand, HLO_Token:$token, - ChannelHandle:$channel_id, + ChannelHandle:$channel_handle, DefaultValuedAttr:$is_host_transfer ); @@ -769,7 +769,7 @@ def HLO_RecvOp : HLO_Op<"recv", []> { let arguments = (ins HLO_Token:$token, - ChannelHandle:$channel_id, + ChannelHandle:$channel_handle, DefaultValuedAttr:$is_host_transfer ); diff --git a/tests/ops.mlir b/tests/ops.mlir index a437aa0..7c33d2f 100644 --- a/tests/ops.mlir +++ b/tests/ops.mlir @@ -682,7 +682,7 @@ func @map_unranked(%arg0: tensor<*xf32>, %arg1: tensor<*xf32>) -> tensor<*xf32> func @recv_invalid_number_of_results(%token: !mhlo.token) -> tuple, tensor, !mhlo.token> { // expected-error@+1 {{result is expected to be a tuple of size 2, but got 3}} %0 = "mhlo.recv"(%token) { - channel_id = { + channel_handle = { handle = 5 : i64, type = 3 : i64 // Host to device channel }, @@ -696,7 +696,7 @@ func @recv_invalid_number_of_results(%token: !mhlo.token) -> tuple tuple, tensor> { // expected-error@+1 {{second element of result tuple is expected to be of token type, but got 'tensor'}} %0 = "mhlo.recv"(%token) { - channel_id = { + channel_handle = { handle = 5 : i64, type = 3 : i64 // Host to device channel },