diff --git a/include/mlir-hlo/Dialect/mhlo/IR/chlo_ops.td b/include/mlir-hlo/Dialect/mhlo/IR/chlo_ops.td index 96b9437..22106d9 100644 --- a/include/mlir-hlo/Dialect/mhlo/IR/chlo_ops.td +++ b/include/mlir-hlo/Dialect/mhlo/IR/chlo_ops.td @@ -76,11 +76,9 @@ class HLOClient_Op traits> : //===----------------------------------------------------------------------===// class HLOClient_BroadcastBinaryElementwiseOp< - string mnemonic, list traits> : - HLOClient_Op])> { + string mnemonic, list traits> : HLOClient_Op]> { let arguments = (ins HLO_Tensor:$lhs, HLO_Tensor:$rhs, @@ -411,8 +409,8 @@ def HLOClient_BroadcastComplexOp : HLOClient_BroadcastBinaryElementwiseOp< class HLOClient_UnaryElementwiseOp traits, Type ArgTensorType, Type ResultTensorType> : HLOClient_Op { + traits # [InferFusibilityOpInterface, NoSideEffect, + SameOperandsAndResultShape]> { let arguments = (ins ArgTensorType:$operand); let results = (outs ResultTensorType:$result);