[mlir] feat: add mlirFuncSetResultAttr (#125972)

cc @ftynse @wsmoses

Author: avik-pal

mlir/include/mlir-c/Dialect/Func.h

@@ -35,6 +35,10 @@ MLIR_CAPI_EXPORTED void mlirFuncSetArgAttr(MlirOperation op, intptr_t pos,
                                            MlirStringRef name,
                                            MlirAttribute attr);
 
+MLIR_CAPI_EXPORTED void mlirFuncSetResultAttr(MlirOperation op, intptr_t pos,
+                                              MlirStringRef name,
+                                              MlirAttribute attr);
+
 #ifdef __cplusplus
 }
 #endif

mlir/lib/CAPI/Dialect/Func.cpp

@@ -19,3 +19,9 @@ void mlirFuncSetArgAttr(MlirOperation op, intptr_t pos, MlirStringRef name,
   llvm::cast<mlir::func::FuncOp>(unwrap(op))
       .setArgAttr(pos, unwrap(name), unwrap(attr));
 }
+
+void mlirFuncSetResultAttr(MlirOperation op, intptr_t pos, MlirStringRef name,
+                           MlirAttribute attr) {
+  llvm::cast<mlir::func::FuncOp>(unwrap(op))
+      .setResultAttr(pos, unwrap(name), unwrap(attr));
+}