Using getContext() instead of getInstruction()->getContext() on Operation (NFC)

PiperOrigin-RevId: 240088209

4 files changed, 6 insertions, 8 deletions

diff --git a/mlir/lib/AffineOps/AffineOps.cpp b/mlir/lib/AffineOps/AffineOps.cpp
index 0c8dcb5ab56..4badde9012b 100644
--- a/mlir/lib/AffineOps/AffineOps.cpp
+++ b/mlir/lib/AffineOps/AffineOps.cpp
@@ -1063,7 +1063,7 @@ Block *AffineForOp::createBody() {
   // Create a new block for the body, and add an argument for the induction
   // variable.
   Block *body = new Block();
-  body->addArgument(IndexType::get(getInstruction()->getContext()));
+  body->addArgument(IndexType::get(getContext()));
   bodyRegion.push_back(body);
   return body;
 }
@@ -1139,13 +1139,11 @@ int64_t AffineForOp::getConstantUpperBound() {
 }
 
 void AffineForOp::setConstantLowerBound(int64_t value) {
-  setLowerBound(
-      {}, AffineMap::getConstantMap(value, getInstruction()->getContext()));
+  setLowerBound({}, AffineMap::getConstantMap(value, getContext()));
 }
 
 void AffineForOp::setConstantUpperBound(int64_t value) {
-  setUpperBound(
-      {}, AffineMap::getConstantMap(value, getInstruction()->getContext()));
+  setUpperBound({}, AffineMap::getConstantMap(value, getContext()));
 }
 
 AffineForOp::operand_range AffineForOp::getLowerBoundOperands() {
diff --git a/mlir/lib/Analysis/LoopAnalysis.cpp b/mlir/lib/Analysis/LoopAnalysis.cpp
index 78f8aeb978a..ba9a29177fe 100644
--- a/mlir/lib/Analysis/LoopAnalysis.cpp
+++ b/mlir/lib/Analysis/LoopAnalysis.cpp
@@ -244,7 +244,7 @@ static bool isContiguousAccess(Value &iv, LoadOrStoreOp memoryOp,
   auto layoutMap = memRefType.getAffineMaps();
   // TODO(ntv): remove dependence on Builder once we support non-identity
   // layout map.
-  Builder b(memoryOp->getInstruction()->getContext());
+  Builder b(memoryOp.getContext());
   if (layoutMap.size() >= 2 ||
       (layoutMap.size() == 1 &&
        !(layoutMap[0] ==
diff --git a/mlir/lib/StandardOps/Ops.cpp b/mlir/lib/StandardOps/Ops.cpp
index 67c51d54789..a14f3a24e82 100644
--- a/mlir/lib/StandardOps/Ops.cpp
+++ b/mlir/lib/StandardOps/Ops.cpp
@@ -720,7 +720,7 @@ void CmpIOp::print(OpAsmPrinter *p) {
   assert(predicateValue >= static_cast<int>(CmpIPredicate::FirstValidValue) &&
          predicateValue < static_cast<int>(CmpIPredicate::NumPredicates) &&
          "unknown predicate index");
-  Builder b(getInstruction()->getContext());
+  Builder b(getContext());
   auto predicateStringAttr =
       b.getStringAttr(getPredicateNames()[predicateValue]);
   p->printAttribute(predicateStringAttr);
diff --git a/mlir/lib/Transforms/MaterializeVectors.cpp b/mlir/lib/Transforms/MaterializeVectors.cpp
index d1374e92dc8..0a7eaabbb09 100644
--- a/mlir/lib/Transforms/MaterializeVectors.cpp
+++ b/mlir/lib/Transforms/MaterializeVectors.cpp
@@ -454,7 +454,7 @@ static AffineMap projectedPermutationMap(VectorTransferOpTy transfer,
          "Shape and ratio not of the same size");
   unsigned dim = 0;
   SmallVector<AffineExpr, 4> keep;
-  MLIRContext *context = transfer->getInstruction()->getContext();
+  MLIRContext *context = transfer.getContext();
   functional::zipApply(
       [&dim, &keep, context](int64_t shape, int64_t ratio) {
         assert(shape >= ratio && "shape dim must be greater than ratio dim");