Add a getLoc() method on mlir::Value that returns the loc of the defining Operations if any, otherwise an unknown location

--

PiperOrigin-RevId: 241354085

mlir/include/mlir/IR/Value.h

@@ -77,6 +77,10 @@ public:
   /// defines it.
   Operation *getDefiningOp();
 
+  /// If this value is the result of an operation, use it as a location,
+  /// otherwise return an unknown location.
+  Location getLoc();
+
   using use_iterator = ValueUseIterator<OpOperand>;
   using use_range = llvm::iterator_range<use_iterator>;
 

mlir/lib/IR/Value.cpp

@@ -38,6 +38,13 @@ Function *Value::getFunction() {
   }
 }
 
+Location Value::getLoc() {
+  if (auto *op = getDefiningOp()) {
+    return op->getLoc();
+  }
+  return UnknownLoc::get(getContext());
+}
+
 //===----------------------------------------------------------------------===//
 // IRObjectWithUseList implementation.
 //===----------------------------------------------------------------------===//