ir gen

import%20marimo%0A%0A__generated_with%20%3D%20%220.23.15%22%0Aapp%20%3D%20marimo.App()%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%20from%20sympy%20import%20S%2C%20symbols%2C%20Expr%2C%20Add%2C%20Mul%2C%20Sum%2C%20Integer%2C%20Float%2C%20E%2C%20I%2C%20re%2C%20im%2C%20Abs%2C%20Pow%2C%20Rational%2C%20Function%2C%20UnevaluatedExpr%0A%20%20%20%20from%20sympy.core.symbol%20import%20Symbol%0A%0A%20%20%20%20from%20xdsl.ir%20import%20Attribute%2C%20Operation%2C%20SSAValue%2C%20Region%2C%20Block%2C%20ParametrizedAttribute%0A%20%20%20%20from%20xdsl.pattern_rewriter%20import%20PatternRewriter%2C%20RewritePattern%2C%20op_type_rewrite_pattern%2C%20PatternRewriteWalker%2C%20GreedyRewritePatternApplier%0A%20%20%20%20from%20xdsl.transforms.dead_code_elimination%20import%20region_dce%0A%20%20%20%20from%20xdsl.traits%20import%20Pure%0A%20%20%20%20from%20xdsl.irdl%20import%20irdl_op_definition%2C%20traits_def%2C%20IRDLOperation%2C%20irdl_attr_definition%2C%20operand_def%2C%20result_def%0A%20%20%20%20from%20xdsl.dialects.builtin%20import%20ModuleOp%2C%20Float64Type%2C%20FloatAttr%2C%20IntegerType%2C%20IntegerAttr%0A%20%20%20%20from%20xdsl.dialects.func%20import%20FuncOp%2C%20ReturnOp%0A%20%20%20%20from%20xdsl.dialects.arith%20import%20AddfOp%2C%20SubfOp%2C%20MulfOp%2C%20ConstantOp%2C%20AddiOp%2C%20MuliOp%2C%20SIToFPOp%2C%20FloatingPointLikeBinaryOperation%2C%20DivfOp%2C%20SelectOp%2C%20CmpfOp%0A%20%20%20%20from%20xdsl.dialects.scf%20import%20ForOp%2C%20YieldOp%2C%20IfOp%0A%20%20%20%20from%20xdsl.dialects.math%20import%20PowFOp%2C%20SqrtOp%0A%20%20%20%20from%20xdsl.builder%20import%20Builder%2C%20InsertPoint%0A%0A%20%20%20%20return%20(%0A%20%20%20%20%20%20%20%20Abs%2C%0A%20%20%20%20%20%20%20%20Add%2C%0A%20%20%20%20%20%20%20%20AddfOp%2C%0A%20%20%20%20%20%20%20%20AddiOp%2C%0A%20%20%20%20%20%20%20%20Attribute%2C%0A%20%20%20%20%20%20%20%20Block%2C%0A%20%20%20%20%20%20%20%20Builder%2C%0A%20%20%20%20%20%20%20%20CmpfOp%2C%0A%20%20%20%20%20%20%20%20ConstantOp%2C%0A%20%20%20%20%20%20%20%20Expr%2C%0A%20%20%20%20%20%20%20%20Float%2C%0A%20%20%20%20%20%20%20%20Float64Type%2C%0A%20%20%20%20%20%20%20%20FloatAttr%2C%0A%20%20%20%20%20%20%20%20ForOp%2C%0A%20%20%20%20%20%20%20%20FuncOp%2C%0A%20%20%20%20%20%20%20%20IfOp%2C%0A%20%20%20%20%20%20%20%20InsertPoint%2C%0A%20%20%20%20%20%20%20%20Integer%2C%0A%20%20%20%20%20%20%20%20IntegerAttr%2C%0A%20%20%20%20%20%20%20%20IntegerType%2C%0A%20%20%20%20%20%20%20%20ModuleOp%2C%0A%20%20%20%20%20%20%20%20Mul%2C%0A%20%20%20%20%20%20%20%20MulfOp%2C%0A%20%20%20%20%20%20%20%20MuliOp%2C%0A%20%20%20%20%20%20%20%20Pow%2C%0A%20%20%20%20%20%20%20%20PowFOp%2C%0A%20%20%20%20%20%20%20%20Region%2C%0A%20%20%20%20%20%20%20%20ReturnOp%2C%0A%20%20%20%20%20%20%20%20SIToFPOp%2C%0A%20%20%20%20%20%20%20%20SSAValue%2C%0A%20%20%20%20%20%20%20%20SelectOp%2C%0A%20%20%20%20%20%20%20%20SubfOp%2C%0A%20%20%20%20%20%20%20%20Sum%2C%0A%20%20%20%20%20%20%20%20Symbol%2C%0A%20%20%20%20%20%20%20%20YieldOp%2C%0A%20%20%20%20%20%20%20%20symbols%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%20%23%20Writing%20an%20IR%20emitter%20for%20SymPy%0A%0A%20%20%20%20SymPy%20is%20a%20Python%20library%20for%20symbolic%20mathematics.%20The%20goal%20here%20is%20to%20convert%20SymPy%0A%20%20%20%20expressions%20to%20MLIR%20core%20dialects%2C%20namely%20%60builtin%60%2C%20%60func%60%2C%20and%20%60arith%60.%20In%20this%20exercise%2C%20we%20will%20only%20handle%20a%20subset%20of%20SymPy%20that%20only%20deals%20with%20real%20numbers%20for%20simplicity.%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%20Some%20SymPy%20examples%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(symbols)%3A%0A%20%20%20%20%23%20Define%20variables%20that%20can%20be%20used%20in%20the%20rest%20of%20the%20notebook%0A%20%20%20%20%23%20x%2C%20y%2C%20z%2C%20t%20are%20non-zero%20reals%2C%20while%20a%2C%20b%2C%20c%2C%20d%20are%20non-zero%20integers%0A%20%20%20%20x%2C%20y%2C%20z%2C%20t%20%3D%20symbols(%22x%20y%20z%20t%22%2C%20real%3DTrue%2C%20zero%3DFalse)%0A%20%20%20%20a%2C%20b%2C%20c%2C%20d%20%3D%20symbols(%22a%20b%20c%20d%22%2C%20integer%3DTrue%2C%20zero%3DFalse)%0A%0A%20%20%20%20%23%20SymPy%20uses%20overloaded%20Python%20operators%20to%20define%20expressions.%0A%20%20%20%20%23%20Expressions%20are%20automatically%20simplified%20by%20SymPy.%0A%20%20%20%20print('%22x%20%2B%20y%20*%20z%22%20-%3E%20'%2C%20x%20%2B%20y%20*%20z)%0A%20%20%20%20print('%22x%20%2B%20x%22%20-%3E%20'%2C%20x%20%2B%20x)%0A%20%20%20%20print('%22x%20-%20x%22%20-%3E%20'%2C%20x%20-%20x)%0A%20%20%20%20return%20a%2C%20b%2C%20x%2C%20y%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%20Expected%20MLIR%20output%0A%0A%20%20%20%20We%20want%20to%20only%20output%20code%20from%20the%20%60arith%60%20dialect%20(and%20%60math.powf%60)%2C%20which%20deals%20with%20arithmetic.%0A%0A%20%20%20%20Here%20is%20the%20expected%20MLIR%20output%20for%20the%20expression%20%60x%20%2B%20x%60%3A%0A%20%20%20%20%60%60%60%0A%20%20%20%20builtin.module%20%7B%0A%20%20%20%20%20%20func.func%20%40main(%25x%3A%20f64)%20-%3E%20f64%20%7B%0A%20%20%20%20%20%20%20%20%25add%20%3D%20arith.addf%20%25x%2C%20%25x%20%3A%20f64%0A%20%20%20%20%20%20%20%20func.return%20%25add%20%3A%20f64%0A%20%20%20%20%20%20%7D%0A%20%20%20%20%7D%0A%20%20%20%20%60%60%60%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(%22%22%22%0A%20%20%20%20The%20operations%20you%20will%20have%20to%20use%20have%20the%20following%20constructors.%3A%0A%0A%20%20%20%20*%20arith.constant%3A%20ConstantOp(IntegerAttr(int_cst%2C%20IntegerType(64)))%0A%20%20%20%20*%20arith.constant%3A%20ConstantOp(FloatAttr(float_cst%2C%20Float64Type()))%0A%20%20%20%20*%20arith.addi%3A%20AddiOp(lhs%2C%20rhs)%0A%20%20%20%20*%20arith.muli%3A%20MuliOp(lhs%2C%20rhs)%0A%20%20%20%20*%20arith.addf%3A%20AddfOp(lhs%2C%20rhs)%0A%20%20%20%20*%20arith.mulf%3A%20MulfOp(lhs%2C%20rhs)%0A%20%20%20%20*%20math.powf%3A%20PowFOp(lhs%2C%20rhs)%0A%20%20%20%20*%20arith.select%3A%20SelectOp(cond%2C%20lhs%2C%20rhs)%0A%20%20%20%20%20%20%20%20*%20Represents%20the%20formula%20%60if%20cond%20then%20lhs%20else%20rhs%60%0A%20%20%20%20*%20arith.cmpf%3A%20CmpfOp(lhs%2C%20rhs%2C%20%22olt%22)%0A%20%20%20%20%20%20%20%20*%20Represents%20%60lhs%20%3C%20rhs%60%20with%20floating%20points%0A%20%20%20%20*%20scf.if%3A%20IfOp(cond%2C%20region1%2C%20region2)%0A%20%20%20%20%20%20%20%20*%20Regions%20should%20have%20a%20single%20block%2C%20without%20block%20arguments.%20The%20last%20operation%20in%20the%20regions%20should%20be%20an%20%60scf.yield%60%20with%20constructor%20%60YieldOp(%5Bresult%5D)%60%0A%20%20%20%20%20%20%20%20*%20%60.results%5B0%5D%60%20is%20used%20to%20get%20the%20result%20of%20%60IfOp%60%0A%20%20%20%20*%20scf.for%3A%20ForOp(lower_bound%2C%20upper_bound%2C%20step%2C%20%5Bfirst_acc%5D%2C%20region)%0A%20%20%20%20%20%20%20%20*%20%60first_acc%60%20is%20the%20initial%20value%20of%20the%20accumulator%0A%20%20%20%20%20%20%20%20*%20%60region%60%20should%20have%20a%20single%20block%20with%20two%20block%20arguments.%20One%20for%20the%20value%20we%20iterate%20on%20(we%20use%20%60i64%60)%2C%20and%20the%20second%20for%20the%20accumulator%20(we%20use%20%60f64%60%20here).%0A%20%20%20%20%20%20%20%20*%20%60.results%5B0%5D%60%20is%20used%20to%20get%20the%20result%20of%20%60ForOp%60%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%20Introspecting%20SymPy%20expressions%0A%0A%20%20%20%20SymPy%20expressions%20all%20inherit%20from%20%60Expr%60.%20For%20all%20our%20examples%2C%20%60Expr%60%20are%20a%20tree%20of%20expressions%2C%20where%20nodes%20are%20function%20calls%20and%20leaves%20are%20either%20symbols%20or%20constants.%0A%0A%20%20%20%20*%20Function%20calls%20can%20be%20introspected%20using%20the%20%60func%60%20and%20%60args%60%20fields%2C%20which%20respectively%20return%20the%20associated%20function%20and%20its%20arguments%20(its%20node%20children).%20Additionally%2C%20the%20type%20of%20a%20function%20call%20is%20equal%20to%20the%20value%20returned%20by%20%60func%60.%0A%20%20%20%20*%20Constants%20can%20be%20converted%20to%20floats%20using%20the%20%60float%60%20function.%0A%20%20%20%20*%20Each%20symbol%20is%20a%20function%20argument.%0A%0A%20%20%20%20Here%20are%20some%20examples%20of%20SymPy%20introspection%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(x)%3A%0A%20%20%20%20%23%20A%20SymPy%20expression%0A%20%20%20%20expression%20%3D%20x%20%2B%20x%0A%0A%20%20%20%20print(%22expression%3A%22%2C%20expression)%0A%0A%20%20%20%20%23%20The%20func%20argument%20is%20equal%20to%20the%20type%20of%20the%20AST%20node%0A%20%20%20%20print(%22type%3A%22%2C%20type(expression))%0A%20%20%20%20print(%22func%3A%22%2C%20expression.func)%0A%0A%20%20%20%20%23%20The%20operation%20argument%20and%20their%20types%3A%0A%20%20%20%20print(%22args%3A%22%2C%20expression.args)%0A%20%20%20%20print(%22type(args%5B0%5D)%3A%22%2C%20type(expression.args%5B0%5D))%0A%20%20%20%20print(%22type(args%5B1%5D)%3A%22%2C%20type(expression.args%5B1%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%20Exercise%3A%20Emitting%20MLIR%20IR%20from%20SymPy%0A%0A%20%20%20%20Your%20goal%20is%20to%20convert%20SymPy%20AST%20nodes%20to%20MLIR%20IR%2C%20using%20xDSL.%0A%20%20%20%20We%20are%20giving%20you%20most%20of%20the%20boilerplate%2C%20so%20you%20only%20have%20to%20focus%20on%20emitting%20IR%20for%20each%20SymPy%20AST%20node.%0A%0A%20%20%20%20The%20following%20function%20will%20print%20an%20expression%2C%20and%20print%20the%20resulting%20MLIR%20that%20you%20are%20emitting%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(Expr%2C%20emit_ir)%3A%0A%20%20%20%20def%20print_ir(expr%3A%20Expr)%3A%0A%20%20%20%20%20%20%20%20%23%20Print%20the%20SymPy%20expression%0A%20%20%20%20%20%20%20%20print(expr)%0A%0A%20%20%20%20%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%23%20Converts%20the%20SymPy%20expression%20to%20an%20MLIR%20%60builtin.module%60%20operation%0A%20%20%20%20%20%20%20%20%20%20%20%20op%20%3D%20emit_ir(expr)%0A%20%20%20%20%20%20%20%20except%20NotImplementedError%20as%20e%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20print(e%2C%20%22%5Cn%5Cn%22)%0A%20%20%20%20%20%20%20%20%20%20%20%20return%0A%0A%20%20%20%20%20%20%20%20%23%20Check%20that%20the%20operation%20verifies%2C%20and%20prints%20the%20operation%0A%20%20%20%20%20%20%20%20op.verify()%0A%20%20%20%20%20%20%20%20print(op)%0A%0A%20%20%20%20%20%20%20%20%23%20Print%20a%20separator%0A%20%20%20%20%20%20%20%20print(%22%5Cn%5Cn%22)%0A%0A%20%20%20%20return%20(print_ir%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(%22%22%22%0A%20%20%20%20The%20following%20function%20returns%20the%20MLIR%20type%20of%20the%20expression%20result.%20As%20we%20are%20only%20handling%20integer%20and%20real%20types%2C%20we%20only%20returns%20either%20the%20%60i64%60%20or%20%60f64%60%20types%2C%20which%20correspond%20to%2064bits%20integers%20and%20floating%20points.%0A%0A%20%20%20%20For%20instance%2C%20the%20type%20of%20%60x%20%2B%20y%60%20is%20%60f64%60%2C%20as%20%60x%60%20and%20%60y%60%20are%20floating%20points.%20The%20type%20of%20%60a%20*%20b%60%20is%20%60i64%60%2C%20as%20both%20%60a%60%20and%20%60b%60%20are%20integers.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(Attribute%2C%20Expr%2C%20Float64Type%2C%20IntegerType)%3A%0A%20%20%20%20%23%20Get%20the%20MLIR%20type%20for%20a%20SymPy%20expression%0A%20%20%20%20def%20get_mlir_type(expr%3A%20Expr)%20-%3E%20Attribute%3A%0A%20%20%20%20%20%20%20%20if%20expr.is_integer%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20return%20IntegerType(64)%0A%20%20%20%20%20%20%20%20elif%20expr.is_extended_real%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20return%20Float64Type()%0A%20%20%20%20%20%20%20%20else%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20raise%20Exception(f%22Unknown%20MLIR%20type%20for%20expression%20%7Bexpr%7D.%20Please%20make%20sure%20there%20cannot%20be%20a%20division%20by%20zero%2C%20or%20a%20power%20of%20a%20negative%20value.%22)%0A%0A%20%20%20%20return%20(get_mlir_type%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%20following%20function%20(%60emit_ir%60)%20should%20be%20called%20to%20emit%20MLIR%20IR%20from%20a%20SymPy%20expression.%20It%20takes%20a%20SymPy%20expression%2C%20creates%20a%20module%20and%20a%20function%2C%20and%20starts%20the%20recursion%20on%20the%20SymPy%20AST%20to%20emit%20MLIR%20IR.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(%0A%20%20%20%20Builder%2C%0A%20%20%20%20Expr%2C%0A%20%20%20%20FuncOp%2C%0A%20%20%20%20InsertPoint%2C%0A%20%20%20%20ModuleOp%2C%0A%20%20%20%20ReturnOp%2C%0A%20%20%20%20emit_op%2C%0A%20%20%20%20get_mlir_type%2C%0A)%3A%0A%20%20%20%20def%20emit_ir(expr%3A%20Expr)%20-%3E%20ModuleOp%3A%0A%20%20%20%20%20%20%20%20%23%20Create%20a%20module%2C%20and%20create%20a%20builder%20at%20the%20beginning%20of%20its%20only%20block%0A%20%20%20%20%20%20%20%20module%20%3D%20ModuleOp(%5B%5D)%0A%20%20%20%20%20%20%20%20builder%20%3D%20Builder(InsertPoint.at_end(module.body.block))%0A%0A%20%20%20%20%20%20%20%20%23%20Create%20the%20MLIR%20types%20for%20each%20symbol.%0A%20%20%20%20%20%20%20%20arg_types%20%3D%20%5Bget_mlir_type(arg)%20for%20arg%20in%20expr.free_symbols%5D%0A%0A%20%20%20%20%20%20%20%20%23%20Create%20a%20new%20function%20and%20inserts%20it%20inside%20the%20module.%0A%20%20%20%20%20%20%20%20func%20%3D%20FuncOp(%22main%22%2C%20(arg_types%2C%20%5Bget_mlir_type(expr)%5D))%0A%20%20%20%20%20%20%20%20builder.insert(func)%0A%0A%20%20%20%20%20%20%20%20%23%20Associate%20each%20symbol%20with%20its%20MLIR%20name.%0A%20%20%20%20%20%20%20%20arg_values%20%3D%20%7Barg%3A%20value%20for%20arg%2C%20value%20in%20zip(expr.free_symbols%2C%20func.args)%7D%0A%0A%20%20%20%20%20%20%20%20%23%20Set%20the%20name%20for%20each%20function%20argument.%20This%20is%20only%20to%20get%20better%20names%0A%20%20%20%20%20%20%20%20%23%20for%20IR%20values.%0A%20%20%20%20%20%20%20%20for%20arg%2C%20value%20in%20arg_values.items()%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20value.name_hint%20%3D%20arg.name%0A%0A%20%20%20%20%20%20%20%20%23%20Set%20the%20builder%20insertion%20point%20inside%20the%20function.%0A%20%20%20%20%20%20%20%20builder.insertion_point%20%3D%20InsertPoint.at_end(func.body.block)%0A%0A%20%20%20%20%20%20%20%20%23%20Convert%20the%20expression%20into%20MLIR%20IR%20inside%20the%20function.%0A%20%20%20%20%20%20%20%20result%20%3D%20emit_op(expr%2C%20builder%2C%20arg_values)%0A%0A%20%20%20%20%20%20%20%20%23%20Insert%20a%20return%20statement%20at%20the%20end%20of%20the%20function.%0A%20%20%20%20%20%20%20%20builder.insert(ReturnOp(result))%0A%20%20%20%20%20%20%20%20return%20module%0A%0A%20%20%20%20return%20(emit_ir%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%20Finally%2C%20here%20are%20the%20functions%20that%20you%20should%20complete.%20%60emit_op%60%20is%20fully%20complete%2C%20and%20emits%20the%20necessary%20IR%20for%20a%20SymPy%20expression.%20%60emit_integer_op%60%20and%20%60emit_float_op%60%20emits%20the%20operations%20for%20integer%20and%20float%20operations%2C%20and%20are%20only%20partially%20implemented.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(%0A%20%20%20%20Builder%2C%0A%20%20%20%20Expr%2C%0A%20%20%20%20Float%2C%0A%20%20%20%20Float64Type%2C%0A%20%20%20%20Integer%2C%0A%20%20%20%20IntegerType%2C%0A%20%20%20%20SIToFPOp%2C%0A%20%20%20%20SSAValue%2C%0A%20%20%20%20Symbol%2C%0A%20%20%20%20get_mlir_type%2C%0A)%3A%0A%20%20%20%20def%20emit_op(%0A%20%20%20%20%20%20%20%20expr%3A%20Expr%2C%0A%20%20%20%20%20%20%20%20builder%3A%20Builder%2C%0A%20%20%20%20%20%20%20%20args%3A%20dict%5BSymbol%2C%20SSAValue%5D%2C%0A%20%20%20%20)%20-%3E%20SSAValue%3A%0A%20%20%20%20%20%20%20%20type%20%3D%20get_mlir_type(expr)%0A%20%20%20%20%20%20%20%20if%20isinstance(type%2C%20IntegerType)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20return%20emit_integer_op(expr%2C%20builder%2C%20args)%0A%20%20%20%20%20%20%20%20elif%20isinstance(type%2C%20Float64Type)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20return%20emit_real_op(expr%2C%20builder%2C%20args)%0A%20%20%20%20%20%20%20%20else%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20raise%20Exception(%22Unknown%20function%20to%20emit%20IR%20for%20MLIR%20type%20%22%2C%20type)%0A%0A%20%20%20%20def%20emit_integer_op(%0A%20%20%20%20%20%20%20%20expr%3A%20Expr%2C%0A%20%20%20%20%20%20%20%20builder%3A%20Builder%2C%0A%20%20%20%20%20%20%20%20args%3A%20dict%5BSymbol%2C%20SSAValue%5D%2C%0A%20%20%20%20)%20-%3E%20SSAValue%3A%0A%20%20%20%20%20%20%20%20%23%20Handle%20symbolic%20values%0A%20%20%20%20%20%20%20%20if%20isinstance(expr%2C%20Symbol)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20return%20args%5Bexpr%5D%0A%0A%20%20%20%20%20%20%20%20%23%20Handle%20constants%0A%20%20%20%20%20%20%20%20if%20isinstance(expr%2C%20Integer)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%23%20Hint%3A%20int(expr)%20returns%20the%20value%20of%20the%20%60expr%60%20constant%0A%20%20%20%20%20%20%20%20%20%20%20%20raise%20NotImplementedError(%22Integer%20constants%20are%20not%20implemented%22)%0A%0A%20%20%20%20%20%20%20%20%23%20Hint%3A%20Here%2C%20implement%20support%20for%20Add%20and%20Mul%0A%0A%20%20%20%20%20%20%20%20raise%20NotImplementedError(f%22No%20IR%20emitter%20for%20integer%20function%20%7Bexpr.func%7D%22)%0A%0A%20%20%20%20def%20emit_real_op(%0A%20%20%20%20%20%20%20%20expr%3A%20Expr%2C%0A%20%20%20%20%20%20%20%20builder%3A%20Builder%2C%0A%20%20%20%20%20%20%20%20args%3A%20dict%5BSymbol%2C%20SSAValue%5D%2C%0A%20%20%20%20)%20-%3E%20SSAValue%3A%0A%20%20%20%20%20%20%20%20%23%20If%20the%20expression%20is%20an%20integer%20expression%2C%20emit%20it%20and%20then%20convert%20it%0A%20%20%20%20%20%20%20%20%23%20to%20a%20float%20expression.%0A%20%20%20%20%20%20%20%20if%20expr.is_integer%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20res%20%3D%20emit_integer_op(expr%2C%20builder%2C%20args)%0A%20%20%20%20%20%20%20%20%20%20%20%20op%20%3D%20builder.insert(SIToFPOp(res%2C%20Float64Type()))%0A%20%20%20%20%20%20%20%20%20%20%20%20return%20op.result%0A%0A%20%20%20%20%20%20%20%20%23%20Handle%20constants%0A%20%20%20%20%20%20%20%20if%20isinstance(expr%2C%20Float)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%23%20Hint%3A%20float(expr)%20returns%20the%20value%20of%20the%20%60expr%60%20constant%0A%20%20%20%20%20%20%20%20%20%20%20%20raise%20NotImplementedError(%22Float%20constants%20are%20not%20implemented%22)%0A%0A%20%20%20%20%20%20%20%20%23%20Handle%20symbolic%20values%0A%20%20%20%20%20%20%20%20if%20isinstance(expr%2C%20Symbol)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20return%20args%5Bexpr%5D%0A%0A%20%20%20%20%20%20%20%20%23%20Hint%3A%20Here%2C%20implement%20support%20for%20Add%2C%20Mul%2C%20and%20Pow%20(and%20later%20Abs%20and%20Sum)%0A%0A%20%20%20%20%20%20%20%20raise%20NotImplementedError(f%22No%20IR%20emitter%20for%20float%20function%20%7Bexpr.func%7D%22)%0A%0A%20%20%20%20return%20(emit_op%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo)%3A%0A%20%20%20%20mo.md(%22%22%22%0A%20%20%20%20Here%20are%20a%20few%20simple%20examples%20that%20you%20should%20support%20first.%20For%20each%20test%2C%20the%20expression%20is%20printed%2C%20then%20either%20the%20MLIR%20code%2C%20or%20an%20error.%20Each%20of%20the%20operators%20used%20in%20these%20tests%20should%20only%20be%20converted%20to%20a%20single%20MLIR%20operation.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(Float%2C%20Integer%2C%20a%2C%20b%2C%20print_ir%2C%20x%2C%20y)%3A%0A%20%20%20%20print_ir(Float(2))%0A%20%20%20%20print_ir(Integer(2))%0A%0A%20%20%20%20%23%20Adds%20two%20integers%0A%20%20%20%20print_ir(a%20%2B%20b)%0A%0A%20%20%20%20%23%20Adds%20two%20integers%20with%20a%20real%0A%20%20%20%20print_ir(a%20%2B%20b%20%2B%20x)%0A%0A%20%20%20%20%23%20Multiplies%20two%20reals%0A%20%20%20%20print_ir(x%20*%20y)%0A%0A%20%20%20%20%23%20Multiplies%20three%20reals%0A%20%20%20%20print_ir(a%20*%20b%20%2B%20x)%0A%0A%20%20%20%20%23%20Add%20two%20reals%0A%20%20%20%20print_ir(x%20%2B%20x)%0A%0A%20%20%20%20%23%20Square%20a%20real%0A%20%20%20%20print_ir(x%20**%204)%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%20The%20following%20expression%20requires%20to%20handle%20the%20AST%20node%20%60Abs%60.%20Instead%20of%20converting%20it%20to%20%60math.absf%60%20operation%2C%20the%20task%20is%20to%20write%20it%20using%20the%20formula%20%60x%20%3C%200%20%3F%200-x%20%3A%20x%60%20using%20only%20%60arith%60%20operations.%20Hint%2C%20you%20should%20use%20%60arith.select%60%20for%20expressing%20the%20conditional.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(Abs%2C%20print_ir%2C%20x%2C%20y)%3A%0A%20%20%20%20print_ir(Abs(x%20%2B%20y))%0A%20%20%20%20print_ir((x%20**%202)%20**%20y)%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%20Supporting%20operations%20with%20regions%0A%0A%20%20%20%20Your%20next%20task%20is%20to%20handle%20operations%20that%20may%20have%20regions.%0A%0A%20%20%20%20As%20a%20first%20step%2C%20rewrite%20the%20lowering%20to%20%60Abs%60%20to%20output%20an%20%60scf.if%60%20instead%20of%20an%20%60arith.select%60.%20Then%2C%20as%20a%20harder%20task%2C%20support%20the%20%60Sum%60%20operation%20using%20an%20%60scf.for%60%20loop.%0A%0A%20%20%20%20Here%20are%20a%20few%20examples%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(Abs%2C%20Sum%2C%20a%2C%20b%2C%20print_ir%2C%20x%2C%20y)%3A%0A%20%20%20%20print_ir(Abs(x%20%2B%20y))%0A%20%20%20%20print_ir((x%20**%202)%20**%20y)%0A%0A%20%20%20%20%23%20The%20sum%20of%20all%20numbers%20from%200%20to%2010%20(excluded)%0A%20%20%20%20%23%20You%20can%20access%20the%20Sum%20arguments%20with%20%60args%5B0%5D%60%2C%20and%20%60args%5B1%5D%5B0%5D%60%2C%20%60args%5B1%5D%5B1%5D%60%20and%20%60args%5B1%5D%5B2%5D%60.%0A%20%20%20%20print_ir(Sum(a%2C%20(a%2C%200%2C%2010)))%0A%0A%20%20%20%20%23%20The%20triangle%20sum%20from%200%20to%20a%20(excluded)%0A%20%20%20%20print_ir(Sum(b%2Bb%2C%20(b%2C%200%2C%20a)))%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%20Solutions%0A%0A%20%20%20%20Here%20is%20a%20solution%20for%20the%20implementation%20of%20%60emit_real_op%60%20and%20%60emit_integer_op%60.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(%0A%20%20%20%20Abs%2C%0A%20%20%20%20Add%2C%0A%20%20%20%20AddfOp%2C%0A%20%20%20%20AddiOp%2C%0A%20%20%20%20Block%2C%0A%20%20%20%20Builder%2C%0A%20%20%20%20CmpfOp%2C%0A%20%20%20%20ConstantOp%2C%0A%20%20%20%20Expr%2C%0A%20%20%20%20Float%2C%0A%20%20%20%20Float64Type%2C%0A%20%20%20%20FloatAttr%2C%0A%20%20%20%20ForOp%2C%0A%20%20%20%20IfOp%2C%0A%20%20%20%20InsertPoint%2C%0A%20%20%20%20Integer%2C%0A%20%20%20%20IntegerAttr%2C%0A%20%20%20%20IntegerType%2C%0A%20%20%20%20Mul%2C%0A%20%20%20%20MulfOp%2C%0A%20%20%20%20MuliOp%2C%0A%20%20%20%20Pow%2C%0A%20%20%20%20PowFOp%2C%0A%20%20%20%20Region%2C%0A%20%20%20%20SIToFPOp%2C%0A%20%20%20%20SSAValue%2C%0A%20%20%20%20SelectOp%2C%0A%20%20%20%20SubfOp%2C%0A%20%20%20%20Sum%2C%0A%20%20%20%20Symbol%2C%0A%20%20%20%20YieldOp%2C%0A)%3A%0A%20%20%20%20def%20solution()%3A%0A%20%20%20%20%20%20%20%20def%20emit_integer_op(%0A%20%20%20%20%20%20%20%20%20%20%20%20expr%3A%20Expr%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20builder%3A%20Builder%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20args%3A%20dict%5BSymbol%2C%20SSAValue%5D%2C%0A%20%20%20%20%20%20%20%20)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%23%20Handle%20symbolic%20values%0A%20%20%20%20%20%20%20%20%20%20%20%20if%20isinstance(expr%2C%20Symbol)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20return%20args%5Bexpr%5D%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20%23%20Handle%20constants%0A%20%20%20%20%20%20%20%20%20%20%20%20if%20isinstance(expr%2C%20Integer)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20constant_op%20%3D%20builder.insert(%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20ConstantOp(IntegerAttr(int(expr)%2C%20IntegerType(64)))%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20return%20constant_op.result%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20if%20isinstance(expr%2C%20Add)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20lhs%20%3D%20emit_integer_op(expr.args%5B0%5D%2C%20builder%2C%20args)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20rhs%20%3D%20emit_integer_op(expr.args%5B1%5D%2C%20builder%2C%20args)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20add_op%20%3D%20builder.insert(AddiOp(lhs%2C%20rhs))%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20return%20add_op.result%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20if%20isinstance(expr%2C%20Mul)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20lhs%20%3D%20emit_integer_op(expr.args%5B0%5D%2C%20builder%2C%20args)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20rhs%20%3D%20emit_integer_op(expr.args%5B1%5D%2C%20builder%2C%20args)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20add_op%20%3D%20builder.insert(MuliOp(lhs%2C%20rhs))%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20return%20add_op.result%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20raise%20NotImplementedError(f%22No%20IR%20emitter%20for%20integer%20function%20%7Bexpr.func%7D%22)%0A%0A%20%20%20%20%20%20%20%20def%20emit_real_op(%0A%20%20%20%20%20%20%20%20%20%20%20%20expr%3A%20Expr%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20builder%3A%20Builder%2C%0A%20%20%20%20%20%20%20%20%20%20%20%20args%3A%20dict%5BSymbol%2C%20SSAValue%5D%2C%0A%20%20%20%20%20%20%20%20)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%23%20If%20the%20expression%20is%20an%20integer%20expression%2C%20emits%20it%20and%20then%20convert%20it%0A%20%20%20%20%20%20%20%20%20%20%20%20%23%20back%20to%20a%20float%20expression.%0A%20%20%20%20%20%20%20%20%20%20%20%20if%20expr.is_integer%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20res%20%3D%20emit_integer_op(expr%2C%20builder%2C%20args)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20op%20%3D%20builder.insert(SIToFPOp(res%2C%20Float64Type()))%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20return%20op.result%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20%23%20Handle%20constants%0A%20%20%20%20%20%20%20%20%20%20%20%20if%20isinstance(expr%2C%20Float)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20constant_op%20%3D%20builder.insert(%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20ConstantOp(FloatAttr(float(expr)%2C%20Float64Type()))%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20return%20constant_op.result%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20%23%20Handle%20symbolic%20values%0A%20%20%20%20%20%20%20%20%20%20%20%20if%20isinstance(expr%2C%20Symbol)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20return%20args%5Bexpr%5D%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20if%20isinstance(expr%2C%20Add)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20lhs%20%3D%20emit_real_op(expr.args%5B0%5D%2C%20builder%2C%20args)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20rhs%20%3D%20emit_real_op(expr.args%5B1%5D%2C%20builder%2C%20args)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20add_op%20%3D%20builder.insert(AddfOp(lhs%2C%20rhs))%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20return%20add_op.result%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20if%20isinstance(expr%2C%20Mul)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20lhs%20%3D%20emit_real_op(expr.args%5B0%5D%2C%20builder%2C%20args)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20rhs%20%3D%20emit_real_op(expr.args%5B1%5D%2C%20builder%2C%20args)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20mul_op%20%3D%20builder.insert(MulfOp(lhs%2C%20rhs))%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20return%20mul_op.result%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20if%20isinstance(expr%2C%20Pow)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20lhs%20%3D%20emit_real_op(expr.args%5B0%5D%2C%20builder%2C%20args)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20rhs%20%3D%20emit_real_op(expr.args%5B1%5D%2C%20builder%2C%20args)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20pow_op%20%3D%20builder.insert(PowFOp(lhs%2C%20rhs))%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20return%20pow_op.result%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20if%20isinstance(expr%2C%20Abs)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%23%20The%20arith.select%20solution%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20if%20False%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20arg%20%3D%20emit_real_op(expr.args%5B0%5D%2C%20builder%2C%20args)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20zero%20%3D%20builder.insert(ConstantOp(FloatAttr(0%2C%20Float64Type()))).result%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20neg%20%3D%20builder.insert(SubfOp(zero%2C%20arg)).result%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20is_neg%20%3D%20builder.insert(CmpfOp(arg%2C%20zero%2C%20%22olt%22)).result%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20select%20%3D%20builder.insert(SelectOp(is_neg%2C%20neg%2C%20arg)).result%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20return%20select%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%23%20The%20scf.if%20solution%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20arg%20%3D%20emit_real_op(expr.args%5B0%5D%2C%20builder%2C%20args)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20zero%20%3D%20builder.insert(ConstantOp(FloatAttr(0%2C%20Float64Type()))).result%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20is_neg%20%3D%20builder.insert(CmpfOp(arg%2C%20zero%2C%20%22olt%22)).result%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20lhs_region%20%3D%20Region(%5BBlock()%5D)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20builder2%20%3D%20Builder(InsertPoint.at_end(lhs_region.block))%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20neg%20%3D%20builder2.insert(SubfOp(zero%2C%20arg)).result%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20builder2.insert(YieldOp(neg))%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20rhs_region%20%3D%20Region(%5BBlock()%5D)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20builder3%20%3D%20Builder(InsertPoint.at_end(rhs_region.block))%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20builder3.insert(YieldOp(arg))%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20if_res%20%3D%20builder.insert(%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20IfOp(is_neg%2C%20%5BFloat64Type()%5D%2C%20lhs_region%2C%20rhs_region)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20).results%5B0%5D%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20return%20if_res%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20if%20isinstance(expr%2C%20Sum)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20zero%20%3D%20builder.insert(ConstantOp(FloatAttr(0%2C%20Float64Type()))).result%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20lb%20%3D%20emit_integer_op(expr.args%5B1%5D%5B1%5D%2C%20builder%2C%20args)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20ub%20%3D%20emit_integer_op(expr.args%5B1%5D%5B2%5D%2C%20builder%2C%20args)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20step%20%3D%20builder.insert(ConstantOp(IntegerAttr(1%2C%20IntegerType(64)))).result%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20region%20%3D%20Region(%5BBlock(arg_types%3D%5BIntegerType(64)%2C%20Float64Type()%5D)%5D)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20accumulator%20%3D%20region.block.args%5B1%5D%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20b2%20%3D%20Builder(InsertPoint.at_end(region.block))%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20arg%20%3D%20emit_real_op(%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20expr.args%5B0%5D%2C%20b2%2C%20args%20%7C%20%7Bexpr.args%5B1%5D%5B0%5D%3A%20region.block.args%5B0%5D%7D%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20)%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20add%20%3D%20b2.insert(AddfOp(arg%2C%20accumulator)).result%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20b2.insert(YieldOp(add))%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20return%20builder.insert(ForOp(lb%2C%20ub%2C%20step%2C%20%5Bzero%5D%2C%20region)).results%5B0%5D%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20raise%20NotImplementedError(f%22No%20IR%20emitter%20for%20float%20function%20%7Bexpr.func%7D%22)%0A%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