Created
March 1, 2021 15:02
-
-
Save seibert/0be8825985400acbae5681aef2b2ee79 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| { | |
| "cells": [ | |
| { | |
| "cell_type": "code", | |
| "execution_count": 1, | |
| "id": "confidential-vancouver", | |
| "metadata": {}, | |
| "outputs": [], | |
| "source": [ | |
| "from mlir_graphblas import MlirOptCli\n", | |
| "import logging\n", | |
| "logging.basicConfig()\n", | |
| "logging.getLogger('mlir_graphblas').setLevel(logging.DEBUG)" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 2, | |
| "id": "dutch-diagnosis", | |
| "metadata": {}, | |
| "outputs": [], | |
| "source": [ | |
| "opt = MlirOptCli()" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 3, | |
| "id": "cardiac-cabin", | |
| "metadata": {}, | |
| "outputs": [], | |
| "source": [ | |
| "add_sd_mlir = b'''\\\n", | |
| "#trait_add = {\n", | |
| " indexing_maps = [\n", | |
| " affine_map<(i,j) -> (i,j)>, // A\n", | |
| " affine_map<(i,j) -> (i,j)>, // B\n", | |
| " affine_map<(i,j) -> (i,j)> // X (out)\n", | |
| " ],\n", | |
| " //sparse = [\n", | |
| " // [ \"D\", \"D\" ], // A\n", | |
| " // [ \"D\", \"D\" ], // B\n", | |
| " // [ \"D\", \"D\" ] // X\n", | |
| " //],\n", | |
| " iterator_types = [\"parallel\", \"parallel\"],\n", | |
| " doc = \"X(i,j) = A(i,j) OP B(i,j)\"\n", | |
| "}\n", | |
| "func @add_func(%arga: tensor<32x16xf32>, %argb: tensor<32x16xf32>) -> tensor<32x16xf32> {\n", | |
| " %0 = linalg.generic #trait_add\n", | |
| " ins(%arga, %argb: tensor<32x16xf32>, tensor<32x16xf32>)\n", | |
| " outs(%arga: tensor<32x16xf32>) {\n", | |
| " ^bb(%a: f32, %b: f32, %s: f32):\n", | |
| " %0 = addf %a, %b : f32\n", | |
| " linalg.yield %0 : f32\n", | |
| " } -> tensor<32x16xf32>\n", | |
| " return %0 : tensor<32x16xf32>\n", | |
| "}\n", | |
| "'''" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 4, | |
| "id": "driven-medium", | |
| "metadata": {}, | |
| "outputs": [ | |
| { | |
| "name": "stderr", | |
| "output_type": "stream", | |
| "text": [ | |
| "DEBUG:mlir_graphblas:RUN: ['mlir-opt', '--linalg-bufferize']\n", | |
| "DEBUG:mlir_graphblas:RUN: ['mlir-opt', '--func-bufferize']\n", | |
| "DEBUG:mlir_graphblas:RUN: ['mlir-opt', '--finalizing-bufferize']\n", | |
| "DEBUG:mlir_graphblas:RUN: ['mlir-opt', '--convert-linalg-to-loops']\n", | |
| "DEBUG:mlir_graphblas:RUN: ['mlir-opt', '--convert-scf-to-std']\n", | |
| "DEBUG:mlir_graphblas:RUN: ['mlir-opt', '--convert-std-to-llvm']\n" | |
| ] | |
| } | |
| ], | |
| "source": [ | |
| "passes = [\n", | |
| " '--linalg-bufferize',\n", | |
| " '--func-bufferize',\n", | |
| " '--finalizing-bufferize',\n", | |
| " '--convert-linalg-to-loops',\n", | |
| " '--convert-scf-to-std',\n", | |
| " '--convert-std-to-llvm',\n", | |
| "]\n", | |
| "result = opt.debug_passes(add_sd_mlir, passes)" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 33, | |
| "id": "absent-oakland", | |
| "metadata": { | |
| "scrolled": false | |
| }, | |
| "outputs": [ | |
| { | |
| "data": { | |
| "application/javascript": [ | |
| "\n", | |
| "(function(root) {\n", | |
| " function now() {\n", | |
| " return new Date();\n", | |
| " }\n", | |
| "\n", | |
| " var force = true;\n", | |
| "\n", | |
| " if (typeof root._bokeh_onload_callbacks === \"undefined\" || force === true) {\n", | |
| " root._bokeh_onload_callbacks = [];\n", | |
| " root._bokeh_is_loading = undefined;\n", | |
| " }\n", | |
| "\n", | |
| " if (typeof (root._bokeh_timeout) === \"undefined\" || force === true) {\n", | |
| " root._bokeh_timeout = Date.now() + 5000;\n", | |
| " root._bokeh_failed_load = false;\n", | |
| " }\n", | |
| "\n", | |
| " function run_callbacks() {\n", | |
| " try {\n", | |
| " root._bokeh_onload_callbacks.forEach(function(callback) {\n", | |
| " if (callback != null)\n", | |
| " callback();\n", | |
| " });\n", | |
| " } finally {\n", | |
| " delete root._bokeh_onload_callbacks\n", | |
| " }\n", | |
| " console.debug(\"Bokeh: all callbacks have finished\");\n", | |
| " }\n", | |
| "\n", | |
| " function load_libs(css_urls, js_urls, callback) {\n", | |
| " if (css_urls == null) css_urls = [];\n", | |
| " if (js_urls == null) js_urls = [];\n", | |
| "\n", | |
| " root._bokeh_onload_callbacks.push(callback);\n", | |
| " if (root._bokeh_is_loading > 0) {\n", | |
| " console.debug(\"Bokeh: BokehJS is being loaded, scheduling callback at\", now());\n", | |
| " return null;\n", | |
| " }\n", | |
| " if (js_urls == null || js_urls.length === 0) {\n", | |
| " run_callbacks();\n", | |
| " return null;\n", | |
| " }\n", | |
| " console.debug(\"Bokeh: BokehJS not loaded, scheduling load and callback at\", now());\n", | |
| " root._bokeh_is_loading = css_urls.length + js_urls.length;\n", | |
| "\n", | |
| " function on_load() {\n", | |
| " root._bokeh_is_loading--;\n", | |
| " if (root._bokeh_is_loading === 0) {\n", | |
| " console.debug(\"Bokeh: all BokehJS libraries/stylesheets loaded\");\n", | |
| " run_callbacks()\n", | |
| " }\n", | |
| " }\n", | |
| "\n", | |
| " function on_error() {\n", | |
| " console.error(\"failed to load \" + url);\n", | |
| " }\n", | |
| "\n", | |
| " for (var i = 0; i < css_urls.length; i++) {\n", | |
| " var url = css_urls[i];\n", | |
| " const element = document.createElement(\"link\");\n", | |
| " element.onload = on_load;\n", | |
| " element.onerror = on_error;\n", | |
| " element.rel = \"stylesheet\";\n", | |
| " element.type = \"text/css\";\n", | |
| " element.href = url;\n", | |
| " console.debug(\"Bokeh: injecting link tag for BokehJS stylesheet: \", url);\n", | |
| " document.body.appendChild(element);\n", | |
| " }\n", | |
| "\n", | |
| " var skip = [];\n", | |
| " if (window.requirejs) {\n", | |
| " require([], function() {\n", | |
| " })\n", | |
| " }\n", | |
| " for (var i = 0; i < js_urls.length; i++) {\n", | |
| " var url = js_urls[i];\n", | |
| " if (skip.indexOf(url) >= 0) { on_load(); continue; }\n", | |
| " var element = document.createElement('script');\n", | |
| " element.onload = on_load;\n", | |
| " element.onerror = on_error;\n", | |
| " element.async = false;\n", | |
| " element.src = url;\n", | |
| " console.debug(\"Bokeh: injecting script tag for BokehJS library: \", url);\n", | |
| " document.head.appendChild(element);\n", | |
| " }\n", | |
| "\tif (!js_urls.length) {\n", | |
| " on_load()\n", | |
| " }\n", | |
| " };\n", | |
| "\n", | |
| " function inject_raw_css(css) {\n", | |
| " const element = document.createElement(\"style\");\n", | |
| " element.appendChild(document.createTextNode(css));\n", | |
| " document.body.appendChild(element);\n", | |
| " }\n", | |
| "\n", | |
| " var js_urls = [\"https://cdn.bokeh.org/bokeh/release/bokeh-2.2.3.min.js\", \"https://cdn.bokeh.org/bokeh/release/bokeh-widgets-2.2.3.min.js\", \"https://cdn.bokeh.org/bokeh/release/bokeh-tables-2.2.3.min.js\", \"https://unpkg.com/@holoviz/panel@^0.10.3/dist/panel.min.js\"];\n", | |
| " var css_urls = [\"https://unpkg.com/@holoviz/[email protected]/dist/css/alerts.css\", \"https://unpkg.com/@holoviz/[email protected]/dist/css/card.css\", \"https://unpkg.com/@holoviz/[email protected]/dist/css/widgets.css\", \"https://unpkg.com/@holoviz/[email protected]/dist/css/markdown.css\", \"https://unpkg.com/@holoviz/[email protected]/dist/css/json.css\", \"https://unpkg.com/@holoviz/[email protected]/dist/css/dataframe.css\"];\n", | |
| "\n", | |
| " var inline_js = [\n", | |
| " function(Bokeh) {\n", | |
| " Bokeh.set_log_level(\"info\");\n", | |
| " },\n", | |
| " function(Bokeh) {} // ensure no trailing comma for IE\n", | |
| " ];\n", | |
| "\n", | |
| " function run_inline_js() {\n", | |
| " if ((root.Bokeh !== undefined) || (force === true)) {\n", | |
| " for (var i = 0; i < inline_js.length; i++) {\n", | |
| " inline_js[i].call(root, root.Bokeh);\n", | |
| " }} else if (Date.now() < root._bokeh_timeout) {\n", | |
| " setTimeout(run_inline_js, 100);\n", | |
| " } else if (!root._bokeh_failed_load) {\n", | |
| " console.log(\"Bokeh: BokehJS failed to load within specified timeout.\");\n", | |
| " root._bokeh_failed_load = true;\n", | |
| " }\n", | |
| " }\n", | |
| "\n", | |
| " if (root._bokeh_is_loading === 0) {\n", | |
| " console.debug(\"Bokeh: BokehJS loaded, going straight to plotting\");\n", | |
| " run_inline_js();\n", | |
| " } else {\n", | |
| " load_libs(css_urls, js_urls, function() {\n", | |
| " console.debug(\"Bokeh: BokehJS plotting callback run at\", now());\n", | |
| " run_inline_js();\n", | |
| " });\n", | |
| " }\n", | |
| "}(window));" | |
| ], | |
| "application/vnd.holoviews_load.v0+json": "\n(function(root) {\n function now() {\n return new Date();\n }\n\n var force = true;\n\n if (typeof root._bokeh_onload_callbacks === \"undefined\" || force === true) {\n root._bokeh_onload_callbacks = [];\n root._bokeh_is_loading = undefined;\n }\n\n if (typeof (root._bokeh_timeout) === \"undefined\" || force === true) {\n root._bokeh_timeout = Date.now() + 5000;\n root._bokeh_failed_load = false;\n }\n\n function run_callbacks() {\n try {\n root._bokeh_onload_callbacks.forEach(function(callback) {\n if (callback != null)\n callback();\n });\n } finally {\n delete root._bokeh_onload_callbacks\n }\n console.debug(\"Bokeh: all callbacks have finished\");\n }\n\n function load_libs(css_urls, js_urls, callback) {\n if (css_urls == null) css_urls = [];\n if (js_urls == null) js_urls = [];\n\n root._bokeh_onload_callbacks.push(callback);\n if (root._bokeh_is_loading > 0) {\n console.debug(\"Bokeh: BokehJS is being loaded, scheduling callback at\", now());\n return null;\n }\n if (js_urls == null || js_urls.length === 0) {\n run_callbacks();\n return null;\n }\n console.debug(\"Bokeh: BokehJS not loaded, scheduling load and callback at\", now());\n root._bokeh_is_loading = css_urls.length + js_urls.length;\n\n function on_load() {\n root._bokeh_is_loading--;\n if (root._bokeh_is_loading === 0) {\n console.debug(\"Bokeh: all BokehJS libraries/stylesheets loaded\");\n run_callbacks()\n }\n }\n\n function on_error() {\n console.error(\"failed to load \" + url);\n }\n\n for (var i = 0; i < css_urls.length; i++) {\n var url = css_urls[i];\n const element = document.createElement(\"link\");\n element.onload = on_load;\n element.onerror = on_error;\n element.rel = \"stylesheet\";\n element.type = \"text/css\";\n element.href = url;\n console.debug(\"Bokeh: injecting link tag for BokehJS stylesheet: \", url);\n document.body.appendChild(element);\n }\n\n var skip = [];\n if (window.requirejs) {\n require([], function() {\n })\n }\n for (var i = 0; i < js_urls.length; i++) {\n var url = js_urls[i];\n if (skip.indexOf(url) >= 0) { on_load(); continue; }\n var element = document.createElement('script');\n element.onload = on_load;\n element.onerror = on_error;\n element.async = false;\n element.src = url;\n console.debug(\"Bokeh: injecting script tag for BokehJS library: \", url);\n document.head.appendChild(element);\n }\n\tif (!js_urls.length) {\n on_load()\n }\n };\n\n function inject_raw_css(css) {\n const element = document.createElement(\"style\");\n element.appendChild(document.createTextNode(css));\n document.body.appendChild(element);\n }\n\n var js_urls = [\"https://cdn.bokeh.org/bokeh/release/bokeh-2.2.3.min.js\", \"https://cdn.bokeh.org/bokeh/release/bokeh-widgets-2.2.3.min.js\", \"https://cdn.bokeh.org/bokeh/release/bokeh-tables-2.2.3.min.js\", \"https://unpkg.com/@holoviz/panel@^0.10.3/dist/panel.min.js\"];\n var css_urls = [\"https://unpkg.com/@holoviz/[email protected]/dist/css/alerts.css\", \"https://unpkg.com/@holoviz/[email protected]/dist/css/card.css\", \"https://unpkg.com/@holoviz/[email protected]/dist/css/widgets.css\", \"https://unpkg.com/@holoviz/[email protected]/dist/css/markdown.css\", \"https://unpkg.com/@holoviz/[email protected]/dist/css/json.css\", \"https://unpkg.com/@holoviz/[email protected]/dist/css/dataframe.css\"];\n\n var inline_js = [\n function(Bokeh) {\n Bokeh.set_log_level(\"info\");\n },\n function(Bokeh) {} // ensure no trailing comma for IE\n ];\n\n function run_inline_js() {\n if ((root.Bokeh !== undefined) || (force === true)) {\n for (var i = 0; i < inline_js.length; i++) {\n inline_js[i].call(root, root.Bokeh);\n }} else if (Date.now() < root._bokeh_timeout) {\n setTimeout(run_inline_js, 100);\n } else if (!root._bokeh_failed_load) {\n console.log(\"Bokeh: BokehJS failed to load within specified timeout.\");\n root._bokeh_failed_load = true;\n }\n }\n\n if (root._bokeh_is_loading === 0) {\n console.debug(\"Bokeh: BokehJS loaded, going straight to plotting\");\n run_inline_js();\n } else {\n load_libs(css_urls, js_urls, function() {\n console.debug(\"Bokeh: BokehJS plotting callback run at\", now());\n run_inline_js();\n });\n }\n}(window));" | |
| }, | |
| "metadata": {}, | |
| "output_type": "display_data" | |
| }, | |
| { | |
| "data": { | |
| "application/javascript": [ | |
| "\n", | |
| "if ((window.PyViz === undefined) || (window.PyViz instanceof HTMLElement)) {\n", | |
| " window.PyViz = {comms: {}, comm_status:{}, kernels:{}, receivers: {}, plot_index: []}\n", | |
| "}\n", | |
| "\n", | |
| "\n", | |
| " function JupyterCommManager() {\n", | |
| " }\n", | |
| "\n", | |
| " JupyterCommManager.prototype.register_target = function(plot_id, comm_id, msg_handler) {\n", | |
| " if (window.comm_manager || ((window.Jupyter !== undefined) && (Jupyter.notebook.kernel != null))) {\n", | |
| " var comm_manager = window.comm_manager || Jupyter.notebook.kernel.comm_manager;\n", | |
| " comm_manager.register_target(comm_id, function(comm) {\n", | |
| " comm.on_msg(msg_handler);\n", | |
| " });\n", | |
| " } else if ((plot_id in window.PyViz.kernels) && (window.PyViz.kernels[plot_id])) {\n", | |
| " window.PyViz.kernels[plot_id].registerCommTarget(comm_id, function(comm) {\n", | |
| " comm.onMsg = msg_handler;\n", | |
| " });\n", | |
| " } else if (typeof google != 'undefined' && google.colab.kernel != null) {\n", | |
| " google.colab.kernel.comms.registerTarget(comm_id, (comm) => {\n", | |
| " var messages = comm.messages[Symbol.asyncIterator]();\n", | |
| " function processIteratorResult(result) {\n", | |
| " var message = result.value;\n", | |
| " console.log(message)\n", | |
| " var content = {data: message.data, comm_id};\n", | |
| " var buffers = []\n", | |
| " for (var buffer of message.buffers || []) {\n", | |
| " buffers.push(new DataView(buffer))\n", | |
| " }\n", | |
| " var metadata = message.metadata || {};\n", | |
| " var msg = {content, buffers, metadata}\n", | |
| " msg_handler(msg);\n", | |
| " return messages.next().then(processIteratorResult);\n", | |
| " }\n", | |
| " return messages.next().then(processIteratorResult);\n", | |
| " })\n", | |
| " }\n", | |
| " }\n", | |
| "\n", | |
| " JupyterCommManager.prototype.get_client_comm = function(plot_id, comm_id, msg_handler) {\n", | |
| " if (comm_id in window.PyViz.comms) {\n", | |
| " return window.PyViz.comms[comm_id];\n", | |
| " } else if (window.comm_manager || ((window.Jupyter !== undefined) && (Jupyter.notebook.kernel != null))) {\n", | |
| " var comm_manager = window.comm_manager || Jupyter.notebook.kernel.comm_manager;\n", | |
| " var comm = comm_manager.new_comm(comm_id, {}, {}, {}, comm_id);\n", | |
| " if (msg_handler) {\n", | |
| " comm.on_msg(msg_handler);\n", | |
| " }\n", | |
| " } else if ((plot_id in window.PyViz.kernels) && (window.PyViz.kernels[plot_id])) {\n", | |
| " var comm = window.PyViz.kernels[plot_id].connectToComm(comm_id);\n", | |
| " comm.open();\n", | |
| " if (msg_handler) {\n", | |
| " comm.onMsg = msg_handler;\n", | |
| " }\n", | |
| " } else if (typeof google != 'undefined' && google.colab.kernel != null) {\n", | |
| " var comm_promise = google.colab.kernel.comms.open(comm_id)\n", | |
| " comm_promise.then((comm) => {\n", | |
| " window.PyViz.comms[comm_id] = comm;\n", | |
| " if (msg_handler) {\n", | |
| " var messages = comm.messages[Symbol.asyncIterator]();\n", | |
| " function processIteratorResult(result) {\n", | |
| " var message = result.value;\n", | |
| " var content = {data: message.data};\n", | |
| " var metadata = message.metadata || {comm_id};\n", | |
| " var msg = {content, metadata}\n", | |
| " msg_handler(msg);\n", | |
| " return messages.next().then(processIteratorResult);\n", | |
| " }\n", | |
| " return messages.next().then(processIteratorResult);\n", | |
| " }\n", | |
| " }) \n", | |
| " var sendClosure = (data, metadata, buffers, disposeOnDone) => {\n", | |
| " return comm_promise.then((comm) => {\n", | |
| " comm.send(data, metadata, buffers, disposeOnDone);\n", | |
| " });\n", | |
| " };\n", | |
| " var comm = {\n", | |
| " send: sendClosure\n", | |
| " };\n", | |
| " }\n", | |
| " window.PyViz.comms[comm_id] = comm;\n", | |
| " return comm;\n", | |
| " }\n", | |
| " window.PyViz.comm_manager = new JupyterCommManager();\n", | |
| " \n", | |
| "\n", | |
| "\n", | |
| "var JS_MIME_TYPE = 'application/javascript';\n", | |
| "var HTML_MIME_TYPE = 'text/html';\n", | |
| "var EXEC_MIME_TYPE = 'application/vnd.holoviews_exec.v0+json';\n", | |
| "var CLASS_NAME = 'output';\n", | |
| "\n", | |
| "/**\n", | |
| " * Render data to the DOM node\n", | |
| " */\n", | |
| "function render(props, node) {\n", | |
| " var div = document.createElement(\"div\");\n", | |
| " var script = document.createElement(\"script\");\n", | |
| " node.appendChild(div);\n", | |
| " node.appendChild(script);\n", | |
| "}\n", | |
| "\n", | |
| "/**\n", | |
| " * Handle when a new output is added\n", | |
| " */\n", | |
| "function handle_add_output(event, handle) {\n", | |
| " var output_area = handle.output_area;\n", | |
| " var output = handle.output;\n", | |
| " if ((output.data == undefined) || (!output.data.hasOwnProperty(EXEC_MIME_TYPE))) {\n", | |
| " return\n", | |
| " }\n", | |
| " var id = output.metadata[EXEC_MIME_TYPE][\"id\"];\n", | |
| " var toinsert = output_area.element.find(\".\" + CLASS_NAME.split(' ')[0]);\n", | |
| " if (id !== undefined) {\n", | |
| " var nchildren = toinsert.length;\n", | |
| " var html_node = toinsert[nchildren-1].children[0];\n", | |
| " html_node.innerHTML = output.data[HTML_MIME_TYPE];\n", | |
| " var scripts = [];\n", | |
| " var nodelist = html_node.querySelectorAll(\"script\");\n", | |
| " for (var i in nodelist) {\n", | |
| " if (nodelist.hasOwnProperty(i)) {\n", | |
| " scripts.push(nodelist[i])\n", | |
| " }\n", | |
| " }\n", | |
| "\n", | |
| " scripts.forEach( function (oldScript) {\n", | |
| " var newScript = document.createElement(\"script\");\n", | |
| " var attrs = [];\n", | |
| " var nodemap = oldScript.attributes;\n", | |
| " for (var j in nodemap) {\n", | |
| " if (nodemap.hasOwnProperty(j)) {\n", | |
| " attrs.push(nodemap[j])\n", | |
| " }\n", | |
| " }\n", | |
| " attrs.forEach(function(attr) { newScript.setAttribute(attr.name, attr.value) });\n", | |
| " newScript.appendChild(document.createTextNode(oldScript.innerHTML));\n", | |
| " oldScript.parentNode.replaceChild(newScript, oldScript);\n", | |
| " });\n", | |
| " if (JS_MIME_TYPE in output.data) {\n", | |
| " toinsert[nchildren-1].children[1].textContent = output.data[JS_MIME_TYPE];\n", | |
| " }\n", | |
| " output_area._hv_plot_id = id;\n", | |
| " if ((window.Bokeh !== undefined) && (id in Bokeh.index)) {\n", | |
| " window.PyViz.plot_index[id] = Bokeh.index[id];\n", | |
| " } else {\n", | |
| " window.PyViz.plot_index[id] = null;\n", | |
| " }\n", | |
| " } else if (output.metadata[EXEC_MIME_TYPE][\"server_id\"] !== undefined) {\n", | |
| " var bk_div = document.createElement(\"div\");\n", | |
| " bk_div.innerHTML = output.data[HTML_MIME_TYPE];\n", | |
| " var script_attrs = bk_div.children[0].attributes;\n", | |
| " for (var i = 0; i < script_attrs.length; i++) {\n", | |
| " toinsert[toinsert.length - 1].childNodes[1].setAttribute(script_attrs[i].name, script_attrs[i].value);\n", | |
| " }\n", | |
| " // store reference to server id on output_area\n", | |
| " output_area._bokeh_server_id = output.metadata[EXEC_MIME_TYPE][\"server_id\"];\n", | |
| " }\n", | |
| "}\n", | |
| "\n", | |
| "/**\n", | |
| " * Handle when an output is cleared or removed\n", | |
| " */\n", | |
| "function handle_clear_output(event, handle) {\n", | |
| " var id = handle.cell.output_area._hv_plot_id;\n", | |
| " var server_id = handle.cell.output_area._bokeh_server_id;\n", | |
| " if (((id === undefined) || !(id in PyViz.plot_index)) && (server_id !== undefined)) { return; }\n", | |
| " var comm = window.PyViz.comm_manager.get_client_comm(\"hv-extension-comm\", \"hv-extension-comm\", function () {});\n", | |
| " if (server_id !== null) {\n", | |
| " comm.send({event_type: 'server_delete', 'id': server_id});\n", | |
| " return;\n", | |
| " } else if (comm !== null) {\n", | |
| " comm.send({event_type: 'delete', 'id': id});\n", | |
| " }\n", | |
| " delete PyViz.plot_index[id];\n", | |
| " if ((window.Bokeh !== undefined) & (id in window.Bokeh.index)) {\n", | |
| " var doc = window.Bokeh.index[id].model.document\n", | |
| " doc.clear();\n", | |
| " const i = window.Bokeh.documents.indexOf(doc);\n", | |
| " if (i > -1) {\n", | |
| " window.Bokeh.documents.splice(i, 1);\n", | |
| " }\n", | |
| " }\n", | |
| "}\n", | |
| "\n", | |
| "/**\n", | |
| " * Handle kernel restart event\n", | |
| " */\n", | |
| "function handle_kernel_cleanup(event, handle) {\n", | |
| " delete PyViz.comms[\"hv-extension-comm\"];\n", | |
| " window.PyViz.plot_index = {}\n", | |
| "}\n", | |
| "\n", | |
| "/**\n", | |
| " * Handle update_display_data messages\n", | |
| " */\n", | |
| "function handle_update_output(event, handle) {\n", | |
| " handle_clear_output(event, {cell: {output_area: handle.output_area}})\n", | |
| " handle_add_output(event, handle)\n", | |
| "}\n", | |
| "\n", | |
| "function register_renderer(events, OutputArea) {\n", | |
| " function append_mime(data, metadata, element) {\n", | |
| " // create a DOM node to render to\n", | |
| " var toinsert = this.create_output_subarea(\n", | |
| " metadata,\n", | |
| " CLASS_NAME,\n", | |
| " EXEC_MIME_TYPE\n", | |
| " );\n", | |
| " this.keyboard_manager.register_events(toinsert);\n", | |
| " // Render to node\n", | |
| " var props = {data: data, metadata: metadata[EXEC_MIME_TYPE]};\n", | |
| " render(props, toinsert[0]);\n", | |
| " element.append(toinsert);\n", | |
| " return toinsert\n", | |
| " }\n", | |
| "\n", | |
| " events.on('output_added.OutputArea', handle_add_output);\n", | |
| " events.on('output_updated.OutputArea', handle_update_output);\n", | |
| " events.on('clear_output.CodeCell', handle_clear_output);\n", | |
| " events.on('delete.Cell', handle_clear_output);\n", | |
| " events.on('kernel_ready.Kernel', handle_kernel_cleanup);\n", | |
| "\n", | |
| " OutputArea.prototype.register_mime_type(EXEC_MIME_TYPE, append_mime, {\n", | |
| " safe: true,\n", | |
| " index: 0\n", | |
| " });\n", | |
| "}\n", | |
| "\n", | |
| "if (window.Jupyter !== undefined) {\n", | |
| " try {\n", | |
| " var events = require('base/js/events');\n", | |
| " var OutputArea = require('notebook/js/outputarea').OutputArea;\n", | |
| " if (OutputArea.prototype.mime_types().indexOf(EXEC_MIME_TYPE) == -1) {\n", | |
| " register_renderer(events, OutputArea);\n", | |
| " }\n", | |
| " } catch(err) {\n", | |
| " }\n", | |
| "}\n" | |
| ], | |
| "application/vnd.holoviews_load.v0+json": "\nif ((window.PyViz === undefined) || (window.PyViz instanceof HTMLElement)) {\n window.PyViz = {comms: {}, comm_status:{}, kernels:{}, receivers: {}, plot_index: []}\n}\n\n\n function JupyterCommManager() {\n }\n\n JupyterCommManager.prototype.register_target = function(plot_id, comm_id, msg_handler) {\n if (window.comm_manager || ((window.Jupyter !== undefined) && (Jupyter.notebook.kernel != null))) {\n var comm_manager = window.comm_manager || Jupyter.notebook.kernel.comm_manager;\n comm_manager.register_target(comm_id, function(comm) {\n comm.on_msg(msg_handler);\n });\n } else if ((plot_id in window.PyViz.kernels) && (window.PyViz.kernels[plot_id])) {\n window.PyViz.kernels[plot_id].registerCommTarget(comm_id, function(comm) {\n comm.onMsg = msg_handler;\n });\n } else if (typeof google != 'undefined' && google.colab.kernel != null) {\n google.colab.kernel.comms.registerTarget(comm_id, (comm) => {\n var messages = comm.messages[Symbol.asyncIterator]();\n function processIteratorResult(result) {\n var message = result.value;\n console.log(message)\n var content = {data: message.data, comm_id};\n var buffers = []\n for (var buffer of message.buffers || []) {\n buffers.push(new DataView(buffer))\n }\n var metadata = message.metadata || {};\n var msg = {content, buffers, metadata}\n msg_handler(msg);\n return messages.next().then(processIteratorResult);\n }\n return messages.next().then(processIteratorResult);\n })\n }\n }\n\n JupyterCommManager.prototype.get_client_comm = function(plot_id, comm_id, msg_handler) {\n if (comm_id in window.PyViz.comms) {\n return window.PyViz.comms[comm_id];\n } else if (window.comm_manager || ((window.Jupyter !== undefined) && (Jupyter.notebook.kernel != null))) {\n var comm_manager = window.comm_manager || Jupyter.notebook.kernel.comm_manager;\n var comm = comm_manager.new_comm(comm_id, {}, {}, {}, comm_id);\n if (msg_handler) {\n comm.on_msg(msg_handler);\n }\n } else if ((plot_id in window.PyViz.kernels) && (window.PyViz.kernels[plot_id])) {\n var comm = window.PyViz.kernels[plot_id].connectToComm(comm_id);\n comm.open();\n if (msg_handler) {\n comm.onMsg = msg_handler;\n }\n } else if (typeof google != 'undefined' && google.colab.kernel != null) {\n var comm_promise = google.colab.kernel.comms.open(comm_id)\n comm_promise.then((comm) => {\n window.PyViz.comms[comm_id] = comm;\n if (msg_handler) {\n var messages = comm.messages[Symbol.asyncIterator]();\n function processIteratorResult(result) {\n var message = result.value;\n var content = {data: message.data};\n var metadata = message.metadata || {comm_id};\n var msg = {content, metadata}\n msg_handler(msg);\n return messages.next().then(processIteratorResult);\n }\n return messages.next().then(processIteratorResult);\n }\n }) \n var sendClosure = (data, metadata, buffers, disposeOnDone) => {\n return comm_promise.then((comm) => {\n comm.send(data, metadata, buffers, disposeOnDone);\n });\n };\n var comm = {\n send: sendClosure\n };\n }\n window.PyViz.comms[comm_id] = comm;\n return comm;\n }\n window.PyViz.comm_manager = new JupyterCommManager();\n \n\n\nvar JS_MIME_TYPE = 'application/javascript';\nvar HTML_MIME_TYPE = 'text/html';\nvar EXEC_MIME_TYPE = 'application/vnd.holoviews_exec.v0+json';\nvar CLASS_NAME = 'output';\n\n/**\n * Render data to the DOM node\n */\nfunction render(props, node) {\n var div = document.createElement(\"div\");\n var script = document.createElement(\"script\");\n node.appendChild(div);\n node.appendChild(script);\n}\n\n/**\n * Handle when a new output is added\n */\nfunction handle_add_output(event, handle) {\n var output_area = handle.output_area;\n var output = handle.output;\n if ((output.data == undefined) || (!output.data.hasOwnProperty(EXEC_MIME_TYPE))) {\n return\n }\n var id = output.metadata[EXEC_MIME_TYPE][\"id\"];\n var toinsert = output_area.element.find(\".\" + CLASS_NAME.split(' ')[0]);\n if (id !== undefined) {\n var nchildren = toinsert.length;\n var html_node = toinsert[nchildren-1].children[0];\n html_node.innerHTML = output.data[HTML_MIME_TYPE];\n var scripts = [];\n var nodelist = html_node.querySelectorAll(\"script\");\n for (var i in nodelist) {\n if (nodelist.hasOwnProperty(i)) {\n scripts.push(nodelist[i])\n }\n }\n\n scripts.forEach( function (oldScript) {\n var newScript = document.createElement(\"script\");\n var attrs = [];\n var nodemap = oldScript.attributes;\n for (var j in nodemap) {\n if (nodemap.hasOwnProperty(j)) {\n attrs.push(nodemap[j])\n }\n }\n attrs.forEach(function(attr) { newScript.setAttribute(attr.name, attr.value) });\n newScript.appendChild(document.createTextNode(oldScript.innerHTML));\n oldScript.parentNode.replaceChild(newScript, oldScript);\n });\n if (JS_MIME_TYPE in output.data) {\n toinsert[nchildren-1].children[1].textContent = output.data[JS_MIME_TYPE];\n }\n output_area._hv_plot_id = id;\n if ((window.Bokeh !== undefined) && (id in Bokeh.index)) {\n window.PyViz.plot_index[id] = Bokeh.index[id];\n } else {\n window.PyViz.plot_index[id] = null;\n }\n } else if (output.metadata[EXEC_MIME_TYPE][\"server_id\"] !== undefined) {\n var bk_div = document.createElement(\"div\");\n bk_div.innerHTML = output.data[HTML_MIME_TYPE];\n var script_attrs = bk_div.children[0].attributes;\n for (var i = 0; i < script_attrs.length; i++) {\n toinsert[toinsert.length - 1].childNodes[1].setAttribute(script_attrs[i].name, script_attrs[i].value);\n }\n // store reference to server id on output_area\n output_area._bokeh_server_id = output.metadata[EXEC_MIME_TYPE][\"server_id\"];\n }\n}\n\n/**\n * Handle when an output is cleared or removed\n */\nfunction handle_clear_output(event, handle) {\n var id = handle.cell.output_area._hv_plot_id;\n var server_id = handle.cell.output_area._bokeh_server_id;\n if (((id === undefined) || !(id in PyViz.plot_index)) && (server_id !== undefined)) { return; }\n var comm = window.PyViz.comm_manager.get_client_comm(\"hv-extension-comm\", \"hv-extension-comm\", function () {});\n if (server_id !== null) {\n comm.send({event_type: 'server_delete', 'id': server_id});\n return;\n } else if (comm !== null) {\n comm.send({event_type: 'delete', 'id': id});\n }\n delete PyViz.plot_index[id];\n if ((window.Bokeh !== undefined) & (id in window.Bokeh.index)) {\n var doc = window.Bokeh.index[id].model.document\n doc.clear();\n const i = window.Bokeh.documents.indexOf(doc);\n if (i > -1) {\n window.Bokeh.documents.splice(i, 1);\n }\n }\n}\n\n/**\n * Handle kernel restart event\n */\nfunction handle_kernel_cleanup(event, handle) {\n delete PyViz.comms[\"hv-extension-comm\"];\n window.PyViz.plot_index = {}\n}\n\n/**\n * Handle update_display_data messages\n */\nfunction handle_update_output(event, handle) {\n handle_clear_output(event, {cell: {output_area: handle.output_area}})\n handle_add_output(event, handle)\n}\n\nfunction register_renderer(events, OutputArea) {\n function append_mime(data, metadata, element) {\n // create a DOM node to render to\n var toinsert = this.create_output_subarea(\n metadata,\n CLASS_NAME,\n EXEC_MIME_TYPE\n );\n this.keyboard_manager.register_events(toinsert);\n // Render to node\n var props = {data: data, metadata: metadata[EXEC_MIME_TYPE]};\n render(props, toinsert[0]);\n element.append(toinsert);\n return toinsert\n }\n\n events.on('output_added.OutputArea', handle_add_output);\n events.on('output_updated.OutputArea', handle_update_output);\n events.on('clear_output.CodeCell', handle_clear_output);\n events.on('delete.Cell', handle_clear_output);\n events.on('kernel_ready.Kernel', handle_kernel_cleanup);\n\n OutputArea.prototype.register_mime_type(EXEC_MIME_TYPE, append_mime, {\n safe: true,\n index: 0\n });\n}\n\nif (window.Jupyter !== undefined) {\n try {\n var events = require('base/js/events');\n var OutputArea = require('notebook/js/outputarea').OutputArea;\n if (OutputArea.prototype.mime_types().indexOf(EXEC_MIME_TYPE) == -1) {\n register_renderer(events, OutputArea);\n }\n } catch(err) {\n }\n}\n" | |
| }, | |
| "metadata": {}, | |
| "output_type": "display_data" | |
| }, | |
| { | |
| "data": {}, | |
| "metadata": {}, | |
| "output_type": "display_data" | |
| }, | |
| { | |
| "data": { | |
| "application/vnd.holoviews_exec.v0+json": "", | |
| "text/html": [ | |
| "<div id='1001'>\n", | |
| "\n", | |
| "\n", | |
| "\n", | |
| "\n", | |
| "\n", | |
| " <div class=\"bk-root\" id=\"06327318-09b9-4fc5-977f-e92aba9d7588\" data-root-id=\"1001\"></div>\n", | |
| "</div>\n", | |
| "<script type=\"application/javascript\">(function(root) {\n", | |
| " function embed_document(root) {\n", | |
| " var docs_json = {\"3edfffc3-603e-4fff-9a0e-b7b33a9cc9e7\":{\"roots\":{\"references\":[{\"attributes\":{\"button_type\":\"primary\",\"icon\":null,\"label\":\"\\u25c0\",\"margin\":[5,10,5,10],\"min_width\":200,\"subscribed_events\":[\"button_click\"],\"width\":200},\"id\":\"1005\",\"type\":\"Button\"},{\"attributes\":{\"children\":[{\"id\":\"1003\"},{\"id\":\"1004\"},{\"id\":\"1007\"}],\"margin\":[0,0,0,0],\"name\":\"Column00017\"},\"id\":\"1002\",\"type\":\"Column\"},{\"attributes\":{\"margin\":[0,0,0,0],\"tabs\":[{\"id\":\"1010\"},{\"id\":\"1014\"},{\"id\":\"1022\"}]},\"id\":\"1001\",\"type\":\"Tabs\"},{\"attributes\":{\"client_comm_id\":\"e53f4b8724be440d9d0577c84e459a0c\",\"comm_id\":\"2ee34e3606cc490c987ae8772c048076\",\"plot_id\":\"1001\"},\"id\":\"1023\",\"type\":\"panel.models.comm_manager.CommManager\"},{\"attributes\":{\"reload\":false},\"id\":\"1024\",\"type\":\"panel.models.location.Location\"},{\"attributes\":{\"children\":[{\"id\":\"1016\"},{\"id\":\"1019\"}],\"margin\":[0,0,0,0],\"name\":\"Row00030\"},\"id\":\"1015\",\"type\":\"Row\"},{\"attributes\":{\"margin\":[5,10,5,10],\"options\":[\"Initial\",\"linalg-bufferize\",\"func-bufferize\",\"finalizing-bufferize\",\"convert-linalg-to-loops\",\"convert-scf-to-std\",\"convert-std-to-llvm\"],\"title\":\"Passes\",\"value\":\"Initial\"},\"id\":\"1012\",\"type\":\"Select\"},{\"attributes\":{\"children\":[{\"id\":\"1008\"},{\"id\":\"1009\"}],\"margin\":[0,0,0,0],\"max_width\":400,\"name\":\"Row00016\"},\"id\":\"1007\",\"type\":\"Row\"},{\"attributes\":{\"child\":{\"id\":\"1011\"},\"name\":\"Column00021\",\"title\":\"Single\"},\"id\":\"1014\",\"type\":\"Panel\"},{\"attributes\":{\"margin\":[5,5,5,5],\"name\":\"HTML00027\",\"text\":\"\\n<html>\\n<head>\\n<script type="text/javascript"\\n src="https://cdnjs.cloudflare.com/ajax/libs/codemirror/5.52.2/codemirror.min.js">\\n</script>\\n\\n<script type="text/javascript"\\n src="https://cdnjs.cloudflare.com/ajax/libs/codemirror/5.52.2/mode/javascript/javascript.min.js">\\n</script>\\n</head>\\n<body>\\n\\n<div>\\n<textarea id="foobar_double2">\\n#map = affine_map<(d0, d1) -> (d0, d1)>\\nmodule {\\n func @add_func(%arg0: tensor<32x16xf32>, %arg1: tensor<32x16xf32>) -> tensor<32x16xf32> {\\n %0 = tensor_to_memref %arg0 : memref<32x16xf32>\\n %1 = tensor_to_memref %arg1 : memref<32x16xf32>\\n %2 = tensor_to_memref %arg0 : memref<32x16xf32>\\n %3 = alloc() : memref<32x16xf32>\\n linalg.generic {doc = "X(i,j) = A(i,j) OP B(i,j)", indexing_maps = [#map, #map, #map], iterator_types = ["parallel", "parallel"]} ins(%0, %1 : memref<32x16xf32>, memref<32x16xf32>) outs(%3 : memref<32x16xf32>) {\\n ^bb0(%arg2: f32, %arg3: f32, %arg4: f32): // no predecessors\\n %5 = addf %arg2, %arg3 : f32\\n linalg.yield %5 : f32\\n }\\n %4 = tensor_load %3 : memref<32x16xf32>\\n return %4 : tensor<32x16xf32>\\n }\\n}\\n\\n\\n</textarea>\\n</div>\\n\\n<script>\\nvar config = {\\n lineNumbers: true,\\n tabSize: 2,\\n readOnly: true,\\n mode: 'javascript',\\n};\\nvar myTextArea = document.querySelector('#foobar_double2');\\nvar cm = CodeMirror.fromTextArea(myTextArea, config);\\n</script>\\n</body>\\n</html>\\n\"},\"id\":\"1021\",\"type\":\"panel.models.markup.HTML\"},{\"attributes\":{\"children\":[{\"id\":\"1017\"},{\"id\":\"1018\"}],\"margin\":[0,0,0,0],\"name\":\"Column00025\"},\"id\":\"1016\",\"type\":\"Column\"},{\"attributes\":{\"child\":{\"id\":\"1002\"},\"name\":\"Column00017\",\"title\":\"Sequential\"},\"id\":\"1010\",\"type\":\"Panel\"},{\"attributes\":{\"margin\":[5,10,5,10],\"options\":[\"Initial\",\"linalg-bufferize\",\"func-bufferize\",\"finalizing-bufferize\",\"convert-linalg-to-loops\",\"convert-scf-to-std\",\"convert-std-to-llvm\"],\"title\":\"Passes\",\"value\":\"Initial\"},\"id\":\"1017\",\"type\":\"Select\"},{\"attributes\":{\"children\":[{\"id\":\"1005\"},{\"id\":\"1006\"}],\"margin\":[0,0,0,0],\"name\":\"Row00011\"},\"id\":\"1004\",\"type\":\"Row\"},{\"attributes\":{\"button_type\":\"primary\",\"icon\":null,\"label\":\"\\u25b6\",\"margin\":[5,10,5,10],\"min_width\":200,\"subscribed_events\":[\"button_click\"],\"width\":200},\"id\":\"1006\",\"type\":\"Button\"},{\"attributes\":{\"children\":[{\"id\":\"1020\"},{\"id\":\"1021\"}],\"margin\":[0,0,0,0],\"name\":\"Column00029\"},\"id\":\"1019\",\"type\":\"Column\"},{\"attributes\":{\"margin\":[5,10,5,10],\"options\":[\"linalg-bufferize\",\"func-bufferize\",\"finalizing-bufferize\",\"convert-linalg-to-loops\",\"convert-scf-to-std\",\"convert-std-to-llvm\"],\"title\":\"Passes\",\"value\":\"linalg-bufferize\"},\"id\":\"1003\",\"type\":\"Select\"},{\"attributes\":{\"child\":{\"id\":\"1015\"},\"name\":\"Row00030\",\"title\":\"Double\"},\"id\":\"1022\",\"type\":\"Panel\"},{\"attributes\":{\"children\":[{\"id\":\"1012\"},{\"id\":\"1013\"}],\"margin\":[0,0,0,0],\"name\":\"Column00021\"},\"id\":\"1011\",\"type\":\"Column\"},{\"attributes\":{\"margin\":[5,5,5,5],\"name\":\"HTML00014\",\"text\":\"\\n<html>\\n<head>\\n<script type="text/javascript"\\n src="https://cdnjs.cloudflare.com/ajax/libs/codemirror/5.52.2/codemirror.min.js">\\n</script>\\n\\n<script type="text/javascript"\\n src="https://cdnjs.cloudflare.com/ajax/libs/codemirror/5.52.2/mode/javascript/javascript.min.js">\\n</script>\\n</head>\\n<body>\\n\\n<div>\\n<textarea id="foobar_seq2">\\n#map = affine_map<(d0, d1) -> (d0, d1)>\\nmodule {\\n func @add_func(%arg0: tensor<32x16xf32>, %arg1: tensor<32x16xf32>) -> tensor<32x16xf32> {\\n %0 = tensor_to_memref %arg0 : memref<32x16xf32>\\n %1 = tensor_to_memref %arg1 : memref<32x16xf32>\\n %2 = tensor_to_memref %arg0 : memref<32x16xf32>\\n %3 = alloc() : memref<32x16xf32>\\n linalg.generic {doc = "X(i,j) = A(i,j) OP B(i,j)", indexing_maps = [#map, #map, #map], iterator_types = ["parallel", "parallel"]} ins(%0, %1 : memref<32x16xf32>, memref<32x16xf32>) outs(%3 : memref<32x16xf32>) {\\n ^bb0(%arg2: f32, %arg3: f32, %arg4: f32): // no predecessors\\n %5 = addf %arg2, %arg3 : f32\\n linalg.yield %5 : f32\\n }\\n %4 = tensor_load %3 : memref<32x16xf32>\\n return %4 : tensor<32x16xf32>\\n }\\n}\\n\\n\\n</textarea>\\n</div>\\n\\n<script>\\nvar config = {\\n lineNumbers: true,\\n tabSize: 2,\\n readOnly: true,\\n mode: 'javascript',\\n};\\nvar myTextArea = document.querySelector('#foobar_seq2');\\nvar cm = CodeMirror.fromTextArea(myTextArea, config);\\n</script>\\n</body>\\n</html>\\n\"},\"id\":\"1009\",\"type\":\"panel.models.markup.HTML\"},{\"attributes\":{\"margin\":[5,5,5,5],\"name\":\"HTML00012\",\"text\":\"\\n<html>\\n<head>\\n<script type="text/javascript"\\n src="https://cdnjs.cloudflare.com/ajax/libs/codemirror/5.52.2/codemirror.min.js">\\n</script>\\n\\n<script type="text/javascript"\\n src="https://cdnjs.cloudflare.com/ajax/libs/codemirror/5.52.2/mode/javascript/javascript.min.js">\\n</script>\\n</head>\\n<body>\\n\\n<div>\\n<textarea id="foobar_seq1">\\n#trait_add = {\\n indexing_maps = [\\n affine_map<(i,j) -> (i,j)>, // A\\n affine_map<(i,j) -> (i,j)>, // B\\n affine_map<(i,j) -> (i,j)> // X (out)\\n ],\\n //sparse = [\\n // [ "D", "D" ], // A\\n // [ "D", "D" ], // B\\n // [ "D", "D" ] // X\\n //],\\n iterator_types = ["parallel", "parallel"],\\n doc = "X(i,j) = A(i,j) OP B(i,j)"\\n}\\nfunc @add_func(%arga: tensor<32x16xf32>, %argb: tensor<32x16xf32>) -> tensor<32x16xf32> {\\n %0 = linalg.generic #trait_add\\n ins(%arga, %argb: tensor<32x16xf32>, tensor<32x16xf32>)\\n outs(%arga: tensor<32x16xf32>) {\\n ^bb(%a: f32, %b: f32, %s: f32):\\n %0 = addf %a, %b : f32\\n linalg.yield %0 : f32\\n } -> tensor<32x16xf32>\\n return %0 : tensor<32x16xf32>\\n}\\n\\n</textarea>\\n</div>\\n\\n<script>\\nvar config = {\\n lineNumbers: true,\\n tabSize: 2,\\n readOnly: true,\\n mode: 'javascript',\\n};\\nvar myTextArea = document.querySelector('#foobar_seq1');\\nvar cm = CodeMirror.fromTextArea(myTextArea, config);\\n</script>\\n</body>\\n</html>\\n\"},\"id\":\"1008\",\"type\":\"panel.models.markup.HTML\"},{\"attributes\":{\"margin\":[5,10,5,10],\"options\":[\"Initial\",\"linalg-bufferize\",\"func-bufferize\",\"finalizing-bufferize\",\"convert-linalg-to-loops\",\"convert-scf-to-std\",\"convert-std-to-llvm\"],\"title\":\"Passes\",\"value\":\"linalg-bufferize\"},\"id\":\"1020\",\"type\":\"Select\"},{\"attributes\":{\"margin\":[5,5,5,5],\"name\":\"HTML00023\",\"text\":\"\\n<html>\\n<head>\\n<script type="text/javascript"\\n src="https://cdnjs.cloudflare.com/ajax/libs/codemirror/5.52.2/codemirror.min.js">\\n</script>\\n\\n<script type="text/javascript"\\n src="https://cdnjs.cloudflare.com/ajax/libs/codemirror/5.52.2/mode/javascript/javascript.min.js">\\n</script>\\n</head>\\n<body>\\n\\n<div>\\n<textarea id="foobar_double1">\\n#trait_add = {\\n indexing_maps = [\\n affine_map<(i,j) -> (i,j)>, // A\\n affine_map<(i,j) -> (i,j)>, // B\\n affine_map<(i,j) -> (i,j)> // X (out)\\n ],\\n //sparse = [\\n // [ "D", "D" ], // A\\n // [ "D", "D" ], // B\\n // [ "D", "D" ] // X\\n //],\\n iterator_types = ["parallel", "parallel"],\\n doc = "X(i,j) = A(i,j) OP B(i,j)"\\n}\\nfunc @add_func(%arga: tensor<32x16xf32>, %argb: tensor<32x16xf32>) -> tensor<32x16xf32> {\\n %0 = linalg.generic #trait_add\\n ins(%arga, %argb: tensor<32x16xf32>, tensor<32x16xf32>)\\n outs(%arga: tensor<32x16xf32>) {\\n ^bb(%a: f32, %b: f32, %s: f32):\\n %0 = addf %a, %b : f32\\n linalg.yield %0 : f32\\n } -> tensor<32x16xf32>\\n return %0 : tensor<32x16xf32>\\n}\\n\\n</textarea>\\n</div>\\n\\n<script>\\nvar config = {\\n lineNumbers: true,\\n tabSize: 2,\\n readOnly: true,\\n mode: 'javascript',\\n};\\nvar myTextArea = document.querySelector('#foobar_double1');\\nvar cm = CodeMirror.fromTextArea(myTextArea, config);\\n</script>\\n</body>\\n</html>\\n\"},\"id\":\"1018\",\"type\":\"panel.models.markup.HTML\"},{\"attributes\":{\"margin\":[5,5,5,5],\"name\":\"HTML00019\",\"text\":\"\\n<html>\\n<head>\\n<script type="text/javascript"\\n src="https://cdnjs.cloudflare.com/ajax/libs/codemirror/5.52.2/codemirror.min.js">\\n</script>\\n\\n<script type="text/javascript"\\n src="https://cdnjs.cloudflare.com/ajax/libs/codemirror/5.52.2/mode/javascript/javascript.min.js">\\n</script>\\n</head>\\n<body>\\n\\n<div>\\n<textarea id="foobar_single">\\n#trait_add = {\\n indexing_maps = [\\n affine_map<(i,j) -> (i,j)>, // A\\n affine_map<(i,j) -> (i,j)>, // B\\n affine_map<(i,j) -> (i,j)> // X (out)\\n ],\\n //sparse = [\\n // [ "D", "D" ], // A\\n // [ "D", "D" ], // B\\n // [ "D", "D" ] // X\\n //],\\n iterator_types = ["parallel", "parallel"],\\n doc = "X(i,j) = A(i,j) OP B(i,j)"\\n}\\nfunc @add_func(%arga: tensor<32x16xf32>, %argb: tensor<32x16xf32>) -> tensor<32x16xf32> {\\n %0 = linalg.generic #trait_add\\n ins(%arga, %argb: tensor<32x16xf32>, tensor<32x16xf32>)\\n outs(%arga: tensor<32x16xf32>) {\\n ^bb(%a: f32, %b: f32, %s: f32):\\n %0 = addf %a, %b : f32\\n linalg.yield %0 : f32\\n } -> tensor<32x16xf32>\\n return %0 : tensor<32x16xf32>\\n}\\n\\n</textarea>\\n</div>\\n\\n<script>\\nvar config = {\\n lineNumbers: true,\\n tabSize: 2,\\n readOnly: true,\\n mode: 'javascript',\\n};\\nvar myTextArea = document.querySelector('#foobar_single');\\nvar cm = CodeMirror.fromTextArea(myTextArea, config);\\n</script>\\n</body>\\n</html>\\n\"},\"id\":\"1013\",\"type\":\"panel.models.markup.HTML\"}],\"root_ids\":[\"1001\",\"1023\",\"1024\"]},\"title\":\"Bokeh Application\",\"version\":\"2.2.3\"}};\n", | |
| " var render_items = [{\"docid\":\"3edfffc3-603e-4fff-9a0e-b7b33a9cc9e7\",\"root_ids\":[\"1001\"],\"roots\":{\"1001\":\"06327318-09b9-4fc5-977f-e92aba9d7588\"}}];\n", | |
| " root.Bokeh.embed.embed_items_notebook(docs_json, render_items);\n", | |
| " }\n", | |
| " if (root.Bokeh !== undefined && root.Bokeh.Panel !== undefined) {\n", | |
| " embed_document(root);\n", | |
| " } else {\n", | |
| " var attempts = 0;\n", | |
| " var timer = setInterval(function(root) {\n", | |
| " if (root.Bokeh !== undefined && root.Bokeh.Panel !== undefined) {\n", | |
| " clearInterval(timer);\n", | |
| " embed_document(root);\n", | |
| " } else if (document.readyState == \"complete\") {\n", | |
| " attempts++;\n", | |
| " if (attempts > 100) {\n", | |
| " clearInterval(timer);\n", | |
| " console.log(\"Bokeh: ERROR: Unable to run BokehJS code because BokehJS library is missing\");\n", | |
| " }\n", | |
| " }\n", | |
| " }, 10, root)\n", | |
| " }\n", | |
| "})(window);</script>" | |
| ], | |
| "text/plain": [ | |
| "Tabs\n", | |
| " [0] Column\n", | |
| " [0] Select(name='Passes', options=['linalg-bufferize', ...], value='linalg-bufferize')\n", | |
| " [1] Row\n", | |
| " [0] Button(button_type='primary', name='◀', width=200)\n", | |
| " [1] Button(button_type='primary', name='▶', width=200)\n", | |
| " [2] Row(max_width=400)\n", | |
| " [0] HTML(str)\n", | |
| " [1] HTML(str)\n", | |
| " [1] Column\n", | |
| " [0] Select(name='Passes', options=['Initial', 'linalg-buffer...], value='Initial')\n", | |
| " [1] HTML(str)\n", | |
| " [2] Row\n", | |
| " [0] Column\n", | |
| " [0] Select(name='Passes', options=['Initial', 'linalg-buffer...], value='Initial')\n", | |
| " [1] HTML(str)\n", | |
| " [1] Column\n", | |
| " [0] Select(name='Passes', options=['Initial', 'linalg-buffer...], value='linalg-bufferize')\n", | |
| " [1] HTML(str)" | |
| ] | |
| }, | |
| "execution_count": 33, | |
| "metadata": { | |
| "application/vnd.holoviews_exec.v0+json": { | |
| "id": "1001" | |
| } | |
| }, | |
| "output_type": "execute_result" | |
| } | |
| ], | |
| "source": [ | |
| "result.explore()" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 6, | |
| "id": "continued-allocation", | |
| "metadata": {}, | |
| "outputs": [ | |
| { | |
| "name": "stdout", | |
| "output_type": "stream", | |
| "text": [ | |
| "module attributes {llvm.data_layout = \"\"} {\n", | |
| " llvm.func @malloc(i64) -> !llvm.ptr<i8>\n", | |
| " llvm.func @add_func(%arg0: !llvm.ptr<f32>, %arg1: !llvm.ptr<f32>, %arg2: i64, %arg3: i64, %arg4: i64, %arg5: i64, %arg6: i64, %arg7: !llvm.ptr<f32>, %arg8: !llvm.ptr<f32>, %arg9: i64, %arg10: i64, %arg11: i64, %arg12: i64, %arg13: i64) -> !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)> {\n", | |
| " %0 = llvm.mlir.undef : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %1 = llvm.insertvalue %arg0, %0[0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %2 = llvm.insertvalue %arg1, %1[1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %3 = llvm.insertvalue %arg2, %2[2] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %4 = llvm.insertvalue %arg3, %3[3, 0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %5 = llvm.insertvalue %arg5, %4[4, 0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %6 = llvm.insertvalue %arg4, %5[3, 1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %7 = llvm.insertvalue %arg6, %6[4, 1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %8 = llvm.mlir.undef : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %9 = llvm.insertvalue %arg7, %8[0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %10 = llvm.insertvalue %arg8, %9[1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %11 = llvm.insertvalue %arg9, %10[2] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %12 = llvm.insertvalue %arg10, %11[3, 0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %13 = llvm.insertvalue %arg12, %12[4, 0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %14 = llvm.insertvalue %arg11, %13[3, 1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %15 = llvm.insertvalue %arg13, %14[4, 1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %16 = llvm.mlir.constant(32 : index) : i64\n", | |
| " %17 = llvm.mlir.constant(16 : index) : i64\n", | |
| " %18 = llvm.mlir.constant(0 : index) : i64\n", | |
| " %19 = llvm.mlir.constant(1 : index) : i64\n", | |
| " %20 = llvm.mlir.constant(32 : index) : i64\n", | |
| " %21 = llvm.mlir.constant(16 : index) : i64\n", | |
| " %22 = llvm.mlir.constant(1 : index) : i64\n", | |
| " %23 = llvm.mlir.constant(512 : index) : i64\n", | |
| " %24 = llvm.mlir.null : !llvm.ptr<f32>\n", | |
| " %25 = llvm.getelementptr %24[%23] : (!llvm.ptr<f32>, i64) -> !llvm.ptr<f32>\n", | |
| " %26 = llvm.ptrtoint %25 : !llvm.ptr<f32> to i64\n", | |
| " %27 = llvm.call @malloc(%26) : (i64) -> !llvm.ptr<i8>\n", | |
| " %28 = llvm.bitcast %27 : !llvm.ptr<i8> to !llvm.ptr<f32>\n", | |
| " %29 = llvm.mlir.undef : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %30 = llvm.insertvalue %28, %29[0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %31 = llvm.insertvalue %28, %30[1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %32 = llvm.mlir.constant(0 : index) : i64\n", | |
| " %33 = llvm.insertvalue %32, %31[2] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %34 = llvm.insertvalue %20, %33[3, 0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %35 = llvm.insertvalue %21, %34[3, 1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %36 = llvm.insertvalue %21, %35[4, 0] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %37 = llvm.insertvalue %22, %36[4, 1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " llvm.br ^bb1(%18 : i64)\n", | |
| " ^bb1(%38: i64): // 2 preds: ^bb0, ^bb5\n", | |
| " %39 = llvm.icmp \"slt\" %38, %16 : i64\n", | |
| " llvm.cond_br %39, ^bb2, ^bb6\n", | |
| " ^bb2: // pred: ^bb1\n", | |
| " llvm.br ^bb3(%18 : i64)\n", | |
| " ^bb3(%40: i64): // 2 preds: ^bb2, ^bb4\n", | |
| " %41 = llvm.icmp \"slt\" %40, %17 : i64\n", | |
| " llvm.cond_br %41, ^bb4, ^bb5\n", | |
| " ^bb4: // pred: ^bb3\n", | |
| " %42 = llvm.extractvalue %7[1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %43 = llvm.mlir.constant(16 : index) : i64\n", | |
| " %44 = llvm.mul %38, %43 : i64\n", | |
| " %45 = llvm.add %44, %40 : i64\n", | |
| " %46 = llvm.getelementptr %42[%45] : (!llvm.ptr<f32>, i64) -> !llvm.ptr<f32>\n", | |
| " %47 = llvm.load %46 : !llvm.ptr<f32>\n", | |
| " %48 = llvm.extractvalue %15[1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %49 = llvm.mlir.constant(16 : index) : i64\n", | |
| " %50 = llvm.mul %38, %49 : i64\n", | |
| " %51 = llvm.add %50, %40 : i64\n", | |
| " %52 = llvm.getelementptr %48[%51] : (!llvm.ptr<f32>, i64) -> !llvm.ptr<f32>\n", | |
| " %53 = llvm.load %52 : !llvm.ptr<f32>\n", | |
| " %54 = llvm.fadd %47, %53 : f32\n", | |
| " %55 = llvm.extractvalue %37[1] : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " %56 = llvm.mlir.constant(16 : index) : i64\n", | |
| " %57 = llvm.mul %38, %56 : i64\n", | |
| " %58 = llvm.add %57, %40 : i64\n", | |
| " %59 = llvm.getelementptr %55[%58] : (!llvm.ptr<f32>, i64) -> !llvm.ptr<f32>\n", | |
| " llvm.store %54, %59 : !llvm.ptr<f32>\n", | |
| " %60 = llvm.add %40, %19 : i64\n", | |
| " llvm.br ^bb3(%60 : i64)\n", | |
| " ^bb5: // pred: ^bb3\n", | |
| " %61 = llvm.add %38, %19 : i64\n", | |
| " llvm.br ^bb1(%61 : i64)\n", | |
| " ^bb6: // pred: ^bb1\n", | |
| " llvm.return %37 : !llvm.struct<(ptr<f32>, ptr<f32>, i64, array<2 x i64>, array<2 x i64>)>\n", | |
| " }\n", | |
| "}\n", | |
| "\n", | |
| "\n" | |
| ] | |
| } | |
| ], | |
| "source": [ | |
| "print(result.stages[-1])" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 7, | |
| "id": "consecutive-prisoner", | |
| "metadata": {}, | |
| "outputs": [], | |
| "source": [ | |
| "with open('out.mlir', 'w') as f:\n", | |
| " f.write(result.stages[-1])" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 8, | |
| "id": "filled-portal", | |
| "metadata": {}, | |
| "outputs": [], | |
| "source": [ | |
| "! mlir-translate --mlir-to-llvmir out.mlir > out.llvm" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 9, | |
| "id": "verbal-unknown", | |
| "metadata": {}, | |
| "outputs": [], | |
| "source": [ | |
| "import llvmlite.binding as llvm\n", | |
| "\n", | |
| "llvm.initialize()\n", | |
| "llvm.initialize_native_target()\n", | |
| "llvm.initialize_native_asmprinter() # yes, even this one" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 16, | |
| "id": "civil-homework", | |
| "metadata": {}, | |
| "outputs": [], | |
| "source": [ | |
| "with open('out.llvm') as f:\n", | |
| " llvm_ir = f.read()" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 12, | |
| "id": "plastic-irrigation", | |
| "metadata": {}, | |
| "outputs": [], | |
| "source": [ | |
| "def create_execution_engine():\n", | |
| " \"\"\"\n", | |
| " Create an ExecutionEngine suitable for JIT code generation on\n", | |
| " the host CPU. The engine is reusable for an arbitrary number of\n", | |
| " modules.\n", | |
| " \"\"\"\n", | |
| " # Create a target machine representing the host\n", | |
| " target = llvm.Target.from_default_triple()\n", | |
| " target_machine = target.create_target_machine()\n", | |
| " # And an execution engine with an empty backing module\n", | |
| " backing_mod = llvm.parse_assembly(\"\")\n", | |
| " engine = llvm.create_mcjit_compiler(backing_mod, target_machine)\n", | |
| " return engine\n", | |
| "\n", | |
| "def compile_ir(engine, llvm_ir):\n", | |
| " \"\"\"\n", | |
| " Compile the LLVM IR string with the given engine.\n", | |
| " The compiled module object is returned.\n", | |
| " \"\"\"\n", | |
| " # Create a LLVM module object from the IR\n", | |
| " mod = llvm.parse_assembly(llvm_ir)\n", | |
| " mod.verify()\n", | |
| " # Now add the module and make sure it is ready for execution\n", | |
| " engine.add_module(mod)\n", | |
| " engine.finalize_object()\n", | |
| " engine.run_static_constructors()\n", | |
| " return mod" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 13, | |
| "id": "flying-charter", | |
| "metadata": {}, | |
| "outputs": [], | |
| "source": [ | |
| "engine = create_execution_engine()" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 17, | |
| "id": "awful-clearance", | |
| "metadata": {}, | |
| "outputs": [], | |
| "source": [ | |
| "mod = compile_ir(engine, llvm_ir)" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 18, | |
| "id": "activated-template", | |
| "metadata": {}, | |
| "outputs": [], | |
| "source": [ | |
| "func_ptr = engine.get_function_address('add_func')" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 19, | |
| "id": "anonymous-potter", | |
| "metadata": {}, | |
| "outputs": [ | |
| { | |
| "name": "stdout", | |
| "output_type": "stream", | |
| "text": [ | |
| "; ModuleID = 'LLVMDialectModule'\n", | |
| "source_filename = \"LLVMDialectModule\"\n", | |
| "\n", | |
| "declare i8* @malloc(i64)\n", | |
| "\n", | |
| "declare void @free(i8*)\n", | |
| "\n", | |
| "define { float*, float*, i64, [2 x i64], [2 x i64] } @add_func(float* %0, float* %1, i64 %2, i64 %3, i64 %4, i64 %5, i64 %6, float* %7, float* %8, i64 %9, i64 %10, i64 %11, i64 %12, i64 %13) !dbg !3 {\n", | |
| " %15 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } undef, float* %0, 0, !dbg !7\n", | |
| " %16 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } %15, float* %1, 1, !dbg !9\n", | |
| " %17 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } %16, i64 %2, 2, !dbg !10\n", | |
| " %18 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } %17, i64 %3, 3, 0, !dbg !11\n", | |
| " %19 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } %18, i64 %5, 4, 0, !dbg !12\n", | |
| " %20 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } %19, i64 %4, 3, 1, !dbg !13\n", | |
| " %21 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } %20, i64 %6, 4, 1, !dbg !14\n", | |
| " %22 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } undef, float* %7, 0, !dbg !15\n", | |
| " %23 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } %22, float* %8, 1, !dbg !16\n", | |
| " %24 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } %23, i64 %9, 2, !dbg !17\n", | |
| " %25 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } %24, i64 %10, 3, 0, !dbg !18\n", | |
| " %26 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } %25, i64 %12, 4, 0, !dbg !19\n", | |
| " %27 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } %26, i64 %11, 3, 1, !dbg !20\n", | |
| " %28 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } %27, i64 %13, 4, 1, !dbg !21\n", | |
| " %29 = call i8* @malloc(i64 mul (i64 ptrtoint (float* getelementptr (float, float* null, i32 1) to i64), i64 512)), !dbg !22\n", | |
| " %30 = bitcast i8* %29 to float*, !dbg !23\n", | |
| " %31 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } undef, float* %30, 0, !dbg !24\n", | |
| " %32 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } %31, float* %30, 1, !dbg !25\n", | |
| " %33 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } %32, i64 0, 2, !dbg !26\n", | |
| " %34 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } %33, i64 32, 3, 0, !dbg !27\n", | |
| " %35 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } %34, i64 16, 3, 1, !dbg !28\n", | |
| " %36 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } %35, i64 16, 4, 0, !dbg !29\n", | |
| " %37 = insertvalue { float*, float*, i64, [2 x i64], [2 x i64] } %36, i64 1, 4, 1, !dbg !30\n", | |
| " br label %38, !dbg !31\n", | |
| "\n", | |
| "38: ; preds = %62, %14\n", | |
| " %39 = phi i64 [ %63, %62 ], [ 0, %14 ]\n", | |
| " %40 = icmp slt i64 %39, 32, !dbg !32\n", | |
| " br i1 %40, label %41, label %64, !dbg !33\n", | |
| "\n", | |
| "41: ; preds = %38\n", | |
| " br label %42, !dbg !34\n", | |
| "\n", | |
| "42: ; preds = %45, %41\n", | |
| " %43 = phi i64 [ %61, %45 ], [ 0, %41 ]\n", | |
| " %44 = icmp slt i64 %43, 16, !dbg !35\n", | |
| " br i1 %44, label %45, label %62, !dbg !36\n", | |
| "\n", | |
| "45: ; preds = %42\n", | |
| " %46 = extractvalue { float*, float*, i64, [2 x i64], [2 x i64] } %21, 1, !dbg !37\n", | |
| " %47 = mul i64 %39, 16, !dbg !38\n", | |
| " %48 = add i64 %47, %43, !dbg !39\n", | |
| " %49 = getelementptr float, float* %46, i64 %48, !dbg !40\n", | |
| " %50 = load float, float* %49, align 4, !dbg !41\n", | |
| " %51 = extractvalue { float*, float*, i64, [2 x i64], [2 x i64] } %28, 1, !dbg !42\n", | |
| " %52 = mul i64 %39, 16, !dbg !43\n", | |
| " %53 = add i64 %52, %43, !dbg !44\n", | |
| " %54 = getelementptr float, float* %51, i64 %53, !dbg !45\n", | |
| " %55 = load float, float* %54, align 4, !dbg !46\n", | |
| " %56 = fadd float %50, %55, !dbg !47\n", | |
| " %57 = extractvalue { float*, float*, i64, [2 x i64], [2 x i64] } %37, 1, !dbg !48\n", | |
| " %58 = mul i64 %39, 16, !dbg !49\n", | |
| " %59 = add i64 %58, %43, !dbg !50\n", | |
| " %60 = getelementptr float, float* %57, i64 %59, !dbg !51\n", | |
| " store float %56, float* %60, align 4, !dbg !52\n", | |
| " %61 = add i64 %43, 1, !dbg !53\n", | |
| " br label %42, !dbg !54\n", | |
| "\n", | |
| "62: ; preds = %42\n", | |
| " %63 = add i64 %39, 1, !dbg !55\n", | |
| " br label %38, !dbg !56\n", | |
| "\n", | |
| "64: ; preds = %38\n", | |
| " ret { float*, float*, i64, [2 x i64], [2 x i64] } %37, !dbg !57\n", | |
| "}\n", | |
| "\n", | |
| "!llvm.dbg.cu = !{!0}\n", | |
| "!llvm.module.flags = !{!2}\n", | |
| "\n", | |
| "!0 = distinct !DICompileUnit(language: DW_LANG_C, file: !1, producer: \"mlir\", isOptimized: true, runtimeVersion: 0, emissionKind: FullDebug)\n", | |
| "!1 = !DIFile(filename: \"LLVMDialectModule\", directory: \"/\")\n", | |
| "!2 = !{i32 2, !\"Debug Info Version\", i32 3}\n", | |
| "!3 = distinct !DISubprogram(name: \"add_func\", linkageName: \"add_func\", scope: null, file: !4, line: 3, type: !5, scopeLine: 3, spFlags: DISPFlagDefinition | DISPFlagOptimized, unit: !0, retainedNodes: !6)\n", | |
| "!4 = !DIFile(filename: \"out.mlir\", directory: \"/Users/sseibert/continuum/mlir-graphblas\")\n", | |
| "!5 = !DISubroutineType(types: !6)\n", | |
| "!6 = !{}\n", | |
| "!7 = !DILocation(line: 5, column: 10, scope: !8)\n", | |
| "!8 = !DILexicalBlockFile(scope: !3, file: !4, discriminator: 0)\n", | |
| "!9 = !DILocation(line: 6, column: 10, scope: !8)\n", | |
| "!10 = !DILocation(line: 7, column: 10, scope: !8)\n", | |
| "!11 = !DILocation(line: 8, column: 10, scope: !8)\n", | |
| "!12 = !DILocation(line: 9, column: 10, scope: !8)\n", | |
| "!13 = !DILocation(line: 10, column: 10, scope: !8)\n", | |
| "!14 = !DILocation(line: 11, column: 10, scope: !8)\n", | |
| "!15 = !DILocation(line: 13, column: 10, scope: !8)\n", | |
| "!16 = !DILocation(line: 14, column: 11, scope: !8)\n", | |
| "!17 = !DILocation(line: 15, column: 11, scope: !8)\n", | |
| "!18 = !DILocation(line: 16, column: 11, scope: !8)\n", | |
| "!19 = !DILocation(line: 17, column: 11, scope: !8)\n", | |
| "!20 = !DILocation(line: 18, column: 11, scope: !8)\n", | |
| "!21 = !DILocation(line: 19, column: 11, scope: !8)\n", | |
| "!22 = !DILocation(line: 31, column: 11, scope: !8)\n", | |
| "!23 = !DILocation(line: 32, column: 11, scope: !8)\n", | |
| "!24 = !DILocation(line: 34, column: 11, scope: !8)\n", | |
| "!25 = !DILocation(line: 35, column: 11, scope: !8)\n", | |
| "!26 = !DILocation(line: 37, column: 11, scope: !8)\n", | |
| "!27 = !DILocation(line: 38, column: 11, scope: !8)\n", | |
| "!28 = !DILocation(line: 39, column: 11, scope: !8)\n", | |
| "!29 = !DILocation(line: 40, column: 11, scope: !8)\n", | |
| "!30 = !DILocation(line: 41, column: 11, scope: !8)\n", | |
| "!31 = !DILocation(line: 42, column: 5, scope: !8)\n", | |
| "!32 = !DILocation(line: 44, column: 11, scope: !8)\n", | |
| "!33 = !DILocation(line: 45, column: 5, scope: !8)\n", | |
| "!34 = !DILocation(line: 47, column: 5, scope: !8)\n", | |
| "!35 = !DILocation(line: 49, column: 11, scope: !8)\n", | |
| "!36 = !DILocation(line: 50, column: 5, scope: !8)\n", | |
| "!37 = !DILocation(line: 52, column: 11, scope: !8)\n", | |
| "!38 = !DILocation(line: 54, column: 11, scope: !8)\n", | |
| "!39 = !DILocation(line: 55, column: 11, scope: !8)\n", | |
| "!40 = !DILocation(line: 56, column: 11, scope: !8)\n", | |
| "!41 = !DILocation(line: 57, column: 11, scope: !8)\n", | |
| "!42 = !DILocation(line: 58, column: 11, scope: !8)\n", | |
| "!43 = !DILocation(line: 60, column: 11, scope: !8)\n", | |
| "!44 = !DILocation(line: 61, column: 11, scope: !8)\n", | |
| "!45 = !DILocation(line: 62, column: 11, scope: !8)\n", | |
| "!46 = !DILocation(line: 63, column: 11, scope: !8)\n", | |
| "!47 = !DILocation(line: 64, column: 11, scope: !8)\n", | |
| "!48 = !DILocation(line: 65, column: 11, scope: !8)\n", | |
| "!49 = !DILocation(line: 67, column: 11, scope: !8)\n", | |
| "!50 = !DILocation(line: 68, column: 11, scope: !8)\n", | |
| "!51 = !DILocation(line: 69, column: 11, scope: !8)\n", | |
| "!52 = !DILocation(line: 70, column: 5, scope: !8)\n", | |
| "!53 = !DILocation(line: 71, column: 11, scope: !8)\n", | |
| "!54 = !DILocation(line: 72, column: 5, scope: !8)\n", | |
| "!55 = !DILocation(line: 74, column: 11, scope: !8)\n", | |
| "!56 = !DILocation(line: 75, column: 5, scope: !8)\n", | |
| "!57 = !DILocation(line: 77, column: 5, scope: !8)\n", | |
| "\n" | |
| ] | |
| } | |
| ], | |
| "source": [ | |
| "print(llvm_ir)" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 26, | |
| "id": "alpha-scott", | |
| "metadata": {}, | |
| "outputs": [], | |
| "source": [ | |
| "import ctypes\n", | |
| "from ctypes import CFUNCTYPE, c_float, c_int64, Structure, POINTER\n", | |
| "import numpy as np" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 73, | |
| "id": "conservative-uniform", | |
| "metadata": {}, | |
| "outputs": [], | |
| "source": [ | |
| "c_float_p = np.ctypeslib.ndpointer(np.float32)\n", | |
| "class ReturnType(Structure):\n", | |
| " _fields_ = [\n", | |
| " ('alloc', POINTER(c_float)),\n", | |
| " ('base', POINTER(c_float)),\n", | |
| " ('offset', c_int64),\n", | |
| " ('size0', c_int64),\n", | |
| " ('size1', c_int64),\n", | |
| " ('stride0', c_int64),\n", | |
| " ('stride1', c_int64),\n", | |
| " ]\n", | |
| "\n", | |
| "cfunc = CFUNCTYPE(\n", | |
| " ReturnType,\n", | |
| " # see https://mlir.llvm.org/docs/ConversionToLLVMDialect/#default-convention for how ranked memrefs are passed\n", | |
| " # arg0\n", | |
| " c_float_p, # allocated pointer (for free())\n", | |
| " c_float_p, # base pointer\n", | |
| " c_int64, # offset from base\n", | |
| " c_int64, # dim0 size (elements)\n", | |
| " c_int64, # dim1 size (elements)\n", | |
| " c_int64, # dim0 stride (elements)\n", | |
| " c_int64, # dim1 stride (elements)\n", | |
| " # arg1\n", | |
| " c_float_p, # same as above\n", | |
| " c_float_p,\n", | |
| " c_int64,\n", | |
| " c_int64,\n", | |
| " c_int64,\n", | |
| " c_int64,\n", | |
| " c_int64,\n", | |
| ")(func_ptr)" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 88, | |
| "id": "ongoing-craps", | |
| "metadata": {}, | |
| "outputs": [], | |
| "source": [ | |
| "a = np.arange(32*16, dtype=np.float32).reshape((32,16))\n", | |
| "b = a * 2" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 89, | |
| "id": "fifty-regard", | |
| "metadata": {}, | |
| "outputs": [], | |
| "source": [ | |
| "ret = cfunc(\n", | |
| " a,\n", | |
| " a,\n", | |
| " 0,\n", | |
| " a.shape[0],\n", | |
| " a.shape[1],\n", | |
| " a.strides[0] // a.itemsize,\n", | |
| " a.strides[1] // a.itemsize,\n", | |
| "\n", | |
| " b,\n", | |
| " b,\n", | |
| " 0,\n", | |
| " b.shape[0],\n", | |
| " b.shape[1],\n", | |
| " b.strides[0] // b.itemsize,\n", | |
| " b.strides[1] // b.itemsize,\n", | |
| ")" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 119, | |
| "id": "continent-layout", | |
| "metadata": {}, | |
| "outputs": [ | |
| { | |
| "name": "stdout", | |
| "output_type": "stream", | |
| "text": [ | |
| "alloc : <__main__.LP_c_float object at 0x7fec59e93b00>\n", | |
| "base : <__main__.LP_c_float object at 0x7fec59e93b00>\n", | |
| "offset : 0\n", | |
| "size0 : 32\n", | |
| "size1 : 16\n", | |
| "stride0 : 16\n", | |
| "stride1 : 1\n" | |
| ] | |
| } | |
| ], | |
| "source": [ | |
| "for attr in ['alloc', 'base', 'offset', 'size0', 'size1', 'stride0', 'stride1']:\n", | |
| " print(attr, ':', getattr(ret, attr))" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 129, | |
| "id": "identical-chicago", | |
| "metadata": {}, | |
| "outputs": [], | |
| "source": [ | |
| "out = np.frombuffer(\n", | |
| " (ctypes.c_float * (32 * 16)).from_address(\n", | |
| " ctypes.addressof(ret.base.contents)\n", | |
| " ),\n", | |
| " dtype=np.float32\n", | |
| ").reshape((32, 16))" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 130, | |
| "id": "acquired-entertainment", | |
| "metadata": {}, | |
| "outputs": [ | |
| { | |
| "data": { | |
| "text/plain": [ | |
| "array([[ 0., 3., 6., 9., 12., 15., 18., 21., 24.,\n", | |
| " 27., 30., 33., 36., 39., 42., 45.],\n", | |
| " [ 48., 51., 54., 57., 60., 63., 66., 69., 72.,\n", | |
| " 75., 78., 81., 84., 87., 90., 93.],\n", | |
| " [ 96., 99., 102., 105., 108., 111., 114., 117., 120.,\n", | |
| " 123., 126., 129., 132., 135., 138., 141.],\n", | |
| " [ 144., 147., 150., 153., 156., 159., 162., 165., 168.,\n", | |
| " 171., 174., 177., 180., 183., 186., 189.],\n", | |
| " [ 192., 195., 198., 201., 204., 207., 210., 213., 216.,\n", | |
| " 219., 222., 225., 228., 231., 234., 237.],\n", | |
| " [ 240., 243., 246., 249., 252., 255., 258., 261., 264.,\n", | |
| " 267., 270., 273., 276., 279., 282., 285.],\n", | |
| " [ 288., 291., 294., 297., 300., 303., 306., 309., 312.,\n", | |
| " 315., 318., 321., 324., 327., 330., 333.],\n", | |
| " [ 336., 339., 342., 345., 348., 351., 354., 357., 360.,\n", | |
| " 363., 366., 369., 372., 375., 378., 381.],\n", | |
| " [ 384., 387., 390., 393., 396., 399., 402., 405., 408.,\n", | |
| " 411., 414., 417., 420., 423., 426., 429.],\n", | |
| " [ 432., 435., 438., 441., 444., 447., 450., 453., 456.,\n", | |
| " 459., 462., 465., 468., 471., 474., 477.],\n", | |
| " [ 480., 483., 486., 489., 492., 495., 498., 501., 504.,\n", | |
| " 507., 510., 513., 516., 519., 522., 525.],\n", | |
| " [ 528., 531., 534., 537., 540., 543., 546., 549., 552.,\n", | |
| " 555., 558., 561., 564., 567., 570., 573.],\n", | |
| " [ 576., 579., 582., 585., 588., 591., 594., 597., 600.,\n", | |
| " 603., 606., 609., 612., 615., 618., 621.],\n", | |
| " [ 624., 627., 630., 633., 636., 639., 642., 645., 648.,\n", | |
| " 651., 654., 657., 660., 663., 666., 669.],\n", | |
| " [ 672., 675., 678., 681., 684., 687., 690., 693., 696.,\n", | |
| " 699., 702., 705., 708., 711., 714., 717.],\n", | |
| " [ 720., 723., 726., 729., 732., 735., 738., 741., 744.,\n", | |
| " 747., 750., 753., 756., 759., 762., 765.],\n", | |
| " [ 768., 771., 774., 777., 780., 783., 786., 789., 792.,\n", | |
| " 795., 798., 801., 804., 807., 810., 813.],\n", | |
| " [ 816., 819., 822., 825., 828., 831., 834., 837., 840.,\n", | |
| " 843., 846., 849., 852., 855., 858., 861.],\n", | |
| " [ 864., 867., 870., 873., 876., 879., 882., 885., 888.,\n", | |
| " 891., 894., 897., 900., 903., 906., 909.],\n", | |
| " [ 912., 915., 918., 921., 924., 927., 930., 933., 936.,\n", | |
| " 939., 942., 945., 948., 951., 954., 957.],\n", | |
| " [ 960., 963., 966., 969., 972., 975., 978., 981., 984.,\n", | |
| " 987., 990., 993., 996., 999., 1002., 1005.],\n", | |
| " [1008., 1011., 1014., 1017., 1020., 1023., 1026., 1029., 1032.,\n", | |
| " 1035., 1038., 1041., 1044., 1047., 1050., 1053.],\n", | |
| " [1056., 1059., 1062., 1065., 1068., 1071., 1074., 1077., 1080.,\n", | |
| " 1083., 1086., 1089., 1092., 1095., 1098., 1101.],\n", | |
| " [1104., 1107., 1110., 1113., 1116., 1119., 1122., 1125., 1128.,\n", | |
| " 1131., 1134., 1137., 1140., 1143., 1146., 1149.],\n", | |
| " [1152., 1155., 1158., 1161., 1164., 1167., 1170., 1173., 1176.,\n", | |
| " 1179., 1182., 1185., 1188., 1191., 1194., 1197.],\n", | |
| " [1200., 1203., 1206., 1209., 1212., 1215., 1218., 1221., 1224.,\n", | |
| " 1227., 1230., 1233., 1236., 1239., 1242., 1245.],\n", | |
| " [1248., 1251., 1254., 1257., 1260., 1263., 1266., 1269., 1272.,\n", | |
| " 1275., 1278., 1281., 1284., 1287., 1290., 1293.],\n", | |
| " [1296., 1299., 1302., 1305., 1308., 1311., 1314., 1317., 1320.,\n", | |
| " 1323., 1326., 1329., 1332., 1335., 1338., 1341.],\n", | |
| " [1344., 1347., 1350., 1353., 1356., 1359., 1362., 1365., 1368.,\n", | |
| " 1371., 1374., 1377., 1380., 1383., 1386., 1389.],\n", | |
| " [1392., 1395., 1398., 1401., 1404., 1407., 1410., 1413., 1416.,\n", | |
| " 1419., 1422., 1425., 1428., 1431., 1434., 1437.],\n", | |
| " [1440., 1443., 1446., 1449., 1452., 1455., 1458., 1461., 1464.,\n", | |
| " 1467., 1470., 1473., 1476., 1479., 1482., 1485.],\n", | |
| " [1488., 1491., 1494., 1497., 1500., 1503., 1506., 1509., 1512.,\n", | |
| " 1515., 1518., 1521., 1524., 1527., 1530., 1533.]], dtype=float32)" | |
| ] | |
| }, | |
| "execution_count": 130, | |
| "metadata": {}, | |
| "output_type": "execute_result" | |
| } | |
| ], | |
| "source": [ | |
| "out" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": 131, | |
| "id": "residential-shock", | |
| "metadata": {}, | |
| "outputs": [], | |
| "source": [ | |
| "np.testing.assert_array_equal(out, a + b)" | |
| ] | |
| }, | |
| { | |
| "cell_type": "code", | |
| "execution_count": null, | |
| "id": "appropriate-floating", | |
| "metadata": {}, | |
| "outputs": [], | |
| "source": [] | |
| } | |
| ], | |
| "metadata": { | |
| "kernelspec": { | |
| "display_name": "Python 3", | |
| "language": "python", | |
| "name": "python3" | |
| }, | |
| "language_info": { | |
| "codemirror_mode": { | |
| "name": "ipython", | |
| "version": 3 | |
| }, | |
| "file_extension": ".py", | |
| "mimetype": "text/x-python", | |
| "name": "python", | |
| "nbconvert_exporter": "python", | |
| "pygments_lexer": "ipython3", | |
| "version": "3.7.9" | |
| } | |
| }, | |
| "nbformat": 4, | |
| "nbformat_minor": 5 | |
| } |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment