Add a utility function to Op that allows for checking if an operation has a specific...

PiperOrigin-RevId: 251489498

diff --git a/mlir/include/mlir/IR/OpDefinition.h b/mlir/include/mlir/IR/OpDefinition.h
index 35c248f..7a6b861 100644 (file)
--- a/mlir/include/mlir/IR/OpDefinition.h
+++ b/mlir/include/mlir/IR/OpDefinition.h
@@ -749,6 +749,13 @@ class Op : public OpState,
                typelist_contains<OpTrait::OneResult<ConcreteType>, OpState,
                                  Traits<ConcreteType>...>::value> {
 public:
+  /// Return if this operation contains the provided trait.
+  template <template <typename T> class Trait>
+  static constexpr bool hasTrait() {
+    return typelist_contains<Trait<ConcreteType>, OpState,
+                             Traits<ConcreteType>...>::value;
+  }
+
   /// Return the operation that this refers to.
   Operation *getOperation() { return OpState::getOperation(); }