diff --git a/mlir/include/mlir/Dialect/AffineOps/EDSC/Builders.h b/mlir/include/mlir/Dialect/AffineOps/EDSC/Builders.h --- a/mlir/include/mlir/Dialect/AffineOps/EDSC/Builders.h +++ b/mlir/include/mlir/Dialect/AffineOps/EDSC/Builders.h @@ -92,7 +92,7 @@ } // namespace op -/// Operator overloadings. +/// Arithmetic operator overloadings. template ValueHandle TemplatedIndexedValue::operator+(ValueHandle e) { using op::operator+; @@ -113,7 +113,18 @@ using op::operator/; return static_cast(*this) / e; } +template +ValueHandle TemplatedIndexedValue::operator%(ValueHandle e) { + using op::operator%; + return static_cast(*this) % e; +} +template +ValueHandle TemplatedIndexedValue::operator^(ValueHandle e) { + using op::operator^; + return static_cast(*this) ^ e; +} +/// Assignment-arithmetic operator overloadings. template OperationHandle TemplatedIndexedValue::operator+=(ValueHandle e) { using op::operator+; @@ -134,6 +145,60 @@ using op::operator/; return Store(*this / e, getBase(), {indices.begin(), indices.end()}); } +template +OperationHandle TemplatedIndexedValue::operator%=(ValueHandle e) { + using op::operator%; + return Store(*this % e, getBase(), {indices.begin(), indices.end()}); +} +template +OperationHandle TemplatedIndexedValue::operator^=(ValueHandle e) { + using op::operator^; + return Store(*this ^ e, getBase(), {indices.begin(), indices.end()}); +} + +/// Logical operator overloadings. +template +ValueHandle TemplatedIndexedValue::operator&&(ValueHandle e) { + using op::operator&&; + return static_cast(*this) && e; +} +template +ValueHandle TemplatedIndexedValue::operator||(ValueHandle e) { + using op::operator||; + return static_cast(*this) || e; +} + +/// Comparison operator overloadings. +template +ValueHandle TemplatedIndexedValue::operator==(ValueHandle e) { + using op::operator==; + return static_cast(*this) == e; +} +template +ValueHandle TemplatedIndexedValue::operator!=(ValueHandle e) { + using op::operator!=; + return static_cast(*this) != e; +} +template +ValueHandle TemplatedIndexedValue::operator<(ValueHandle e) { + using op::operator<; + return static_cast(*this) < e; +} +template +ValueHandle TemplatedIndexedValue::operator<=(ValueHandle e) { + using op::operator<=; + return static_cast(*this) <= e; +} +template +ValueHandle TemplatedIndexedValue::operator>(ValueHandle e) { + using op::operator>; + return static_cast(*this) > e; +} +template +ValueHandle TemplatedIndexedValue::operator>=(ValueHandle e) { + using op::operator>=; + return static_cast(*this) >= e; +} } // namespace edsc } // namespace mlir diff --git a/mlir/include/mlir/EDSC/Builders.h b/mlir/include/mlir/EDSC/Builders.h --- a/mlir/include/mlir/EDSC/Builders.h +++ b/mlir/include/mlir/EDSC/Builders.h @@ -538,15 +538,13 @@ ValueHandle getBase() const { return base; } - /// Operator overloadings. + /// Arithmetic operator overloadings. ValueHandle operator+(ValueHandle e); ValueHandle operator-(ValueHandle e); ValueHandle operator*(ValueHandle e); ValueHandle operator/(ValueHandle e); - OperationHandle operator+=(ValueHandle e); - OperationHandle operator-=(ValueHandle e); - OperationHandle operator*=(ValueHandle e); - OperationHandle operator/=(ValueHandle e); + ValueHandle operator%(ValueHandle e); + ValueHandle operator^(ValueHandle e); ValueHandle operator+(TemplatedIndexedValue e) { return *this + static_cast(e); } @@ -559,6 +557,20 @@ ValueHandle operator/(TemplatedIndexedValue e) { return *this / static_cast(e); } + ValueHandle operator%(TemplatedIndexedValue e) { + return *this % static_cast(e); + } + ValueHandle operator^(TemplatedIndexedValue e) { + return *this ^ static_cast(e); + } + + /// Assignment-arithmetic operator overloadings. + OperationHandle operator+=(ValueHandle e); + OperationHandle operator-=(ValueHandle e); + OperationHandle operator*=(ValueHandle e); + OperationHandle operator/=(ValueHandle e); + OperationHandle operator%=(ValueHandle e); + OperationHandle operator^=(ValueHandle e); OperationHandle operator+=(TemplatedIndexedValue e) { return this->operator+=(static_cast(e)); } @@ -571,6 +583,48 @@ OperationHandle operator/=(TemplatedIndexedValue e) { return this->operator/=(static_cast(e)); } + OperationHandle operator%=(TemplatedIndexedValue e) { + return this->operator%=(static_cast(e)); + } + OperationHandle operator^=(TemplatedIndexedValue e) { + return this->operator^=(static_cast(e)); + } + + /// Logical operator overloadings. + ValueHandle operator&&(ValueHandle e); + ValueHandle operator||(ValueHandle e); + ValueHandle operator&&(TemplatedIndexedValue e) { + return *this && static_cast(e); + } + ValueHandle operator||(TemplatedIndexedValue e) { + return *this || static_cast(e); + } + + /// Comparison operator overloadings. + ValueHandle operator==(ValueHandle e); + ValueHandle operator!=(ValueHandle e); + ValueHandle operator<(ValueHandle e); + ValueHandle operator<=(ValueHandle e); + ValueHandle operator>(ValueHandle e); + ValueHandle operator>=(ValueHandle e); + ValueHandle operator==(TemplatedIndexedValue e) { + return *this == static_cast(e); + } + ValueHandle operator!=(TemplatedIndexedValue e) { + return *this != static_cast(e); + } + ValueHandle operator<(TemplatedIndexedValue e) { + return *this < static_cast(e); + } + ValueHandle operator<=(TemplatedIndexedValue e) { + return *this <= static_cast(e); + } + ValueHandle operator>(TemplatedIndexedValue e) { + return *this > static_cast(e); + } + ValueHandle operator>=(TemplatedIndexedValue e) { + return *this >= static_cast(e); + } private: TemplatedIndexedValue(ValueHandle base, ArrayRef indices)