[mlir] Specialize DenseMapInfo and PointerLikeTypeTraits for mlir::TypedValue

[ezhulenev]

Revert accidental formatting changes

[ezhulenev]

Accidentally submitted wrong revision, fixed in https://github.com/llvm/llvm-project/commit/38785647121cef0adde4d3adbe5c5fe21b83d32b

mlir/include/mlir/IR/Value.h

 /// value is either a BlockArgument or the result of an operation. Note: This
 /// class has value-type semantics and is just a simple wrapper around a
 /// ValueImpl that is either owner by a block(in the case of a BlockArgument) or
 /// an Operation(in the case of an OpResult).
 class Value {
 public:
   constexpr Value(detail::ValueImpl *impl = nullptr) : impl(impl) {}
 
-  template <typename U>
-  bool isa() const {
-    return llvm::isa<U>(*this);
-  }
+  template <typename U> bool isa() const { return llvm::isa<U>(*this); }
 
-  template <typename U>
-  U dyn_cast() const {
-    return llvm::dyn_cast<U>(*this);
-  }
+  template <typename U> U dyn_cast() const { return llvm::dyn_cast<U>(*this); }
 
-  template <typename U>
-  U dyn_cast_or_null() const {
+  template <typename U> U dyn_cast_or_null() const {
     return llvm::dyn_cast_if_present<U>(*this);
   }
 
-  template <typename U>
-  U cast() const {
-    return llvm::cast<U>(*this);
-  }
+  template <typename U> U cast() const { return llvm::cast<U>(*this); }
 
   explicit operator bool() const { return impl; }
   bool operator==(const Value &other) const { return impl == other.impl; }
   bool operator!=(const Value &other) const { return !(*this == other); }
 
   /// Return the type of this value.
   Type getType() const { return impl->getType(); }
 
public:
   void setType(Type newType) { impl->setType(newType); }
 
   /// If this value is the result of an operation, return the operation that
   /// defines it.
   Operation *getDefiningOp() const;
 
   /// If this value is the result of an operation of type OpTy, return the
   /// operation that defines it.
-  template <typename OpTy>
-  OpTy getDefiningOp() const {
+  template <typename OpTy> OpTy getDefiningOp() const {
     return llvm::dyn_cast_or_null<OpTy>(getDefiningOp());
   }
 
   /// Return the location of this value.
   Location getLoc() const;
   void setLoc(Location loc);
 
   /// Return the Region in which this Value is defined.
static mlir::OpResult getEmptyKey() {
     void *pointer = llvm::DenseMapInfo<void *>::getEmptyKey();
     return reinterpret_cast<mlir::detail::OpResultImpl *>(pointer);
   }
   static mlir::OpResult getTombstoneKey() {
     void *pointer = llvm::DenseMapInfo<void *>::getTombstoneKey();
     return reinterpret_cast<mlir::detail::OpResultImpl *>(pointer);
   }
 };
+template <typename T>
+struct DenseMapInfo<mlir::detail::TypedValue<T>>
+    : public DenseMapInfo<mlir::Value> {
+  static mlir::detail::TypedValue<T> getEmptyKey() {
+    void *pointer = llvm::DenseMapInfo<void *>::getEmptyKey();
+    return reinterpret_cast<mlir::detail::ValueImpl *>(pointer);
+  }
+  static mlir::detail::TypedValue<T> getTombstoneKey() {
+    void *pointer = llvm::DenseMapInfo<void *>::getTombstoneKey();
+    return reinterpret_cast<mlir::detail::ValueImpl *>(pointer);
+  }
+};
 
 /// Allow stealing the low bits of a value.
-template <>
-struct PointerLikeTypeTraits<mlir::Value> {
+template <> struct PointerLikeTypeTraits<mlir::Value> {
 public:
   static inline void *getAsVoidPointer(mlir::Value value) {
     return const_cast<void *>(value.getAsOpaquePointer());
   }
   static inline mlir::Value getFromVoidPointer(void *pointer) {
     return mlir::Value::getFromOpaquePointer(pointer);
   }
   enum {

 template <>
 struct PointerLikeTypeTraits<mlir::OpResult>
     : public PointerLikeTypeTraits<mlir::Value> {
 public:
   static inline mlir::OpResult getFromVoidPointer(void *pointer) {
     return reinterpret_cast<mlir::detail::OpResultImpl *>(pointer);
   }
 };
+template <typename T>
+struct PointerLikeTypeTraits<mlir::detail::TypedValue<T>>
+    : public PointerLikeTypeTraits<mlir::Value> {
+public:
+  static inline mlir::detail::TypedValue<T> getFromVoidPointer(void *pointer) {
+    return reinterpret_cast<mlir::detail::ValueImpl *>(pointer);
+  }
+};
 
 /// Add support for llvm style casts. We provide a cast between To and From if
 /// From is mlir::Value or derives from it.
 template <typename To, typename From>
 struct CastInfo<
     To, From,
     std::enable_if_t<std::is_same_v<mlir::Value, std::remove_const_t<From>> ||
                      std::is_base_of_v<mlir::Value, From>>>