diff --git a/flang/include/flang/Optimizer/Dialect/CUF/CUFOps.td b/flang/include/flang/Optimizer/Dialect/CUF/CUFOps.td index 9a31ffa2e9471..6f886726b1283 100644 --- a/flang/include/flang/Optimizer/Dialect/CUF/CUFOps.td +++ b/flang/include/flang/Optimizer/Dialect/CUF/CUFOps.td @@ -222,7 +222,8 @@ def cuf_KernelLaunchOp : cuf_Op<"kernel_launch", [CallOpInterface, } void setCalleeFromCallable(mlir::CallInterfaceCallable callee) { - (*this)->setAttr(getCalleeAttrName(), callee.get()); + (*this)->setAttr(getCalleeAttrName(), + llvm::cast(callee)); } mlir::FunctionType getFunctionType(); diff --git a/flang/include/flang/Optimizer/Dialect/FIROps.td b/flang/include/flang/Optimizer/Dialect/FIROps.td index 01f588b3c8ba5..5f0f0b48e892b 100644 --- a/flang/include/flang/Optimizer/Dialect/FIROps.td +++ b/flang/include/flang/Optimizer/Dialect/FIROps.td @@ -2488,8 +2488,9 @@ def fir_CallOp : fir_Op<"call", void setCalleeFromCallable(mlir::CallInterfaceCallable callee) { if (auto calling = (*this)->getAttrOfType(getCalleeAttrName())) - (*this)->setAttr(getCalleeAttrName(), callee.get()); - setOperand(0, callee.get()); + (*this)->setAttr(getCalleeAttrName(), + llvm::cast(callee)); + setOperand(0, llvm::cast(callee)); } }]; }