traversing ir

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%20(xmo%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_()%3A%0A%20%20%20%20from%20xdsl.ir%20import%20Dialect%0A%20%20%20%20from%20collections%20import%20defaultdict%0A%20%20%20%20from%20xdsl.dialects%20import%20builtin%2C%20func%2C%20arith%2C%20scf%0A%20%20%20%20from%20xdsl.context%20import%20Context%0A%20%20%20%20from%20xdsl.parser%20import%20Parser%0A%20%20%20%20from%20collections%20import%20Counter%0A%20%20%20%20from%20xdsl.ir%20import%20OpResult%0A%0A%20%20%20%20return%20(%0A%20%20%20%20%20%20%20%20Context%2C%0A%20%20%20%20%20%20%20%20Counter%2C%0A%20%20%20%20%20%20%20%20Dialect%2C%0A%20%20%20%20%20%20%20%20OpResult%2C%0A%20%20%20%20%20%20%20%20Parser%2C%0A%20%20%20%20%20%20%20%20arith%2C%0A%20%20%20%20%20%20%20%20builtin%2C%0A%20%20%20%20%20%20%20%20defaultdict%2C%0A%20%20%20%20%20%20%20%20func%2C%0A%20%20%20%20%20%20%20%20scf%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%20Traversing%20IR%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%20Parsing%20IR%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_()%3A%0A%20%20%20%20sum_of_squares_text%20%3D%20%22%22%22%5C%0A%20%20%20%20func.func%20%40sum_of_squares(%25n%3A%20index)%20-%3E%20index%20%7B%0A%20%20%20%20%20%20%25zero%20%3D%20arith.constant%200%20%3A%20index%0A%20%20%20%20%20%20%25one%20%3D%20arith.constant%201%20%3A%20index%0A%20%20%20%20%20%20%25res%20%3D%20scf.for%20%25i%20%3D%20%25zero%20to%20%25n%20step%20%25one%20iter_args(%25acc_in%20%3D%20%25zero)%20-%3E%20(index)%20%7B%0A%20%20%20%20%20%20%20%20%25square%20%3D%20arith.muli%20%25i%2C%20%25i%20%3A%20index%0A%20%20%20%20%20%20%20%20%25acc_out%20%3D%20arith.addi%20%25acc_in%2C%20%25square%20%3A%20index%0A%20%20%20%20%20%20%20%20scf.yield%20%25acc_out%20%3A%20index%0A%20%20%20%20%20%20%7D%0A%20%20%20%20%20%20func.return%20%25res%20%3A%20index%0A%20%20%20%20%7D%5C%0A%20%20%20%20%22%22%22%0A%20%20%20%20return%20(sum_of_squares_text%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo%2C%20sum_of_squares_text)%3A%0A%20%20%20%20mo.md(%0A%20%20%20%20%20%20%20%20fr%22%22%22%0A%20%20%20%20In%20this%20notebook%2C%20we'll%20be%20looking%20at%20the%20structure%20of%20the%20following%20module%2C%0A%20%20%20%20which%20implements%20a%20sum-of-squares%20function%3A%0A%0A%20%20%20%20%7Bmo.ui.code_editor(sum_of_squares_text%2C%20language%3D%22javascript%22%2C%20disabled%3DTrue)%7D%0A%20%20%20%20%22%22%22%0A%20%20%20%20)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(Context%2C%20arith%2C%20builtin%2C%20func%2C%20scf)%3A%0A%20%20%20%20%23%20The%20context%20stores%20the%20available%20abstractions%0A%20%20%20%20ctx%20%3D%20Context()%0A%20%20%20%20ctx.load_dialect(builtin.Builtin)%0A%20%20%20%20ctx.load_dialect(func.Func)%0A%20%20%20%20ctx.load_dialect(arith.Arith)%0A%20%20%20%20ctx.load_dialect(scf.Scf)%0A%20%20%20%20return%20(ctx%2C)%0A%0A%0A%40app.cell%0Adef%20_(Parser%2C%20ctx%2C%20sum_of_squares_text%2C%20xmo)%3A%0A%20%20%20%20sum_of_squares_module%20%3D%20Parser(ctx%2C%20sum_of_squares_text).parse_module()%0A%0A%20%20%20%20%23%20We%20can%20then%20parse%20and%20reprint%20the%20same%20module%0A%20%20%20%20xmo.module_html(sum_of_squares_module)%0A%20%20%20%20return%20(sum_of_squares_module%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%20Walking%20the%20IR%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%20The%20Operation%20class%20exposes%20a%20%60walk%60%20function%2C%20which%20lets%20us%20visit%20the%20IR%20from%20top%20to%20bottom.%0A%0A%20%20%20%20For%20example%2C%20to%20get%20the%20names%20of%20all%20the%20operations%20in%20our%20module%2C%20we%20can%20write%20a%20snippet%20like%20this%3A%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(sum_of_squares_module)%3A%0A%20%20%20%20print(%5Bop.name%20for%20op%20in%20sum_of_squares_module.walk()%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%20These%20helpers%20will%20be%20useful%20for%20the%20exercises%20below%3A%0A%0A%20%20%20%20%60Operation%60%3A%0A%0A%20%20%20%20*%20%60results%3A%20Sequence%5BOpResult%5D%60%0A%20%20%20%20*%20%60operands%3A%20Sequence%5BSSAValue%5D%60%0A%0A%20%20%20%20%60Block%60%3A%0A%0A%20%20%20%20*%20%60parent_op%3A%20Operation%20%7C%20None%60%0A%20%20%20%20*%20%60args%3A%20Sequence%5BBlockArgument%5D%60%0A%0A%20%20%20%20%60SSAValue%60%3A%0A%0A%20%20%20%20*%20%60uses%3A%20set%5BUse%5D%60%0A%20%20%20%20*%20%60owner%3A%20Operation%20%7C%20Block%60%0A%0A%20%20%20%20%60OpResult%60%20and%20%60BlockArgument%60%20are%20subclasses%20of%20%60SSAValue%60.%0A%0A%20%20%20%20%60Use%60%3A%0A%0A%20%20%20%20*%20%60operation%3A%20Operation%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%20Exercises%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%20The%20rest%20of%20the%20notebook%20is%20a%20series%20of%20exercises%20for%20you%20to%20get%20acquainted%20with%20the%20API%20of%20xDSL.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo%2C%20sum_of_squares_module%2C%20unique_operations)%3A%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20_sorted%20%3D%20%22%7B%22%20%2B%20%22%2C%20%22.join(sorted(unique_operations(sum_of_squares_module)))%20%2B%20%22%7D%22%0A%20%20%20%20%20%20%20%20_res%20%3D%20str(_sorted)%0A%20%20%20%20except%20Exception%20as%20e%3A%0A%20%20%20%20%20%20%20%20_res%20%3D%20f%22%7Btype(e).__name__%7D%3A%20%7Be%7D%22%0A%0A%20%20%20%20mo.md(%0A%20%20%20%20%20%20%20%20fr%22%22%22%0A%20%20%20%20%20%20%20%20%23%23%23%20Exercise%201.%20Unique%20Operations%0A%0A%20%20%20%20%20%20%20%20Modify%20the%20function%20below%20to%20return%20the%20set%20of%20operations%20used%20in%20the%20module.%0A%0A%20%20%20%20%20%20%20%20%60%60%60%0A%20%20%20%20%20%20%20%20Expected%3A%20%7B%7Barith.addi%2C%20arith.constant%2C%20arith.muli%2C%20builtin.module%2C%20func.func%2C%20func.return%2C%20scf.for%2C%20scf.yield%7D%7D%0A%20%20%20%20%20%20%20%20Result%3A%20%20%20%7B_res%7D%0A%20%20%20%20%20%20%20%20%60%60%60%0A%20%20%20%20%20%20%20%20%22%22%22%0A%20%20%20%20)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(builtin)%3A%0A%20%20%20%20def%20unique_operations(module%3A%20builtin.ModuleOp)%20-%3E%20set%5Bstr%5D%3A%0A%20%20%20%20%20%20%20%20return%20set()%0A%0A%20%20%20%20return%20(unique_operations%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(builtin)%3A%0A%20%20%20%20%23%20Solution%0A%0A%20%20%20%20def%20_unique_operations(module%3A%20builtin.ModuleOp)%20-%3E%20set%5Bstr%5D%3A%0A%20%20%20%20%20%20%20%20return%20%7Bop.name%20for%20op%20in%20module.walk()%7D%0A%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo%2C%20operation_counts%2C%20sum_of_squares_module)%3A%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20_res%20%3D%20str(operation_counts(sum_of_squares_module))%0A%20%20%20%20except%20Exception%20as%20e%3A%0A%20%20%20%20%20%20%20%20_res%20%3D%20f%22%7Btype(e).__name__%7D%3A%20%7Be%7D%22%0A%0A%20%20%20%20mo.md(%0A%20%20%20%20%20%20%20%20fr%22%22%22%0A%20%20%20%20%20%20%20%20%23%23%23%20Exercise%202.%20Operation%20Counter%0A%0A%20%20%20%20%20%20%20%20Modify%20the%20function%20below%20to%20return%20the%20number%20of%20instances%20of%20the%20operations%20in%20the%20module.%0A%0A%20%20%20%20%20%20%20%20%60%60%60%0A%20%20%20%20%20%20%20%20Expected%3A%20%7B%7B'builtin.module'%3A%201%2C%20'func.func'%3A%201%2C%20'arith.constant'%3A%202%2C%20'scf.for'%3A%201%2C%20'arith.muli'%3A%201%2C%20'arith.addi'%3A%201%2C%20'scf.yield'%3A%201%2C%20'func.return'%3A%201%7D%7D%0A%20%20%20%20%20%20%20%20Result%3A%20%20%20%7B_res%7D%0A%20%20%20%20%20%20%20%20%60%60%60%0A%20%20%20%20%20%20%20%20%22%22%22%0A%20%20%20%20)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(builtin)%3A%0A%20%20%20%20%23%20%60Counter%60%20might%20come%20in%20handy%0A%0A%20%20%20%20def%20operation_counts(module%3A%20builtin.ModuleOp)%20-%3E%20dict%5Bstr%2C%20int%5D%3A%0A%20%20%20%20%20%20%20%20return%20%7B%7D%0A%0A%20%20%20%20return%20(operation_counts%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(Counter%2C%20builtin)%3A%0A%20%20%20%20%23%20solution%0A%0A%20%20%20%20def%20_operation_counts(module%3A%20builtin.ModuleOp)%20-%3E%20dict%5Bstr%2C%20int%5D%3A%0A%20%20%20%20%20%20%20%20return%20dict(Counter(op.name%20for%20op%20in%20module.walk()))%0A%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo%2C%20operations_by_dialect%2C%20sum_of_squares_module)%3A%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20_unsorted%20%3D%20operations_by_dialect(sum_of_squares_module)%0A%20%20%20%20%20%20%20%20_sorted%20%3D%20%7B%0A%20%20%20%20%20%20%20%20%20%20%20%20k%3A%20sorted(_unsorted%5Bk%5D)%0A%20%20%20%20%20%20%20%20%20%20%20%20for%20k%20in%20sorted(_unsorted)%0A%20%20%20%20%20%20%20%20%7D%0A%20%20%20%20%20%20%20%20_res%20%3D%20str(_sorted)%0A%20%20%20%20except%20Exception%20as%20e%3A%0A%20%20%20%20%20%20%20%20_res%20%3D%20f%22%7Btype(e).__name__%7D%3A%20%7Be%7D%22%0A%0A%20%20%20%20mo.md(%0A%20%20%20%20%20%20%20%20fr%22%22%22%0A%20%20%20%20%20%20%20%20%23%23%23%20Exercise%203.%20Operations%20By%20Dialect%0A%0A%20%20%20%20%20%20%20%20Modify%20the%20function%20below%20to%20return%20the%20operations%20by%20dialect%20in%20the%20module%0A%0A%20%20%20%20%20%20%20%20%60%60%60%0A%20%20%20%20%20%20%20%20Expected%3A%20%7B%7B'arith'%3A%20%5B'addi'%2C%20'constant'%2C%20'constant'%2C%20'muli'%5D%2C%20'builtin'%3A%20%5B'module'%5D%2C%20'func'%3A%20%5B'func'%2C%20'return'%5D%2C%20'scf'%3A%20%5B'for'%2C%20'yield'%5D%7D%7D%0A%20%20%20%20%20%20%20%20Result%3A%20%20%20%7B_res%7D%0A%20%20%20%20%20%20%20%20%60%60%60%0A%20%20%20%20%20%20%20%20%22%22%22%0A%20%20%20%20)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(builtin)%3A%0A%20%20%20%20%23%20%60defaultdict%60%20might%20come%20in%20handy%0A%0A%20%20%20%20def%20operations_by_dialect(module%3A%20builtin.ModuleOp)%20-%3E%20dict%5Bstr%2C%20list%5Bstr%5D%5D%3A%0A%20%20%20%20%20%20%20%20return%20%7B%7D%0A%0A%20%20%20%20return%20(operations_by_dialect%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(Dialect%2C%20builtin%2C%20defaultdict)%3A%0A%20%20%20%20%23%20solution%0A%0A%20%20%20%20def%20_operations_by_dialect(module%3A%20builtin.ModuleOp)%20-%3E%20dict%5Bstr%2C%20list%5Bstr%5D%5D%3A%0A%20%20%20%20%20%20%20%20res%20%3D%20defaultdict(list)%0A%0A%20%20%20%20%20%20%20%20for%20op%20in%20module.walk()%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20d%2C%20o%20%3D%20Dialect.split_name(op.name)%0A%20%20%20%20%20%20%20%20%20%20%20%20res%5Bd%5D.append(o)%0A%20%20%20%20%20%20%20%20return%20dict(res)%0A%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%20SSA%20Values%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%20SSA%20values%20in%20IR%20are%20either%20defined%20as%20results%20or%20block%20arguments.%0A%20%20%20%20In%20xDSL%2C%20the%20values%20have%20optional%20name%20hints%2C%20and%20are%20numbered%20in%20ascending%20order%20when%20printing%20if%20the%20name%20hint%20is%20missing.%0A%20%20%20%20%22%22%22)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(sum_of_squares_module)%3A%0A%20%20%20%20all_operands%20%3D%20set(val.name_hint%20for%20op%20in%20sum_of_squares_module.walk()%20for%20val%20in%20op.operands)%0A%20%20%20%20all_results%20%3D%20set(val.name_hint%20for%20op%20in%20sum_of_squares_module.walk()%20for%20val%20in%20op.results)%0A%20%20%20%20all_block_args%20%3D%20set(%0A%20%20%20%20%20%20%20%20val.name_hint%0A%20%20%20%20%20%20%20%20for%20op%20in%20sum_of_squares_module.walk()%0A%20%20%20%20%20%20%20%20for%20region%20in%20op.regions%0A%20%20%20%20%20%20%20%20for%20block%20in%20region.blocks%0A%20%20%20%20%20%20%20%20for%20val%20in%20block.args%0A%20%20%20%20)%0A%20%20%20%20all_ssa_values%20%3D%20all_operands%20%7C%20all_results%20%7C%20all_block_args%0A%20%20%20%20return%20(all_ssa_values%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(all_ssa_values%2C%20mo)%3A%0A%20%20%20%20mo.md(%0A%20%20%20%20%20%20%20%20fr%22%22%22%0A%20%20%20%20Here%20are%20all%20the%20name%20hints%20of%20SSA%20values%20in%20our%20module%3A%0A%0A%20%20%20%20%60%60%60%0A%20%20%20%20%7B%7B%7B'%2C%20'.join(sorted(all_ssa_values))%7D%7D%7D%0A%20%20%20%20%60%60%60%0A%20%20%20%20%22%22%22%0A%20%20%20%20)%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(definition_by_use%2C%20mo%2C%20sum_of_squares_module)%3A%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20_unsorted%20%3D%20definition_by_use(sum_of_squares_module)%0A%20%20%20%20%20%20%20%20_sorted%20%3D%20%7B%0A%20%20%20%20%20%20%20%20%20%20%20%20k%3A%20sorted(_unsorted%5Bk%5D)%0A%20%20%20%20%20%20%20%20%20%20%20%20for%20k%20in%20sorted(_unsorted)%0A%20%20%20%20%20%20%20%20%7D%0A%20%20%20%20%20%20%20%20_res%20%3D%20str(_sorted)%0A%20%20%20%20except%20Exception%20as%20e%3A%0A%20%20%20%20%20%20%20%20_res%20%3D%20f%22%7Btype(e).__name__%7D%3A%20%7Be%7D%22%0A%0A%20%20%20%20mo.md(%0A%20%20%20%20%20%20%20%20fr%22%22%22%0A%20%20%20%20%20%20%20%20%23%23%23%20Exercise%204.%20Definition%20By%20Use%0A%0A%20%20%20%20%20%20%20%20Modify%20the%20function%20below%20to%20return%20the%20operation%20that%20defines%20the%20value%20by%20the%20operation%20that%20uses%20it.%0A%20%20%20%20%20%20%20%20If%20the%20operand%20is%20a%20block%20argument%2C%20use%20the%20name%20of%20the%20parent%20operation.%0A%0A%20%20%20%20%20%20%20%20Note%3A%20the%20%60SSAValue%60%20class%20provides%20the%20%60owner%60%20helper%20that%20might%20be%20useful.%0A%0A%0A%20%20%20%20%20%20%20%20%60%60%60%0A%20%20%20%20%20%20%20%20Expected%3A%20%7B%7B'arith.addi'%3A%20%5B'arith.muli'%2C%20'scf.for'%5D%2C%20'arith.muli'%3A%20%5B'scf.for'%2C%20'scf.for'%5D%2C%20'func.return'%3A%20%5B'scf.for'%5D%2C%20'scf.for'%3A%20%5B'arith.constant'%2C%20'arith.constant'%2C%20'arith.constant'%2C%20'func.func'%5D%2C%20'scf.yield'%3A%20%5B'arith.addi'%5D%7D%7D%0A%20%20%20%20%20%20%20%20Result%3A%20%20%20%7B_res%7D%0A%20%20%20%20%20%20%20%20%60%60%60%0A%20%20%20%20%20%20%20%20%22%22%22%0A%20%20%20%20)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(builtin)%3A%0A%20%20%20%20def%20definition_by_use(module%3A%20builtin.ModuleOp)%20-%3E%20dict%5Bstr%2C%20list%5Bstr%5D%5D%3A%0A%20%20%20%20%20%20%20%20return%20%7B%7D%0A%0A%20%20%20%20return%20(definition_by_use%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(OpResult%2C%20builtin%2C%20defaultdict)%3A%0A%20%20%20%20%23%20Solution%0A%0A%20%20%20%20def%20_definition_by_use(module%3A%20builtin.ModuleOp)%20-%3E%20dict%5Bstr%2C%20list%5Bstr%5D%5D%3A%0A%20%20%20%20%20%20%20%20res%20%3D%20defaultdict(list)%0A%0A%20%20%20%20%20%20%20%20for%20use_op%20in%20module.walk()%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20for%20operand%20in%20use_op.operands%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20if%20isinstance(operand%2C%20OpResult)%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20op%20%3D%20operand.op%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20else%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20op%20%3D%20operand.block.parent_op()%0A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20res%5Buse_op.name%5D.append(op.name)%0A%20%20%20%20%20%20%20%20return%20dict(res)%0A%0A%20%20%20%20return%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(mo%2C%20sum_of_squares_module%2C%20uses_by_definition)%3A%0A%20%20%20%20try%3A%0A%20%20%20%20%20%20%20%20_unsorted%20%3D%20uses_by_definition(sum_of_squares_module)%0A%20%20%20%20%20%20%20%20_sorted%20%3D%20%7B%0A%20%20%20%20%20%20%20%20%20%20%20%20k%3A%20sorted(_unsorted%5Bk%5D)%0A%20%20%20%20%20%20%20%20%20%20%20%20for%20k%20in%20sorted(_unsorted)%0A%20%20%20%20%20%20%20%20%7D%0A%20%20%20%20%20%20%20%20_res%20%3D%20str(_sorted)%0A%20%20%20%20except%20Exception%20as%20e%3A%0A%20%20%20%20%20%20%20%20_res%20%3D%20f%22%7Btype(e).__name__%7D%3A%20%7Be%7D%22%0A%0A%20%20%20%20mo.md(%0A%20%20%20%20%20%20%20%20fr%22%22%22%0A%20%20%20%20%20%20%20%20%23%23%23%20Exercise%205.%20Uses%20By%20Definition%0A%0A%20%20%20%20%20%20%20%20Modify%20the%20function%20below%20to%20return%20the%20operations%20that%20use%20the%20result%20by%20the%20operation%20that%20defines%20it.%0A%0A%20%20%20%20%20%20%20%20Note%3A%20the%20%60SSAValue%60%20class%20provides%20a%20%60uses%60%20helper%20that%20might%20be%20useful.%0A%0A%20%20%20%20%20%20%20%20%60%60%60%0A%20%20%20%20%20%20%20%20Expected%3A%20%7B%7B'arith.addi'%3A%20%5B'scf.yield'%5D%2C%20'arith.constant'%3A%20%5B'scf.for'%2C%20'scf.for'%2C%20'scf.for'%5D%2C%20'arith.muli'%3A%20%5B'arith.addi'%5D%2C%20'scf.for'%3A%20%5B'func.return'%5D%7D%7D%0A%20%20%20%20%20%20%20%20Result%3A%20%20%20%7B_sorted%7D%0A%20%20%20%20%20%20%20%20%60%60%60%0A%20%20%20%20%20%20%20%20%22%22%22%0A%20%20%20%20)%0A%20%20%20%20return%0A%0A%0A%40app.cell%0Adef%20_(builtin)%3A%0A%20%20%20%20def%20uses_by_definition(module%3A%20builtin.ModuleOp)%20-%3E%20dict%5Bstr%2C%20list%5Bstr%5D%5D%3A%0A%20%20%20%20%20%20%20%20return%20%7B%7D%0A%0A%20%20%20%20return%20(uses_by_definition%2C)%0A%0A%0A%40app.cell(hide_code%3DTrue)%0Adef%20_(builtin%2C%20defaultdict)%3A%0A%20%20%20%20%23%20Solution%0A%0A%20%20%20%20def%20_uses_by_definition(module%3A%20builtin.ModuleOp)%20-%3E%20dict%5Bstr%2C%20list%5Bstr%5D%5D%3A%0A%20%20%20%20%20%20%20%20res%20%3D%20defaultdict(list)%0A%0A%20%20%20%20%20%20%20%20for%20def_op%20in%20module.walk()%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20for%20result%20in%20def_op.results%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20for%20use%20in%20result.uses%3A%0A%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20%20res%5Bdef_op.name%5D.append(use.operation.name)%0A%0A%20%20%20%20%20%20%20%20return%20dict(res)%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