Use isa and cast as is and get are deprecated

Signed-off-by: Nirvedh Meshram <nirvedh@gmail.com>

compiler/src/iree/compiler/Codegen/Common/ConcretizePadResultShape.cpp

@@ -37,7 +37,7 @@ static Value getAsIndexValue(OpFoldResult attrOrValue, OpBuilder &builder,
       return val;
     matchPattern(val, m_Constant(&attr));
   } else {
-    attr = llvm::cast<IntegerAttr>(attrOrValue.get<Attribute>());
+    attr = llvm::cast<IntegerAttr>(cast<Attribute>(attrOrValue));
   }
   return builder.createOrFold<arith::ConstantIndexOp>(
       loc, attr.getValue().getSExtValue());

compiler/src/iree/compiler/Codegen/Common/DecomposePackUnPackOps.cpp

@@ -146,7 +146,7 @@ static LogicalResult commonRunOnOperation(
 
               // Do nothing if any of inner tile sizes is dynamic.
               if (llvm::any_of(packOp.getMixedTiles(), [](OpFoldResult tile) {
-                    return tile.is<Value>();
+                    return isa<Value>(tile);
                   })) {
                 return {};
               }

compiler/src/iree/compiler/Codegen/Common/TensorToVectorVectorizePad.cpp

@@ -38,7 +38,7 @@ static Value getAsIndexValue(OpFoldResult attrOrValue, OpBuilder &builder,
       return val;
     matchPattern(val, m_Constant(&attr));
   } else {
-    attr = llvm::cast<IntegerAttr>(attrOrValue.get<Attribute>());
+    attr = cast<IntegerAttr>(cast<Attribute>(attrOrValue));
   }
   return builder.createOrFold<arith::ConstantIndexOp>(loc, attr.getInt());
 }
