irdl

import%20marimo%0A%0A__generated_with%20%3D%20%220.23.15%22%0Aapp%20%3D%20marimo.App(width%3D%22medium%22)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_()%3A%0A%20%20%20%20import%20marimo%20as%20mo%0A%0A%20%20%20%20return%20(mo%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Aasync%20def%20_()%3A%0A%20%20%20%20%23%20Get%20the%20current%20notebook%20URL%2C%20drop%20the%20'blob'%20URL%20components%20that%20seem%20to%20be%20added%2C%0A%20%20%20%20%23%20and%20add%20the%20buildnumber%20that%20a%20makethedocs%20PR%20build%20seems%20to%20add.%20This%20allows%20to%20load%0A%20%20%20%20%23%20the%20wheel%20both%20locally%20and%20when%20deployed%20to%20makethedocs.%0A%20%20%20%20async%20def%20import_xdsl()%3A%0A%20%20%20%20%20%20%20%20import%20re%0A%20%20%20%20%20%20%20%20from%20urllib.parse%20import%20urlparse%0A%0A%20%20%20%20%20%20%20%20import%20micropip%0A%20%20%20%20%20%20%20%20from%20marimo%20import%20notebook_location%0A%0A%20%20%20%20%20%20%20%20url%20%3D%20str(notebook_location()).replace(%22blob%3A%22%2C%20%22%22)%0A%20%20%20%20%20%20%20%20print(f%22DEBUG%3A%20notebook%20url%20(full)%3A%20%7Burl%7D%22)%0A%0A%20%20%20%20%20%20%20%20url_parsed%20%3D%20urlparse(url)%0A%20%20%20%20%20%20%20%20scheme%20%3D%20url_parsed.scheme%0A%20%20%20%20%20%20%20%20netloc%20%3D%20url_parsed.netloc%0A%0A%20%20%20%20%20%20%20%20print(f%22DEBUG%3A%20notebook%20url%20(parsed)%3A%20%7Burl_parsed%7D%22)%0A%0A%20%20%20%20%20%20%20%20url%20%3D%20re.sub(%0A%20%20%20%20%20%20%20%20%20%20%20%20%22(%5B%5E%2F%5D)%2F(%5Ba-f0-9-%5D%2B-%5Ba-f0-9-%5D%2B-%5Ba-f0-9-%5D%2B-%5Ba-f0-9-%5D%2B)%22%2C%20%22%5C%5C1%2F%22%2C%20url%2C%20count%3D1%0A%20%20%20%20%20%20%20%20)%0A%20%20%20%20%20%20%20%20buildnumber%20%3D%20re.sub(%22.*--(%5B0-9%2B%5D%2B).*%22%2C%20%22%5C%5C1%22%2C%20url%2C%20count%3D1)%0A%0A%20%20%20%20%20%20%20%20new_url%20%3D%20scheme%20%2B%20%22%3A%2F%2F%22%20%2B%20netloc%0A%0A%20%20%20%20%20%20%20%20if%20buildnumber%20!%3D%20url%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20new_url%20%3D%20new_url%20%2B%20%22%2F%22%20%2B%20buildnumber%20%2B%20%22%2F%22%0A%0A%20%20%20%20%20%20%20%20print(f%22DEBUG%3A%20notebook%20url%20(trimmed)%3A%20%7Bnew_url%7D%22)%0A%0A%20%20%20%20%20%20%20%20await%20micropip.install(%22xdsl%20%40%20%22%20%2B%20new_url%20%2B%20%22%2Fxdsl-0.0.0-py3-none-any.whl%22)%0A%0A%20%20%20%20await%20import_xdsl()%0A%20%20%20%20from%20xdsl.utils%20import%20marimo%20as%20xmo%0A%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_()%3A%0A%20%20%20%20%23%20xDSL%20should%20be%20available%20in%20the%20environment%0A%20%20%20%20from%20xdsl.dialects.arith%20import%20Arith%0A%20%20%20%20from%20xdsl.dialects.builtin%20import%20Builtin%0A%20%20%20%20from%20xdsl.context%20import%20Context%0A%20%20%20%20from%20xdsl.printer%20import%20Printer%0A%0A%20%20%20%20from%20xdsl.dialects.builtin%20import%20i32%0A%20%20%20%20from%20xdsl.utils.exceptions%20import%20VerifyException%0A%0A%20%20%20%20%23%20Context%2C%20containing%20information%20about%20the%20registered%20dialects%0A%20%20%20%20context%20%3D%20Context()%0A%0A%20%20%20%20%23%20Some%20useful%20dialects%0A%20%20%20%20context.load_dialect(Arith)%0A%20%20%20%20context.load_dialect(Builtin)%0A%0A%20%20%20%20%23%20Printer%20used%20to%20pretty-print%20MLIR%20data%20structures%0A%20%20%20%20printer%20%3D%20Printer()%0A%20%20%20%20return%20Printer%2C%20VerifyException%2C%20i32%2C%20printer%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%20IRDL%20tutorial%0A%0A%20%20%20%20%23%23%20An%20Intermediate%20Representation%20Definition%20Language%20(IRDL)%20for%20SSA%20Compilers%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%20Introduction%0A%0A%20%20%20%20xDSL%20is%20an%20extensible%20compiler%2C%20meaning%20that%20new%20operations%2C%20attributes%2C%20and%20types%20can%20be%20added.%20xDSL%20provides%20an%20embedded%20DSL%2C%20IRDL%2C%20to%20define%20new%20dialects.%0A%20%20%20%20This%20tutorial%20aims%20to%20show%20the%20different%20features%20IRDL%20has%2C%20and%20presents%20examples%20on%20how%20to%20use%20them.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%20Attribute%20constraints%0A%0A%20%20%20%20Attribute%20constraints%20represent%20invariants%20over%20attributes%2C%20and%20are%20an%20important%20concept%20for%20defining%20new%20attributes%20and%20operations.%20In%20practice%2C%20an%20attribute%20constraint%20is%20a%20child%20class%20of%20%60AttrConstraint%60%20that%20implements%20a%20%60verify%60%20method.%20The%20method%20takes%20an%20attribute%20to%20verify%20as%20parameter%2C%20and%20a%20dictionary%20associating%20constraint%20variables%20to%20attributes.%20%60verify%60%20does%20not%20return%20anything%2C%20but%20raises%20an%20exception%20if%20the%20invariant%20is%20not%20respected.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%23%20Any%20Constraint%0A%0A%20%20%20%20An%20%60Any%60%20constraint%20will%20never%20trigger%20an%20exception%2C%20and%20will%20always%20pass%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_()%3A%0A%20%20%20%20from%20xdsl.dialects.builtin%20import%20IndexType%2C%20IntegerType%2C%20StringAttr%2C%20i64%0A%20%20%20%20from%20xdsl.irdl%20import%20AnyAttr%2C%20ConstraintContext%0A%0A%20%20%20%20%23%20Construct%20the%20constraint%0A%20%20%20%20any_constraint%20%3D%20AnyAttr()%0A%0A%20%20%20%20%23%20This%20will%20pass%20without%20triggering%20an%20exception%0A%20%20%20%20any_constraint.verify(i64%2C%20ConstraintContext())%0A%20%20%20%20any_constraint.verify(StringAttr(%22ga%22)%2C%20ConstraintContext())%0A%20%20%20%20return%20ConstraintContext%2C%20IndexType%2C%20IntegerType%2C%20StringAttr%2C%20i64%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%23%20Equality%20Constraint%0A%0A%20%20%20%20An%20equality%20constraint%20ensures%20that%20the%20attribute%20is%20equal%20to%20one%20provided%20to%20the%20constraint%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(ConstraintContext%2C%20i32%2C%20i64)%3A%0A%20%20%20%20from%20xdsl.irdl%20import%20EqAttrConstraint%0A%0A%20%20%20%20%23%20Construct%20the%20constraint%0A%20%20%20%20eq_constraint%20%3D%20EqAttrConstraint(i64)%0A%0A%20%20%20%20%23%20This%20will%20pass%20without%20triggering%20an%20exception%0A%20%20%20%20eq_constraint.verify(i64%2C%20ConstraintContext())%0A%0A%20%20%20%20%23%20This%20will%20trigger%20an%20exception%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20eq_constraint.verify(i32%2C%20ConstraintContext())%0A%20%20%20%20except%20Exception%20as%20e%3A%0A%20%20%20%20%20%20%20%20print(e)%0A%20%20%20%20return%20(EqAttrConstraint%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%23%20Base%20Attribute%20Constraint%0A%0A%20%20%20%20A%20base%20attribute%20constraint%20ensures%20that%20the%20attribute%20base%20type%20is%20equal%20to%20an%20expected%20attribute%20base%20type%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(ConstraintContext%2C%20StringAttr%2C%20VerifyException%2C%20i32)%3A%0A%20%20%20%20from%20xdsl.dialects.builtin%20import%20IntAttr%0A%20%20%20%20from%20xdsl.irdl%20import%20BaseAttr%0A%0A%20%20%20%20%23%20Construct%20the%20constraint%0A%20%20%20%20base_constraint%20%3D%20BaseAttr(StringAttr)%0A%0A%20%20%20%20%23%20This%20will%20pass%20without%20triggering%20an%20exception%0A%20%20%20%20base_constraint.verify(StringAttr(%22ga%22)%2C%20ConstraintContext())%0A%20%20%20%20base_constraint.verify(StringAttr(%22bu%22)%2C%20ConstraintContext())%0A%0A%20%20%20%20%23%20This%20will%20trigger%20an%20exception%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20base_constraint.verify(i32%2C%20ConstraintContext())%0A%20%20%20%20except%20VerifyException%20as%20e%3A%0A%20%20%20%20%20%20%20%20print(e)%0A%20%20%20%20return%20BaseAttr%2C%20IntAttr%2C%20base_constraint%0A%0A%0A%40app.cell%0Adef%20_(ConstraintContext%2C%20IntAttr%2C%20VerifyException%2C%20base_constraint)%3A%0A%20%20%20%20%23%20This%20too%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20base_constraint.verify(IntAttr(3)%2C%20ConstraintContext())%0A%20%20%20%20except%20VerifyException%20as%20e%3A%0A%20%20%20%20%20%20%20%20print(e)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%23%20Attribute%20Constraint%20Coercion%0A%0A%20%20%20%20To%20simplify%20the%20definitions%20of%20constraints%2C%20constraint%20constructors%20expecting%20an%20attribute%20constraints%20will%20coerce%20%60Attribute%60%20to%20an%20equality%20attribute%20constraint%2C%20and%20will%20coerce%20an%20%60Attribute%60%20type%20to%20a%20base%20attribute%20constraint.%20this%20is%20done%20using%20the%20%60irdl_to_attr_constraint%60%20function%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(BaseAttr%2C%20EqAttrConstraint%2C%20StringAttr%2C%20i32)%3A%0A%20%20%20%20from%20xdsl.irdl%20import%20AnyOf%2C%20irdl_to_attr_constraint%0A%0A%20%20%20%20assert%20irdl_to_attr_constraint(i32)%20%3D%3D%20EqAttrConstraint(i32)%0A%20%20%20%20assert%20irdl_to_attr_constraint(StringAttr)%20%3D%3D%20BaseAttr(StringAttr)%0A%20%20%20%20assert%20AnyOf.get(i32%2C%20StringAttr)%20%3D%3D%20AnyOf((EqAttrConstraint(i32)%2C%20BaseAttr(StringAttr)))%0A%20%20%20%20return%20AnyOf%2C%20irdl_to_attr_constraint%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%23%20Or%20Constraint%0A%0A%20%20%20%20An%20%60or%60%20constraint%20ensures%20that%20one%20of%20the%20given%20attribute%20constraints%20is%20satisfied%20by%20the%20attribute%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(AnyOf%2C%20ConstraintContext%2C%20StringAttr%2C%20i32%2C%20i64)%3A%0A%20%20%20%20%23%20Construct%20the%20constraint.%20Note%20that%20we%20are%20using%20the%20coercion%20defined%20previously.%0A%20%20%20%20or_constraint%20%3D%20AnyOf.get(i32%2C%20StringAttr)%0A%0A%20%20%20%20%23%20This%20will%20pass%20without%20triggering%20an%20exception%2C%20since%20the%20first%20constraint%20is%20satisfied%0A%20%20%20%20or_constraint.verify(i32%2C%20ConstraintContext())%0A%0A%20%20%20%20%23%20This%20will%20pass%20without%20triggering%20an%20exception%2C%20since%20the%20second%20constraint%20is%20satisfied%0A%20%20%20%20or_constraint.verify(StringAttr(%22ga%22)%2C%20ConstraintContext())%0A%20%20%20%20or_constraint.verify(StringAttr(%22bu%22)%2C%20ConstraintContext())%0A%0A%20%20%20%20%23%20This%20will%20trigger%20an%20exception%2C%20since%20none%20of%20the%20constraints%20are%20satisfied%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20or_constraint.verify(i64%2C%20ConstraintContext())%0A%20%20%20%20except%20Exception%20as%20e%3A%0A%20%20%20%20%20%20%20%20print(e)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%23%20Parametric%20Attribute%20Constraint%0A%0A%20%20%20%20A%20parametric%20attribute%20constraint%20is%20satisfied%20by%20parametric%20attributes%20of%20a%20certain%20base%20type.%20In%20addition%2C%20parametric%20attribute%20constraints%20specify%20constraints%20for%20each%20of%20the%20parameters%20of%20the%20attribute%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(ConstraintContext%2C%20IntAttr%2C%20VerifyException%2C%20i32%2C%20i64)%3A%0A%20%20%20%20from%20xdsl.dialects.builtin%20import%20IntegerAttr%0A%20%20%20%20from%20xdsl.irdl%20import%20ParamAttrConstraint%0A%0A%20%20%20%20%23%20Construct%20the%20constraint.%20Note%20that%20we%20are%20using%20the%20coercion%20defined%20previously.%0A%20%20%20%20param_constraint%20%3D%20ParamAttrConstraint.get(IntegerAttr%2C%20IntAttr%2C%20i32)%0A%0A%20%20%20%20%23%20This%20will%20pass%20without%20triggering%20an%20exception.%0A%20%20%20%20param_constraint.verify(IntegerAttr(IntAttr(42)%2C%20i32)%2C%20ConstraintContext())%0A%20%20%20%20param_constraint.verify(IntegerAttr(IntAttr(23)%2C%20i32)%2C%20ConstraintContext())%0A%0A%20%20%20%20%23%20This%20will%20trigger%20an%20exception%20since%20the%20attribute%20type%20is%20not%20the%20expected%20one%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20param_constraint.verify(i64%2C%20ConstraintContext())%0A%20%20%20%20except%20VerifyException%20as%20e%3A%0A%20%20%20%20%20%20%20%20print(e)%0A%20%20%20%20return%20IntegerAttr%2C%20param_constraint%0A%0A%0A%40app.cell%0Adef%20_(%0A%20%20%20%20ConstraintContext%2C%0A%20%20%20%20IntAttr%2C%0A%20%20%20%20IntegerAttr%2C%0A%20%20%20%20VerifyException%2C%0A%20%20%20%20i64%2C%0A%20%20%20%20param_constraint%2C%0A)%3A%0A%20%20%20%20%23%20This%20will%20trigger%20an%20exception%20since%20the%20second%20parameter%20constraint%20is%20not%20satisfied%0A%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20param_constraint.verify(IntegerAttr(IntAttr(42)%2C%20i64)%2C%20ConstraintContext())%0A%20%20%20%20except%20VerifyException%20as%20e%3A%0A%20%20%20%20%20%20%20%20print(e)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%23%20Constraint%20Variables%0A%0A%20%20%20%20Constraint%20variables%20are%20used%20to%20specify%20equality%20between%20attributes%20in%20operation%20and%20attribute%20definitions.%20They%20also%20contain%20a%20constraint%20that%20must%20be%20satisfied.%0A%20%20%20%20The%20first%20time%20a%20constraint%20variable%20is%20used%2C%20it%20will%20check%20that%20the%20constraint%20is%20satisfied.%20If%20it%20is%20satisfied%2C%20it%20sets%20the%20variable%20to%20the%20given%20attribute.%20If%20a%20constraint%20variable%20is%20already%20set%2C%20it%20will%20check%20that%20the%20given%20attribute%20is%20equal%20to%20the%20one%20already%20set.%20Two%20constraint%20variables%20with%20the%20same%20name%20are%20considered%20equal%2C%20and%20are%20expected%20to%20carry%20the%20same%20constraint.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(BaseAttr%2C%20ConstraintContext%2C%20IndexType%2C%20IntegerType%2C%20VerifyException)%3A%0A%20%20%20%20from%20xdsl.irdl%20import%20VarConstraint%0A%0A%20%20%20%20var_constraint%20%3D%20VarConstraint(%22T%22%2C%20BaseAttr(IntegerType))%0A%20%20%20%20constraint_context%20%3D%20ConstraintContext()%0A%0A%20%20%20%20%23%20The%20underlying%20constraint%20is%20not%20satisfied%20by%20the%20given%20attribute%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20var_constraint.verify(IndexType()%2C%20constraint_context)%0A%20%20%20%20except%20VerifyException%20as%20e%3A%0A%20%20%20%20%20%20%20%20print(e)%0A%20%20%20%20return%20VarConstraint%2C%20constraint_context%2C%20var_constraint%0A%0A%0A%40app.cell%0Adef%20_(constraint_context%2C%20i32%2C%20var_constraint)%3A%0A%20%20%20%20%23%20The%20constraint%20sets%20the%20constraint%20variable%20if%20the%20attribute%20verifies%0A%20%20%20%20var_constraint.verify(i32%2C%20constraint_context)%0A%20%20%20%20print(constraint_context.get_variable(%22T%22))%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(VerifyException%2C%20constraint_context%2C%20i64%2C%20var_constraint)%3A%0A%20%20%20%20%23%20Since%20the%20variable%20is%20set%2C%20the%20constraint%20can%20now%20only%20be%20satisfied%20by%20the%20same%20attribute%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20var_constraint.verify(i64%2C%20constraint_context)%0A%20%20%20%20except%20VerifyException%20as%20e%3A%0A%20%20%20%20%20%20%20%20print(e)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%23%20Custom%20Constraints%0A%0A%20%20%20%20Users%20can%20define%20their%20own%20constraints%20for%20their%20own%20types.%20For%20instance%2C%20here%20is%20the%20definition%20of%20the%20%60ArrayOfConstraint%60%20constraint%2C%20which%20expects%20an%20%60ArrayAttr%60%20where%20all%20elements%20of%20the%20array%20satisfy%20a%20specific%20constraint%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(ConstraintContext%2C%20IntAttr%2C%20VerifyException%2C%20irdl_to_attr_constraint)%3A%0A%20%20%20%20from%20dataclasses%20import%20dataclass%0A%0A%20%20%20%20from%20xdsl.dialects.builtin%20import%20ArrayAttr%0A%20%20%20%20from%20xdsl.ir%20import%20Attribute%0A%20%20%20%20from%20xdsl.irdl%20import%20AttrConstraint%0A%20%20%20%20from%20xdsl.utils.hints%20import%20isa%0A%20%20%20%20from%20typing_extensions%20import%20TypeVar%0A%0A%20%20%20%20%40dataclass(frozen%3DTrue)%0A%20%20%20%20class%20ArrayOfConstraint(AttrConstraint)%3A%0A%20%20%20%20%20%20%20%20%23%20The%20constraint%20that%20needs%20to%20be%20satisfied%20by%20all%20elements%20of%20the%20array%0A%20%20%20%20%20%20%20%20elem_constr%3A%20AttrConstraint%0A%0A%20%20%20%20%20%20%20%20%23%20The%20custom%20init%20applies%20the%20attribute%20constraint%20coercion%0A%20%20%20%20%20%20%20%20def%20__init__(self%2C%20constr%3A%20Attribute%20%7C%20type%5BAttribute%5D%20%7C%20AttrConstraint)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20object.__setattr__(self%2C%20%22elem_constr%22%2C%20irdl_to_attr_constraint(constr))%0A%0A%20%20%20%20%20%20%20%20%23%20Check%20that%20an%20attribute%20satisfies%20the%20constraints%0A%20%20%20%20%20%20%20%20def%20verify(self%2C%20attr%3A%20Attribute%2C%20constraint_context%3A%20ConstraintContext)%20-%3E%20None%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%23%20We%20first%20check%20that%20the%20attribute%20is%20an%20ArrayAttr%0A%20%20%20%20%20%20%20%20%20%20%20%20if%20not%20isa(attr%2C%20ArrayAttr%5BAttribute%5D)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20raise%20VerifyException(f%22expected%20attribute%20ArrayData%20but%20got%20%7Battr%7D%22)%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20%23%20We%20check%20the%20constraint%20for%20all%20elements%20in%20the%20array%0A%20%20%20%20%20%20%20%20%20%20%20%20for%20e%20in%20attr.data%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20self.elem_constr.verify(e%2C%20constraint_context)%0A%0A%20%20%20%20%20%20%20%20def%20mapping_type_vars(self%2C%20type_var_mapping%3A%20dict%5BTypeVar%2C%20AttrConstraint%5D)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20return%20ArrayOfConstraint(self.elem_constr.mapping_type_vars(type_var_mapping))%0A%0A%0A%20%20%20%20array_constraint%20%3D%20ArrayOfConstraint(IntAttr)%0A%0A%20%20%20%20%23%20This%20will%20pass%20without%20triggering%20an%20exception%0A%20%20%20%20array_constraint.verify(ArrayAttr(%5BIntAttr(42)%5D)%2C%20ConstraintContext())%0A%20%20%20%20array_constraint.verify(ArrayAttr(%5BIntAttr(3)%2C%20IntAttr(7)%5D)%2C%20ConstraintContext())%0A%20%20%20%20return%20ArrayAttr%2C%20array_constraint%0A%0A%0A%40app.cell%0Adef%20_(ConstraintContext%2C%20array_constraint%2C%20i32)%3A%0A%20%20%20%20%23%20This%20will%20trigger%20an%20exception%2C%20since%20the%20attribute%20is%20not%20an%20array%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20array_constraint.verify(i32%2C%20ConstraintContext())%0A%20%20%20%20except%20Exception%20as%20e%3A%0A%20%20%20%20%20%20%20%20print(e)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(ArrayAttr%2C%20ConstraintContext%2C%20IntAttr%2C%20StringAttr%2C%20array_constraint)%3A%0A%20%20%20%20%23%20This%20will%20trigger%20an%20exception%2C%20since%20the%20array%20contains%20attribute%20that%20do%20not%20satisfies%20the%20constraint%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20array_constraint.verify(%0A%20%20%20%20%20%20%20%20%20%20%20%20ArrayAttr(%5BIntAttr(42)%2C%20StringAttr(%22ga%22)%5D)%2C%20ConstraintContext()%0A%20%20%20%20%20%20%20%20)%0A%20%20%20%20except%20Exception%20as%20e%3A%0A%20%20%20%20%20%20%20%20print(e)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%20Attribute%20Definition%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%23%20Data%20Attributes%0A%0A%20%20%20%20%60Data%60%20attribute%20types%20are%20defined%20by%20inheriting%20the%20python%20%60Data%60%20class.%20Each%20data%20attribute%20definition%20should%20define%20a%20%60name%60%2C%20and%20two%20methods%20for%20conversion%20to%20a%20string%20representation.%20Here%20is%20for%20example%20the%20definition%20of%20%60IntAttr%60%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(Printer%2C%20printer)%3A%0A%20%20%20%20from%20xdsl.ir%20import%20Data%0A%20%20%20%20from%20xdsl.irdl%20import%20irdl_attr_definition%0A%20%20%20%20from%20xdsl.parser%20import%20AttrParser%0A%0A%0A%20%20%20%20%40irdl_attr_definition%0A%20%20%20%20class%20MyIntAttr(Data%5Bint%5D)%3A%0A%20%20%20%20%20%20%20%20name%20%3D%20%22test.my_int%22%0A%0A%20%20%20%20%20%20%20%20%40classmethod%0A%20%20%20%20%20%20%20%20def%20parse_parameter(cls%2C%20parser%3A%20AttrParser)%20-%3E%20int%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20data%20%3D%20parser.parse_integer()%0A%20%20%20%20%20%20%20%20%20%20%20%20return%20data%0A%0A%20%20%20%20%20%20%20%20def%20print_parameter(self%2C%20printer%3A%20Printer)%20-%3E%20None%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20printer.print_string(f%22%7Bself.data%7D%22)%0A%0A%0A%20%20%20%20MyIntAttr(3).print_parameter(printer)%0A%20%20%20%20return%20(irdl_attr_definition%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%23%20Parametrized%20Attributes%0A%0A%20%20%20%20%60ParametrizedAttribute%60%20attribute%20types%20are%20defined%20using%20the%20%60irdl_attr_definition%60%20decorator%20on%20a%20class.%20Such%20class%20should%20contain%20a%20%60name%60%20field%20specifying%20the%20attribute%20name.%20Parameters%20are%20added%20to%20attribute%20definitions%20by%20defining%20fields%20with%20a%20type%2C%20and%20optionally%20with%20a%20%60param_def%60.%20The%20field%20names%20correspond%20to%20the%20parameter%20names%2C%20and%20%60param_def%60%20contains%20an%20optional%20constraint%20that%20should%20be%20respected%20by%20this%20parameter.%20The%20order%20of%20the%20fields%20correspond%20to%20the%20order%20of%20the%20parameters%20when%20using%20the%20attribute.%20Upon%20construction%20of%20an%20attribute%2C%20all%20constraints%20will%20be%20checked%2C%20and%20an%20exception%20will%20be%20raised%20if%20the%20invariants%20are%20not%20satisfied.%0A%0A%20%20%20%20Here%20is%20an%20example%20of%20an%20integer%20type%20definition%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(EqAttrConstraint%2C%20IntAttr%2C%20StringAttr%2C%20irdl_attr_definition)%3A%0A%20%20%20%20from%20xdsl.ir%20import%20ParametrizedAttribute%0A%20%20%20%20from%20xdsl.irdl%20import%20param_def%0A%0A%0A%20%20%20%20%23%20Represent%20an%20integer%20type%20with%20a%20given%20bitwidth%0A%20%20%20%20%40irdl_attr_definition%0A%20%20%20%20class%20MyIntegerType(ParametrizedAttribute)%3A%0A%20%20%20%20%20%20%20%20%23%20Name%20of%20the%20type.%20This%20is%20used%20for%20printing%20and%20parsing.%0A%20%20%20%20%20%20%20%20name%20%3D%20%22test.integer_type%22%0A%0A%20%20%20%20%20%20%20%20%23%20Only%20parameter%20of%20the%20type%2C%20with%20an%20%60EqAttrConstraint%60%20constraint.%0A%20%20%20%20%20%20%20%20width%3A%20IntAttr%20%3D%20param_def(EqAttrConstraint(IntAttr(32)))%0A%0A%0A%20%20%20%20my_i32%20%3D%20MyIntegerType(IntAttr(32))%0A%0A%20%20%20%20%23%20This%20will%20trigger%20an%20exception%2C%20since%20the%20attribute%20is%20not%20an%20IntAttr%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20MyIntegerType(StringAttr(%22not%20an%20int%22))%0A%20%20%20%20except%20Exception%20as%20e%3A%0A%20%20%20%20%20%20%20%20print(e)%0A%0A%20%20%20%20%23%20This%20will%20trigger%20an%20exception%2C%20since%20the%20attribute%20is%20not%20equal%20to%20the%20expected%20value%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20MyIntegerType(IntAttr(64))%0A%20%20%20%20except%20Exception%20as%20e%3A%0A%20%20%20%20%20%20%20%20print(e)%0A%0A%20%20%20%20print(MyIntegerType(IntAttr(32)))%0A%20%20%20%20return%20(my_i32%2C)%0A%0A%0A%40app.cell%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20Each%20parameter%20can%20be%20accessed%20using%20the%20%60parameters%60%20field.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(my_i32)%3A%0A%20%20%20%20my_i32.parameters%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20Our%20attribute%20definition%20also%20defines%20accessors%20for%20each%20parameter%20based%20on%20the%20name%20given%20in%20the%20%60param_def%60%20field%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(my_i32)%3A%0A%20%20%20%20my_i32.width%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%20Operation%20Definition%0A%0A%20%20%20%20Operations%20are%20defined%20similarly%20to%20%60ParametrizedAttribute%60%2C%20by%20using%20the%20%60irdl_op_definition%60%20decorator.%20The%20decorator%20allows%20the%20definition%20of%20expected%20operands%2C%20results%2C%20attributes%2C%20and%20regions.%20Each%20definition%20should%20contain%20a%20%60name%60%20static%20field%2C%20which%20is%20used%20for%20parsing%20and%20printing%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(printer)%3A%0A%20%20%20%20from%20xdsl.irdl%20import%20IRDLOperation%2C%20irdl_op_definition%0A%0A%0A%20%20%20%20%40irdl_op_definition%0A%20%20%20%20class%20MyEmptyOp(IRDLOperation)%3A%0A%20%20%20%20%20%20%20%20name%20%3D%20%22my_dialect.my_op%22%0A%0A%0A%20%20%20%20my_op%20%3D%20MyEmptyOp.build()%0A%20%20%20%20printer.print_op(my_op)%0A%20%20%20%20return%20IRDLOperation%2C%20irdl_op_definition%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%23%20Operands%20and%20Results%0A%0A%20%20%20%20Operands%20and%20results%20are%20added%20using%20fields%20containing%20%60Operand%60%20and%20%60OpResult%60%2C%20which%20each%20contain%20an%20attribute%20constraint.%20The%20order%20correspond%20to%20the%20operand%20and%20result%20order%2C%20and%20the%20constraint%20applies%20on%20the%20SSA%20variable%20type.%0A%0A%20%20%20%20Here%20is%20an%20example%20of%20an%20operation%20defining%20operands%20and%20a%20result%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(IRDLOperation%2C%20IntegerAttr%2C%20i32%2C%20irdl_op_definition%2C%20printer)%3A%0A%20%20%20%20from%20typing%20import%20Annotated%0A%0A%20%20%20%20from%20xdsl.dialects.arith%20import%20ConstantOp%0A%20%20%20%20from%20xdsl.ir%20import%20OpResult%0A%20%20%20%20from%20xdsl.irdl%20import%20Operand%2C%20operand_def%2C%20result_def%0A%0A%0A%20%20%20%20%40irdl_op_definition%0A%20%20%20%20class%20Addi32Op(IRDLOperation)%3A%0A%20%20%20%20%20%20%20%20name%20%3D%20%22arith.addi32%22%0A%0A%20%20%20%20%20%20%20%20%23%20Definition%20of%20operands%20and%20results.%0A%20%20%20%20%20%20%20%20input1%20%3D%20operand_def(i32)%0A%20%20%20%20%20%20%20%20input2%20%3D%20operand_def(i32)%0A%20%20%20%20%20%20%20%20output%20%3D%20result_def(i32)%0A%0A%0A%20%20%20%20i32_ssa_var%20%3D%20ConstantOp(IntegerAttr(62%2C%2032)%2C%20i32)%0A%20%20%20%20my_addi32%20%3D%20Addi32Op.build(%0A%20%20%20%20%20%20%20%20operands%3D%5Bi32_ssa_var.result%2C%20i32_ssa_var.result%5D%2C%20result_types%3D%5Bi32%5D%0A%20%20%20%20)%0A%20%20%20%20printer.print_op(i32_ssa_var)%0A%20%20%20%20print()%0A%20%20%20%20printer.print_op(my_addi32)%0A%20%20%20%20return%20(%0A%20%20%20%20%20%20%20%20Addi32Op%2C%0A%20%20%20%20%20%20%20%20ConstantOp%2C%0A%20%20%20%20%20%20%20%20i32_ssa_var%2C%0A%20%20%20%20%20%20%20%20my_addi32%2C%0A%20%20%20%20%20%20%20%20operand_def%2C%0A%20%20%20%20%20%20%20%20result_def%2C%0A%20%20%20%20)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20The%20operation%20expects%20exactly%20the%20number%20of%20operands%20and%20results%20defined%20and%20checks%20that%20operands%20and%20results%20satisfy%20their%20invariants%20using%20the%20%60verify%60%20method.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(my_addi32)%3A%0A%20%20%20%20%23%20This%20will%20pass%20since%20the%20invariants%20are%20satisfied%0A%20%20%20%20my_addi32.verify()%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(Addi32%2C%20i32%2C%20i32_ssa_var)%3A%0A%20%20%20%20%23%20Wrong%20number%20of%20operands%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20bad_addi32_a%20%3D%20Addi32.build(operands%3D%5Bi32_ssa_var%5D%2C%20result_types%3D%5Bi32%5D)%0A%20%20%20%20except%20Exception%20as%20e%3A%0A%20%20%20%20%20%20%20%20print(e)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(Addi32Op%2C%20i32%2C%20i32_ssa_var)%3A%0A%20%20%20%20%23%20Wrong%20number%20of%20results%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20bad_addi32_b%20%3D%20Addi32Op.build(%0A%20%20%20%20%20%20%20%20%20%20%20%20operands%3D%5Bi32_ssa_var%2C%20i32_ssa_var%5D%2C%20result_types%3D%5Bi32%2C%20i32%5D%0A%20%20%20%20%20%20%20%20)%0A%20%20%20%20except%20Exception%20as%20e%3A%0A%20%20%20%20%20%20%20%20print(e)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(Addi32Op%2C%20i32_ssa_var%2C%20i64)%3A%0A%20%20%20%20%23%20Wrong%20result%20type%0A%20%20%20%20bad_addi32_c%20%3D%20Addi32Op.build(operands%3D%5Bi32_ssa_var%2C%20i32_ssa_var%5D%2C%20result_types%3D%5Bi64%5D)%0A%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20bad_addi32_c.verify()%0A%20%20%20%20except%20Exception%20as%20e%3A%0A%20%20%20%20%20%20%20%20print(e)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20As%20with%20%60ParametrizedAttribute%60%2C%20each%20operand%20and%20result%20definition%20defines%20accessors%20to%20easily%20access%20operands%20and%20results%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(my_addi32)%3A%0A%20%20%20%20assert%20my_addi32.input1%20%3D%3D%20my_addi32.operands%5B0%5D%0A%20%20%20%20assert%20my_addi32.input2%20%3D%3D%20my_addi32.operands%5B1%5D%0A%20%20%20%20assert%20my_addi32.output%20%3D%3D%20my_addi32.results%5B0%5D%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%23%20Constraint%20Variables%0A%0A%20%20%20%20Constraint%20variables%20can%20directly%20be%20used%20in%20Operation%20and%20Attribute%20definitions%20by%20using%20a%20%60TypeAlias%60%20annotated%20with%20a%20%60ConstraintVar%60.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(%0A%20%20%20%20IRDLOperation%2C%0A%20%20%20%20IntegerType%2C%0A%20%20%20%20VarConstraint%2C%0A%20%20%20%20i32%2C%0A%20%20%20%20i32_ssa_var%2C%0A%20%20%20%20i64%2C%0A%20%20%20%20irdl_op_definition%2C%0A%20%20%20%20operand_def%2C%0A%20%20%20%20result_def%2C%0A)%3A%0A%20%20%20%20from%20typing%20import%20ClassVar%0A%20%20%20%20from%20xdsl.irdl%20import%20base%0A%0A%0A%20%20%20%20%40irdl_op_definition%0A%20%20%20%20class%20BinaryOp(IRDLOperation)%3A%0A%20%20%20%20%20%20%20%20name%20%3D%20%22test.binary_op%22%0A%0A%20%20%20%20%20%20%20%20T%3A%20ClassVar%20%3D%20VarConstraint(%22T%22%2C%20base(IntegerType))%0A%0A%20%20%20%20%20%20%20%20lhs%20%3D%20operand_def(T)%0A%20%20%20%20%20%20%20%20rhs%20%3D%20operand_def(T)%0A%20%20%20%20%20%20%20%20result%20%3D%20result_def(T)%0A%0A%0A%20%20%20%20op%20%3D%20BinaryOp.build(operands%3D%5Bi32_ssa_var%2C%20i32_ssa_var%5D%2C%20result_types%3D%5Bi32%5D)%0A%20%20%20%20op.verify()%0A%0A%20%20%20%20op_incorrect%20%3D%20BinaryOp.build(operands%3D%5Bi32_ssa_var%2C%20i32_ssa_var%5D%2C%20result_types%3D%5Bi64%5D)%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20op_incorrect.verify()%0A%20%20%20%20except%20Exception%20as%20e%3A%0A%20%20%20%20%20%20%20%20print(e)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%23%20Variadic%20Operands%20and%20Results%0A%0A%20%20%20%20Operand%20and%20result%20definitions%20can%20be%20defined%20variadic%2C%20meaning%20that%20their%20definition%20can%20have%20different%20numbers%20of%20operands%20or%20results.%20Variadic%20definitions%20are%20defined%20with%20%60var_operand_def%60%20and%20%60var_result_def%60.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(%0A%20%20%20%20ConstantOp%2C%0A%20%20%20%20IRDLOperation%2C%0A%20%20%20%20IntegerAttr%2C%0A%20%20%20%20i32%2C%0A%20%20%20%20irdl_op_definition%2C%0A%20%20%20%20printer%2C%0A%20%20%20%20result_def%2C%0A)%3A%0A%20%20%20%20from%20xdsl.irdl%20import%20var_operand_def%0A%0A%0A%20%20%20%20%40irdl_op_definition%0A%20%20%20%20class%20AddVariadicOp(IRDLOperation)%3A%0A%20%20%20%20%20%20%20%20name%20%3D%20%22test.add_variadic%22%0A%20%20%20%20%20%20%20%20ops%20%3D%20var_operand_def(i32)%0A%20%20%20%20%20%20%20%20res%20%3D%20result_def(i32)%0A%0A%0A%20%20%20%20i32_ssa_var_b%20%3D%20ConstantOp(IntegerAttr(62%2C%2032)%2C%20i32)%0A%20%20%20%20add_op%20%3D%20AddVariadicOp.build(operands%3D%5B%5Bi32_ssa_var_b%5D%20*%203%5D%2C%20result_types%3D%5Bi32%5D)%0A%20%20%20%20printer.print_op(i32_ssa_var_b)%0A%20%20%20%20print()%0A%20%20%20%20printer.print_op(add_op)%0A%20%20%20%20return%20add_op%2C%20var_operand_def%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20Generated%20accessors%20return%20a%20list%20of%20SSA%20variables%20instead%20of%20a%20single%20variable%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(add_op)%3A%0A%20%20%20%20print(len(add_op.ops))%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20Since%20it%20may%20be%20ambiguous%2C%20it%20is%20not%20possible%20to%20simply%20define%20two%20variadic%20operands%2C%20or%20two%20variadic%20results.%20To%20do%20so%2C%20the%20operation%20definition%20requires%20the%20%60AttrSizedOperandSegments%60%20or%20%60AttrSizedResultSegments%60%20IRDL%20option%2C%20which%20require%20the%20operation%20to%20contain%20a%20%60operandSegmentSizes%60%20or%20%60resultSegmentSizes%60%20attribute%2C%20containing%20the%20size%20of%20the%20variadic%20operands%2C%20and%20variadic%20results.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(%0A%20%20%20%20ConstantOp%2C%0A%20%20%20%20IRDLOperation%2C%0A%20%20%20%20IntegerAttr%2C%0A%20%20%20%20i32%2C%0A%20%20%20%20irdl_op_definition%2C%0A%20%20%20%20result_def%2C%0A%20%20%20%20var_operand_def%2C%0A)%3A%0A%20%20%20%20from%20xdsl.dialects.builtin%20import%20VectorType%0A%20%20%20%20from%20xdsl.irdl%20import%20AttrSizedOperandSegments%0A%0A%0A%20%20%20%20%40irdl_op_definition%0A%20%20%20%20class%20AddVariadic2Op(IRDLOperation)%3A%0A%20%20%20%20%20%20%20%20name%20%3D%20%22test.add_variadic%22%0A%20%20%20%20%20%20%20%20ops1%20%3D%20var_operand_def(i32)%0A%20%20%20%20%20%20%20%20ops2%20%3D%20var_operand_def(i32)%0A%20%20%20%20%20%20%20%20res%20%3D%20result_def(i32)%0A%0A%20%20%20%20%20%20%20%20irdl_options%20%3D%20(AttrSizedOperandSegments()%2C)%0A%0A%0A%20%20%20%20i32_ssa_var_c%20%3D%20ConstantOp(IntegerAttr(62%2C%2032)%2C%20i32)%0A%20%20%20%20add_op2%20%3D%20AddVariadic2Op.build(%0A%20%20%20%20%20%20%20%20operands%3D%5B%5Bi32_ssa_var_c%5D%20*%202%2C%20%5Bi32_ssa_var_c%5D%5D%2C%0A%20%20%20%20%20%20%20%20result_types%3D%5Bi32%5D%2C%0A%20%20%20%20%20%20%20%20attributes%3D%7B%22operandSegmentSizes%22%3A%20VectorType(i32%2C%20%5B2%2C%201%5D)%7D%2C%0A%20%20%20%20)%0A%20%20%20%20print(%22Length%20of%20add_op2.ops1%3A%22%2C%20len(add_op2.ops1))%0A%20%20%20%20print(%22Length%20of%20add_op2.ops2%3A%22%2C%20len(add_op2.ops2))%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20In%20addition%20to%20variadic%20operands%20and%20results%2C%20IRDL%20also%20allows%20the%20definition%20of%20optional%20operands%20and%20results.%20Optional%20operands%20and%20results%20are%20essentially%20variadic%20operands%20and%20regions%20that%20are%20constrained%20to%20have%20either%200%20or%201%20elements.%20They%20are%20defined%20using%20%60OptOperand%60%20and%20%60OptResultDef%60%2C%20and%20define%20accessors%20returning%20optional%20SSA%20variables.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(%0A%20%20%20%20ConstantOp%2C%0A%20%20%20%20IRDLOperation%2C%0A%20%20%20%20IntegerAttr%2C%0A%20%20%20%20i32%2C%0A%20%20%20%20irdl_op_definition%2C%0A%20%20%20%20operand_def%2C%0A%20%20%20%20result_def%2C%0A)%3A%0A%20%20%20%20from%20xdsl.irdl%20import%20OptOperand%2C%20opt_operand_def%0A%0A%0A%20%20%20%20%40irdl_op_definition%0A%20%20%20%20class%20AddVariadic2Op2(IRDLOperation)%3A%0A%20%20%20%20%20%20%20%20name%20%3D%20%22test.add_optional%22%0A%20%20%20%20%20%20%20%20ops1%20%3D%20operand_def(i32)%0A%20%20%20%20%20%20%20%20ops2%20%3D%20opt_operand_def(i32)%0A%20%20%20%20%20%20%20%20res%20%3D%20result_def(i32)%0A%0A%0A%20%20%20%20i32_ssa_var_d%20%3D%20ConstantOp(IntegerAttr(62%2C%2032)%2C%20i32)%0A%20%20%20%20add_op3%20%3D%20AddVariadic2Op2.build(%0A%20%20%20%20%20%20%20%20operands%3D%5Bi32_ssa_var_d%2C%20%5Bi32_ssa_var_d%5D%5D%2C%20result_types%3D%5Bi32%5D%0A%20%20%20%20)%0A%20%20%20%20print(add_op3.ops2)%0A%0A%20%20%20%20add_op4%20%3D%20AddVariadic2Op2.build(operands%3D%5Bi32_ssa_var_d%2C%20%5B%5D%5D%2C%20result_types%3D%5Bi32%5D)%0A%20%20%20%20print(add_op4.ops2)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%23%20Attributes%20Definition%0A%0A%20%20%20%20Attribute%20definitions%20are%20defined%20using%20%60OpAttr%60.%20The%20field%20name%20correspond%20to%20the%20expected%20attribute%20name.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(IRDLOperation%2C%20StringAttr%2C%20irdl_op_definition%2C%20printer)%3A%0A%20%20%20%20from%20xdsl.irdl%20import%20attr_def%0A%0A%0A%20%20%20%20%40irdl_op_definition%0A%20%20%20%20class%20StringAttrOp(IRDLOperation)%3A%0A%20%20%20%20%20%20%20%20name%20%3D%20%22test.string_attr_op%22%0A%20%20%20%20%20%20%20%20value%20%3D%20attr_def(StringAttr)%0A%0A%0A%20%20%20%20my_attr_op%20%3D%20StringAttrOp.build(attributes%3D%7B%22value%22%3A%20StringAttr(%22ga%22)%7D)%0A%20%20%20%20my_attr_op.verify()%0A%20%20%20%20printer.print_op(my_attr_op)%0A%20%20%20%20return%20(StringAttrOp%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20The%20defined%20verifier%20ensures%20that%20the%20attribute%20is%20present%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(StringAttrOp)%3A%0A%20%20%20%20my_attr_op2%20%3D%20StringAttrOp.build()%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20my_attr_op2.verify()%0A%20%20%20%20except%20Exception%20as%20e%3A%0A%20%20%20%20%20%20%20%20print(e)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20And%20expects%20that%20the%20attribute%20respect%20the%20constraint%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(IntAttr%2C%20StringAttrOp)%3A%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20my_attr_op3%20%3D%20StringAttrOp.build(attributes%3D%7B%22value%22%3A%20IntAttr(42)%7D)%0A%20%20%20%20except%20Exception%20as%20e%3A%0A%20%20%20%20%20%20%20%20print(e)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20Note%20that%20contrary%20to%20operands%20and%20results%2C%20other%20attributes%20may%20appear%20in%20an%20operation%2C%20even%20though%20they%20are%20not%20defined%20by%20the%20operation%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(IntAttr%2C%20StringAttr%2C%20StringAttrOp%2C%20printer)%3A%0A%20%20%20%20my_attr_op4%20%3D%20StringAttrOp.build(%0A%20%20%20%20%20%20%20%20attributes%3D%7B%22value%22%3A%20StringAttr(%22ga%22)%2C%20%22other_attr%22%3A%20IntAttr(42)%7D%0A%20%20%20%20)%0A%20%20%20%20my_attr_op4.verify()%0A%20%20%20%20printer.print_op(my_attr_op4)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%23%20Regions%0A%0A%20%20%20%20Regions%20definitions%20are%20defined%20using%20%60Region%60%20or%20%60SingleBlockRegion%60%20annotations.%20The%20second%20definition%20constrains%20the%20region%20to%20contain%20a%20single%20block%2C%20and%20both%20definitions%20allows%20to%20further%20constraint%20the%20region%20by%20giving%20a%20constraint%20for%20the%20entry%20basic%20block%20parameters.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(IRDLOperation%2C%20i32%2C%20irdl_op_definition%2C%20printer)%3A%0A%20%20%20%20from%20xdsl.irdl%20import%20Block%2C%20Region%2C%20region_def%2C%20traits_def%0A%20%20%20%20from%20xdsl.traits%20import%20NoTerminator%0A%0A%0A%20%20%20%20%40irdl_op_definition%0A%20%20%20%20class%20WhileOp(IRDLOperation)%3A%0A%20%20%20%20%20%20%20%20name%20%3D%20%22test.while_op%22%0A%20%20%20%20%20%20%20%20value%20%3D%20region_def()%0A%20%20%20%20%20%20%20%20traits%20%3D%20traits_def(NoTerminator())%0A%0A%0A%20%20%20%20region%20%3D%20Region(Block(arg_types%3D%5Bi32%5D))%0A%20%20%20%20region_op%20%3D%20WhileOp.build(regions%3D%5Bregion%5D)%0A%20%20%20%20region_op.verify()%0A%20%20%20%20printer.print_op(region_op)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(r%22%22%22%0A%20%20%20%20%23%23%23%23%20Additional%20Verifiers%0A%0A%20%20%20%20%60irdl_op_definition%60%20is%20not%20expressive%20enough%20to%20define%20arbitrary%20constraints%2C%20especially%20for%20constraints%20spanning%20over%20multiple%20operand%20and%20result%20definitions.%20To%20circumvent%20that%2C%20definitions%20may%20define%20a%20%60verify_%60%20method%20that%20will%20be%20called%20in%20the%20generated%20verifier%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(%0A%20%20%20%20ConstantOp%2C%0A%20%20%20%20IRDLOperation%2C%0A%20%20%20%20IntegerAttr%2C%0A%20%20%20%20IntegerType%2C%0A%20%20%20%20i32%2C%0A%20%20%20%20i32_ssa_var%2C%0A%20%20%20%20i64%2C%0A%20%20%20%20irdl_op_definition%2C%0A%20%20%20%20operand_def%2C%0A%20%20%20%20result_def%2C%0A)%3A%0A%20%20%20%20from%20xdsl.dialects.arith%20import%20AddiOp%0A%0A%0A%20%20%20%20%40irdl_op_definition%0A%20%20%20%20class%20MyAddiOp(IRDLOperation)%3A%0A%20%20%20%20%20%20%20%20name%20%3D%20%22test.addi%22%0A%20%20%20%20%20%20%20%20input1%20%3D%20operand_def(IntegerType)%0A%20%20%20%20%20%20%20%20input2%20%3D%20operand_def(IntegerType)%0A%20%20%20%20%20%20%20%20output%20%3D%20result_def(IntegerType)%0A%0A%20%20%20%20%20%20%20%20%23%20Ensure%20that%20the%20inputs%20and%20outputs%20have%20the%20same%20type%3A%0A%20%20%20%20%20%20%20%20def%20verify_(self)%20-%3E%20None%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20if%20self.input1.type%20!%3D%20self.input2.type%20or%20self.input2.type%20!%3D%20self.output.type%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20raise%20Exception(%22expect%20all%20input%20and%20output%20types%20to%20be%20equal%22)%0A%0A%0A%20%20%20%20i32_ssa_var_e%20%3D%20ConstantOp(IntegerAttr(62%2C%2032)%2C%20i32)%0A%20%20%20%20add_op5%20%3D%20AddiOp.build(operands%3D%5Bi32_ssa_var_e%2C%20i32_ssa_var_e%5D%2C%20result_types%3D%5Bi32%5D)%0A%20%20%20%20%23%20This%20will%20pass%2C%20since%20all%20operands%20and%20results%20have%20the%20same%20type%0A%20%20%20%20add_op5.verify()%0A%0A%20%20%20%20%23%20This%20will%20raise%20an%20error%2C%20since%20the%20result%20has%20a%20different%20type%20than%20the%20operands%0A%20%20%20%20bad_add_op%20%3D%20AddiOp.build(operands%3D%5Bi32_ssa_var%2C%20i32_ssa_var%5D%2C%20result_types%3D%5Bi64%5D)%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20bad_add_op.verify()%0A%20%20%20%20except%20Exception%20as%20e%3A%0A%20%20%20%20%20%20%20%20print(e)%0A%20%20%20%20return%0A%0A%0Aif%20__name__%20%3D%3D%20%22__main__%22%3A%0A%20%20%20%20app.run()%0A