[Mlir-commits] [mlir] a15618f - [mlir] feat: add `mlirFuncSetResultAttr` (#125972)

[llvmlistbot at llvm.org]

Author: Avik Pal
Date: 2025-02-06T17:33:12-06:00
New Revision: a15618f18cb000cce59df810c9e75153aa5e0aca

URL: https://github.com/llvm/llvm-project/commit/a15618f18cb000cce59df810c9e75153aa5e0aca
DIFF: https://github.com/llvm/llvm-project/commit/a15618f18cb000cce59df810c9e75153aa5e0aca.diff

LOG: [mlir] feat: add `mlirFuncSetResultAttr` (#125972)

cc @ftynse @wsmoses

Added: 
    

Modified: 
    mlir/include/mlir-c/Dialect/Func.h
    mlir/lib/CAPI/Dialect/Func.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/include/mlir-c/Dialect/Func.h b/mlir/include/mlir-c/Dialect/Func.h
index 1df759f0e8e3fe0..001f915af1054f0 100644
--- a/mlir/include/mlir-c/Dialect/Func.h
+++ b/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

diff  --git a/mlir/lib/CAPI/Dialect/Func.cpp b/mlir/lib/CAPI/Dialect/Func.cpp
index 942e090fdfc19d7..8265b61b9842691 100644
--- a/mlir/lib/CAPI/Dialect/Func.cpp
+++ b/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));
+}