@@ -101,12 +101,12 @@ struct VectorizePadWithConditions final
 
     /// Return true if the given `attrOrValue` is a constant zero.
     auto isConstantZero = [](OpFoldResult attrOrValue) {
-      if (attrOrValue.is<Attribute>()) {
-        auto attr = llvm::dyn_cast<IntegerAttr>(attrOrValue.get<Attribute>());
+      if (isa<Attribute>(attrOrValue)) {
+        auto attr = dyn_cast<IntegerAttr>(cast<Attribute>(attrOrValue));
         return attr && attr.getValue().getZExtValue() == 0;
       }
       IntegerAttr attr;
-      return matchPattern(attrOrValue.get<Value>(), m_Constant(&attr)) &&
+      return matchPattern(cast<Value>(attrOrValue), m_Constant(&attr)) &&
              attr.getValue().getZExtValue() == 0;
     };
 

compiler/src/iree/compiler/Codegen/Common/VectorLayoutAnalysis.cpp

@@ -305,7 +305,7 @@ void DistributionLayout::print(raw_ostream &os) const {
 void DistributionLayout::onUpdate(DataFlowSolver *solver) const {
   AnalysisState::onUpdate(solver);
 
-  Value value = anchor.get<Value>();
+  Value value = cast<Value>(anchor);
 
   if (propagation) {
     // Make propagation run again on all users of this value.

compiler/src/iree/compiler/Codegen/LLVMCPU/DispatchABI.cpp

@@ -843,12 +843,12 @@ MemRefDescriptor HALDispatchABI::loadBinding(Operation *forOp, int64_t ordinal,
             currentStrideVal = builder.create<LLVM::ConstantOp>(
                 loc, llvmIndexType, currentStrideInt.value());
           } else {
-            currentStrideVal = currentStride.get<Value>();
+            currentStrideVal = cast<Value>(currentStride);
           }
           currentStride =
               builder.create<LLVM::MulOp>(loc, currentStrideVal, dim)
                   .getResult();
-          desc.setStride(builder, loc, i - 1, currentStride.get<Value>());
+          desc.setStride(builder, loc, i - 1, cast<Value>(currentStride));
         } else {
           currentStride = builder.getIndexAttr(strides[i - 1]);
           desc.setConstantStride(builder, loc, i - 1, strides[i - 1]);

compiler/src/iree/compiler/Codegen/LLVMCPU/KernelDispatch.cpp

@@ -1789,7 +1789,7 @@ static LogicalResult setRootConfig(mlir::FunctionOpInterface entryPointFn,
   DictionaryAttr pipelineConfig;
   auto target = IREE::HAL::ExecutableTargetAttr::lookup(entryPointFn);
   bool hasDynamicInnerTile = llvm::any_of(
-      op.getMixedTiles(), [](OpFoldResult ofr) { return ofr.is<Value>(); });
+      op.getMixedTiles(), [](OpFoldResult ofr) { return isa<Value>(ofr); });
   if (!hasDynamicInnerTile && !isX86(target) && !isRISCV(target)) {
     pipelineConfig = getPipelineConfWithDecompositionAttr(op.getContext());
   }
@@ -1829,7 +1829,7 @@ static LogicalResult setRootConfig(mlir::FunctionOpInterface entryPointFn,
   DictionaryAttr pipelineConfig;
   auto target = IREE::HAL::ExecutableTargetAttr::lookup(entryPointFn);
   bool hasDynamicInnerTile = llvm::any_of(
-      op.getMixedTiles(), [](OpFoldResult ofr) { return ofr.is<Value>(); });
+      op.getMixedTiles(), [](OpFoldResult ofr) { return isa<Value>(ofr); });
   if (!hasDynamicInnerTile && !isX86(target) && !isRISCV(target)) {
     pipelineConfig = getPipelineConfWithDecompositionAttr(op.getContext());
   }

compiler/src/iree/compiler/Codegen/LLVMGPU/ConvertToLLVM.cpp

@@ -433,12 +433,12 @@ class ConvertIREEBindingSubspanOp : public ConvertToLLVMPattern {
               currentStrideVal = rewriter.create<LLVM::ConstantOp>(
                   loc, llvmIndexType, currentStrideInt.value());
             } else {
-              currentStrideVal = currentStride.get<Value>();
+              currentStrideVal = cast<Value>(currentStride);
             }
             currentStride =
                 rewriter.create<LLVM::MulOp>(loc, currentStrideVal, dim)
                     .getResult();
-            desc.setStride(rewriter, loc, i - 1, currentStride.get<Value>());
+            desc.setStride(rewriter, loc, i - 1, cast<Value>(currentStride));
           } else {
             currentStride = rewriter.getIndexAttr(strides[i - 1]);
             desc.setConstantStride(rewriter, loc, i - 1, strides[i - 1]);

compiler/src/iree/compiler/Codegen/LLVMGPU/LLVMGPUTensorPad.cpp

@@ -38,7 +38,7 @@ getPaddedShapeFromTensorLoad(IREE::Flow::DispatchTensorLoadOp tensorLoad,
       FailureOr<int64_t> upperBound =
           ValueBoundsConstraintSet::computeConstantBound(
               presburger::BoundType::UB,
-              {size.get<Value>(), /*dim=*/std::nullopt},
+              {cast<Value>(size), /*dim=*/std::nullopt},
               /*stopCondition=*/nullptr, /*closedUB=*/true);
       if (failed(upperBound))
         return failure();

compiler/src/iree/compiler/Codegen/SPIRV/SPIRVVectorizeLoadStore.cpp

@@ -1000,11 +1000,11 @@ struct ReifyExtractOfCreateMask final
     for (auto [idx, size] :
          llvm::zip_equal(extractOp.getMixedPosition(), maskOp.getOperands())) {
       Value idxVal;
-      if (idx.is<Attribute>()) {
+      if (isa<Attribute>(idx)) {
         idxVal = rewriter.create<arith::ConstantIndexOp>(
-            loc, cast<IntegerAttr>(idx.get<Attribute>()).getInt());
+            loc, cast<IntegerAttr>(cast<Attribute>(idx)).getInt());
       } else {
-        idxVal = idx.get<Value>();
+        idxVal = cast<Value>(idx);
       }
       Value cmpIdx = rewriter.create<arith::CmpIOp>(
           loc, arith::CmpIPredicate::slt, idxVal, size);

compiler/src/iree/compiler/Codegen/Transforms/Transforms.cpp

@@ -67,10 +67,10 @@ static SliceAndDynamicDims cloneOffsetsSizesAndStridesImpl(
     SmallVector<OpFoldResult> clonedOfrs;
     clonedOfrs.reserve(ofrs.size());
     for (auto ofr : ofrs) {
-      if (ofr.is<Attribute>()) {
+      if (isa<Attribute>(ofr)) {
         clonedOfrs.push_back(ofr);
       } else {
-        clonedOfrs.push_back(bvm.lookupOrDefault(ofr.get<Value>()));
+        clonedOfrs.push_back(bvm.lookupOrDefault(cast<Value>(ofr)));
       }
     }
     return clonedOfrs;

compiler/src/iree/compiler/Dialect/Flow/IR/FlowOps.td

@@ -1013,7 +1013,7 @@ def FLOW_CallOp : FLOW_Op<"call", [
 
     /// Set the callee for this operation.
     void setCalleeFromCallable(CallInterfaceCallable callee) {
-      (*this)->setAttr("callee", callee.get<SymbolRefAttr>());
+      (*this)->setAttr("callee", cast<SymbolRefAttr>(callee));
     }
 
     ValueRange getOperandDynamicDims(unsigned idx) {

compiler/src/iree/compiler/Dialect/Flow/Transforms/RegionOpUtils.cpp

@@ -293,7 +293,7 @@ reifyDynamicResultDimsImpl(OpBuilder &b, Value value,
       return failure();
     for (int64_t i = 0; i < shapedType.getRank(); ++i)
       if (shapedType.isDynamicDim(i))
-        dynamicDims.push_back(dims[opResult.getResultNumber()][i].get<Value>());
+        dynamicDims.push_back(cast<Value>(dims[opResult.getResultNumber()][i]));
     return success();
   }
 

compiler/src/iree/compiler/Dialect/Stream/IR/StreamOps.td

@@ -2642,7 +2642,7 @@ def Stream_AsyncCallOp : Stream_Op<"async.call", [
 
     /// Set the callee for this operation.
     void setCalleeFromCallable(CallInterfaceCallable callee) {
-      (*this)->setAttr("callee", callee.get<SymbolRefAttr>());
+      (*this)->setAttr("callee", cast<SymbolRefAttr>(callee));
     }
 
     Value getOperandSize(unsigned idx) {
@@ -3322,7 +3322,7 @@ def Stream_CmdCallOp : Stream_Op<"cmd.call", [
 
     /// Set the callee for this operation.
     void setCalleeFromCallable(CallInterfaceCallable callee) {
-      (*this)->setAttr("callee", callee.get<SymbolRefAttr>());
+      (*this)->setAttr("callee", cast<SymbolRefAttr>(callee));
     }
 
     Value getOperandSize(unsigned idx) {

compiler/src/iree/compiler/Dialect/Util/Analysis/Explorer.cpp

@@ -164,7 +164,7 @@ void Explorer::initializeGlobalInfos() {
 void Explorer::initializeInverseCallGraph() {
   forEachFunctionLikeOp([&](FunctionOpInterface parentOp) {
     parentOp->walk([&](CallOpInterface callOp) {
-      if (callOp.getCallableForCallee().is<Value>()) {
+      if (isa<Value>(callOp.getCallableForCallee())) {
         // Indirect calls can't be tracked in the call graph, so ensure we mark
         // the incomplete flag so that any call graph queries return
         // TraversalResult::INCOMPLETE.
@@ -777,7 +777,7 @@ TraversalResult Explorer::walkDefiningOps(Value value, ResultWalkFn fn,
     // Indirect calls would require us to perform an analysis to first see if we
     // can make them direct or annotate the call sites with the possible
     // targets.
-    if (callOp.getCallableForCallee().is<Value>()) {
+    if (isa<Value>(callOp.getCallableForCallee())) {
       LLVM_DEBUG({
         llvm::dbgs()
             << "  !! traversal incomplete due to unanalyzable indirect call: ";
@@ -786,7 +786,7 @@ TraversalResult Explorer::walkDefiningOps(Value value, ResultWalkFn fn,
       });
       return TraversalResult::INCOMPLETE;
     }
-    auto targetSymbol = callOp.getCallableForCallee().get<SymbolRefAttr>();
+    auto targetSymbol = cast<SymbolRefAttr>(callOp.getCallableForCallee());
     auto targetOp = symbolTables.lookupNearestSymbolFrom<CallableOpInterface>(
         callOp, targetSymbol);
     assert(targetOp && "call target not found");
@@ -1031,7 +1031,7 @@ TraversalResult Explorer::walkTransitiveUses(Value value, UseWalkFn fn,
   // Move across a call to the callee entry block.
   auto traverseCallOp = [&](CallOpInterface callOp, unsigned operandIdx) {
     auto callable = callOp.getCallableForCallee();
-    if (callable.is<Value>()) {
+    if (isa<Value>(callable)) {
       LLVM_DEBUG({
         llvm::dbgs()
             << "  !! traversal incomplete due to unanalyzable indirect call: ";
@@ -1040,7 +1040,7 @@ TraversalResult Explorer::walkTransitiveUses(Value value, UseWalkFn fn,
       });
       return TraversalResult::INCOMPLETE;
     }
-    auto targetSymbol = callable.get<SymbolRefAttr>();
+    auto targetSymbol = cast<SymbolRefAttr>(callable);
     auto targetOp = symbolTables.lookupNearestSymbolFrom<CallableOpInterface>(
         callOp, targetSymbol);
     assert(targetOp && "call target not found");

compiler/src/iree/compiler/Dialect/Util/IR/UtilOps.td

@@ -820,7 +820,7 @@ def Util_CallOp : Util_Op<"call", [
       return (*this)->getAttrOfType<SymbolRefAttr>("callee");
     }
     void setCalleeFromCallable(CallInterfaceCallable callee) {
-      (*this)->setAttr("callee", callee.get<SymbolRefAttr>());
+      (*this)->setAttr("callee", cast<SymbolRefAttr>(callee));
     }
 
     // Clones the call and potentially expands each operand and result.

compiler/src/iree/compiler/Dialect/VM/IR/VMOps.td

@@ -4223,7 +4223,7 @@ class VM_CallBaseOp<string mnemonic, list<Trait> traits = []> :
 
     /// Set the callee for this operation.
     void setCalleeFromCallable(CallInterfaceCallable callee) {
-      (*this)->setAttr("callee", callee.get<SymbolRefAttr>());
+      (*this)->setAttr("callee", cast<SymbolRefAttr>(callee));
     }
 
     /// Get the argument operands to the called function as a mutable range, this is