diff --git a/mlir/include/mlir/Dialect/Complex/IR/ComplexAttributes.td b/mlir/include/mlir/Dialect/Complex/IR/ComplexAttributes.td index ab6074eccb53..afae94c41ad8 100644 --- a/mlir/include/mlir/Dialect/Complex/IR/ComplexAttributes.td +++ b/mlir/include/mlir/Dialect/Complex/IR/ComplexAttributes.td @@ -53,6 +53,12 @@ def Complex_NumberAttr : Complex_Attr<"Number", "number"> { }]> ]; + let extraClassDeclaration = [{ + std::complex getValue() { + return std::complex(getReal(), getImag()); + } + }]; + let genVerifyDecl = 1; let hasCustomAssemblyFormat = 1; let skipDefaultBuilders = 1;