diff --git a/mlir/include/mlir/Dialect/Tosa/IR/TosaTypesBase.td b/mlir/include/mlir/Dialect/Tosa/IR/TosaTypesBase.td --- a/mlir/include/mlir/Dialect/Tosa/IR/TosaTypesBase.td +++ b/mlir/include/mlir/Dialect/Tosa/IR/TosaTypesBase.td @@ -154,6 +154,7 @@ TensorOf.predicate]>, description>; +def Tosa_IntLike : Tosa_TypeLike<[Tosa_Int], "signless-integer-like">; def Tosa_Int8Like : Tosa_TypeLike<[Tosa_Int8], "signless-integer-8-bit-like">; def Tosa_Int16Like : Tosa_TypeLike<[Tosa_Int16], "signless-integer-16-bit-like">; def Tosa_Int32Like : Tosa_TypeLike<[Tosa_Int32], "signless-integer-32-bit-like">; diff --git a/mlir/include/mlir/Dialect/Tosa/IR/TosaUtilOps.td b/mlir/include/mlir/Dialect/Tosa/IR/TosaUtilOps.td --- a/mlir/include/mlir/Dialect/Tosa/IR/TosaUtilOps.td +++ b/mlir/include/mlir/Dialect/Tosa/IR/TosaUtilOps.td @@ -37,14 +37,14 @@ }]; let arguments = (ins - Tosa_Int:$value, - Tosa_Int:$multiplier, + Tosa_IntLike:$value, + Tosa_IntLike:$multiplier, Tosa_Int8Like:$shift, BoolAttr:$double_round ); let results = (outs - Tosa_Int:$output + Tosa_IntLike:$output ); }