diff options
| author | Avik Pal <avik.pal.2017@gmail.com> | 2025-02-06 18:33:12 -0500 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-02-06 17:33:12 -0600 |
| commit | a15618f18cb000cce59df810c9e75153aa5e0aca (patch) | |
| tree | 7d75f183951acdf71276c0489df936159fcdd37e | |
| parent | 14716f2e4b2d771af0fe9163540a8f08ffe1e5ec (diff) | |
| download | llvm-a15618f18cb000cce59df810c9e75153aa5e0aca.zip llvm-a15618f18cb000cce59df810c9e75153aa5e0aca.tar.gz llvm-a15618f18cb000cce59df810c9e75153aa5e0aca.tar.bz2 | |
[mlir] feat: add `mlirFuncSetResultAttr` (#125972)
cc @ftynse @wsmoses
| -rw-r--r-- | mlir/include/mlir-c/Dialect/Func.h | 4 | ||||
| -rw-r--r-- | mlir/lib/CAPI/Dialect/Func.cpp | 6 |
2 files changed, 10 insertions, 0 deletions
diff --git a/mlir/include/mlir-c/Dialect/Func.h b/mlir/include/mlir-c/Dialect/Func.h index 1df759f..001f915 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 942e090..8265b61 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)); +} |