diff --git a/mlir/include/mlir/Dialect/Async/IR/AsyncOps.td b/mlir/include/mlir/Dialect/Async/IR/AsyncOps.td index 33b6792175234..a08f5d6e714ef 100644 --- a/mlir/include/mlir/Dialect/Async/IR/AsyncOps.td +++ b/mlir/include/mlir/Dialect/Async/IR/AsyncOps.td @@ -256,7 +256,7 @@ def Async_CallOp : Async_Op<"call", /// Set the callee for this operation. void setCalleeFromCallable(CallInterfaceCallable callee) { - (*this)->setAttr("callee", callee.get()); + (*this)->setAttr("callee", cast(callee)); } }]; diff --git a/mlir/include/mlir/Dialect/EmitC/IR/EmitC.td b/mlir/include/mlir/Dialect/EmitC/IR/EmitC.td index 744a0dc4770e6..b16f5a8619fe7 100644 --- a/mlir/include/mlir/Dialect/EmitC/IR/EmitC.td +++ b/mlir/include/mlir/Dialect/EmitC/IR/EmitC.td @@ -580,7 +580,7 @@ def EmitC_CallOp : EmitC_Op<"call", /// Set the callee for this operation. void setCalleeFromCallable(CallInterfaceCallable callee) { - (*this)->setAttr("callee", callee.get()); + (*this)->setAttr("callee", cast(callee)); } }]; diff --git a/mlir/include/mlir/Dialect/Func/IR/FuncOps.td b/mlir/include/mlir/Dialect/Func/IR/FuncOps.td index 211201802b08c..4da0efcb13ddf 100644 --- a/mlir/include/mlir/Dialect/Func/IR/FuncOps.td +++ b/mlir/include/mlir/Dialect/Func/IR/FuncOps.td @@ -97,7 +97,7 @@ def CallOp : Func_Op<"call", /// Set the callee for this operation. void setCalleeFromCallable(CallInterfaceCallable callee) { - (*this)->setAttr("callee", callee.get()); + (*this)->setAttr("callee", cast(callee)); } }]; @@ -168,7 +168,7 @@ def CallIndirectOp : Func_Op<"call_indirect", [ /// Set the callee for this operation. void setCalleeFromCallable(CallInterfaceCallable callee) { - setOperand(0, callee.get()); + setOperand(0, cast(callee)); } }]; diff --git a/mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td b/mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td index a3aa53b1fcb85..b2281536aa40b 100644 --- a/mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td +++ b/mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td @@ -339,7 +339,7 @@ def LLVM_GEPOp : LLVM_Op<"getelementptr", [Pure, indices.push_back(value); else indices.push_back( - builder.getInt32(valueOrAttr.get().getInt())); + builder.getInt32(cast(valueOrAttr).getInt())); } Type baseElementType = op.getElemType(); llvm::Type *elementType = moduleTranslation.convertType(baseElementType); diff --git a/mlir/include/mlir/Dialect/Transform/IR/TransformOps.td b/mlir/include/mlir/Dialect/Transform/IR/TransformOps.td index 1eebddca3df4d..77ed6b322451e 100644 --- a/mlir/include/mlir/Dialect/Transform/IR/TransformOps.td +++ b/mlir/include/mlir/Dialect/Transform/IR/TransformOps.td @@ -899,7 +899,7 @@ def IncludeOp : TransformDialectOp<"include", } void setCalleeFromCallable(::mlir::CallInterfaceCallable callee) { - setTargetAttr(callee.get()); + setTargetAttr(cast(callee)); } ::mlir::Operation::operand_range getArgOperands() { diff --git a/mlir/lib/Dialect/Affine/IR/AffineOps.cpp b/mlir/lib/Dialect/Affine/IR/AffineOps.cpp index b45829bcf6d2c..aa2c2041f411f 100644 --- a/mlir/lib/Dialect/Affine/IR/AffineOps.cpp +++ b/mlir/lib/Dialect/Affine/IR/AffineOps.cpp @@ -5069,7 +5069,7 @@ static OpFoldResult computeProduct(Location loc, OpBuilder &builder, if (maybeConst) { result = result * builder.getAffineConstantExpr(*maybeConst); } else { - dynamicPart.push_back(term.get()); + dynamicPart.push_back(cast(term)); result = result * builder.getAffineSymbolExpr(nDynamic++); } } diff --git a/mlir/test/lib/Dialect/Affine/TestReifyValueBounds.cpp b/mlir/test/lib/Dialect/Affine/TestReifyValueBounds.cpp index 2c954ffc4acef..891b3bab8629d 100644 --- a/mlir/test/lib/Dialect/Affine/TestReifyValueBounds.cpp +++ b/mlir/test/lib/Dialect/Affine/TestReifyValueBounds.cpp @@ -150,7 +150,7 @@ static LogicalResult testReifyValueBounds(FunctionOpInterface funcOp, return WalkResult::skip(); } Value constOp = rewriter.create( - op->getLoc(), cast(reified->get()).getInt()); + op->getLoc(), cast(cast(*reified)).getInt()); rewriter.replaceOp(op, constOp); return WalkResult::skip(); }); diff --git a/mlir/test/lib/Dialect/Test/TestOpDefs.cpp b/mlir/test/lib/Dialect/Test/TestOpDefs.cpp index b268e549b93ab..c6be26d0a44d9 100644 --- a/mlir/test/lib/Dialect/Test/TestOpDefs.cpp +++ b/mlir/test/lib/Dialect/Test/TestOpDefs.cpp @@ -1098,7 +1098,7 @@ CallInterfaceCallable TestCallAndStoreOp::getCallableForCallee() { } void TestCallAndStoreOp::setCalleeFromCallable(CallInterfaceCallable callee) { - setCalleeAttr(callee.get()); + setCalleeAttr(cast(callee)); } Operation::operand_range TestCallAndStoreOp::getArgOperands() { @@ -1117,7 +1117,7 @@ CallInterfaceCallable TestCallOnDeviceOp::getCallableForCallee() { } void TestCallOnDeviceOp::setCalleeFromCallable(CallInterfaceCallable callee) { - setCalleeAttr(callee.get()); + setCalleeAttr(cast(callee)); } Operation::operand_range TestCallOnDeviceOp::getArgOperands() { diff --git a/mlir/test/lib/Dialect/Test/TestOps.td b/mlir/test/lib/Dialect/Test/TestOps.td index bafab155eb9d5..0b1f22b3ee932 100644 --- a/mlir/test/lib/Dialect/Test/TestOps.td +++ b/mlir/test/lib/Dialect/Test/TestOps.td @@ -566,7 +566,7 @@ def ConversionCallOp : TEST_Op<"conversion_call_op", } void $cppClass::setCalleeFromCallable(::mlir::CallInterfaceCallable callee) { - (*this)->setAttr("callee", callee.get()); + (*this)->setAttr("callee", cast(callee)); } }]; } diff --git a/mlir/test/lib/Dialect/Test/TestTypes.cpp b/mlir/test/lib/Dialect/Test/TestTypes.cpp index 6e31bb71d04d8..b822e019e09d2 100644 --- a/mlir/test/lib/Dialect/Test/TestTypes.cpp +++ b/mlir/test/lib/Dialect/Test/TestTypes.cpp @@ -295,8 +295,9 @@ TestTypeWithLayoutType::verifyEntries(DataLayoutEntryListRef params, for (DataLayoutEntryInterface entry : params) { // This is for testing purposes only, so assert well-formedness. assert(entry.isTypeEntry() && "unexpected identifier entry"); - assert(llvm::isa(entry.getKey().get()) && - "wrong type passed in"); + assert( + llvm::isa(llvm::cast(entry.getKey())) && + "wrong type passed in"); auto array = llvm::dyn_cast(entry.getValue()); assert(array && array.getValue().size() == 2 && "expected array of two elements");