diff --git a/mlir/include/mlir-c/Dialect/Func.h b/mlir/include/mlir-c/Dialect/Func.h index 1df759f0e8e3f..001f915af1054 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 942e090fdfc19..8265b61b98426 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(unwrap(op)) .setArgAttr(pos, unwrap(name), unwrap(attr)); } + +void mlirFuncSetResultAttr(MlirOperation op, intptr_t pos, MlirStringRef name, + MlirAttribute attr) { + llvm::cast(unwrap(op)) + .setResultAttr(pos, unwrap(name), unwrap(attr)); +}