From 11d18a4a5e0b9d242722b2c57660ce34b52cb517 Mon Sep 17 00:00:00 2001
From: River Riddle <riverriddle@google.com>
Date: Tue, 4 Jun 2019 12:14:43 -0700
Subject: [PATCH] Add a utility function to Op that allows for checking if an
 operation has a specific trait.

PiperOrigin-RevId: 251489498
---
 mlir/include/mlir/IR/OpDefinition.h | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/mlir/include/mlir/IR/OpDefinition.h b/mlir/include/mlir/IR/OpDefinition.h
index 35c248f..7a6b861 100644
--- 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(); }
 
-- 
2.7.4