tutorials/codegen.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# Customizing Code Generation\n",
    "\n",
    "Writing custom backends in DaCe is a powerful feature that allows users to specify their own backends. This could range from customizing different support libraries, through new target architectures and use of specialized hardware, to outputting a completely different language than C.\n",
    "\n",
    "This is made possible due to the modular enumerations and code generation architecture in DaCe. Any enumeration (e.g., for storage types and schedulers) can be extended through the API, in order to enable custom, project-specific behavior, and code generators (similarly to transformations) have a registry that can also be extended at runtime. \n",
    "\n",
    "Code generation generally follows a recursive procedure, starting from the top-level SDFG:\n",
    "  * Generate frame-code and entry point for SDFG\n",
    "  * Generate code for array allocation/deallocation based on lifetime (SDFG-wide and persistent lifetime)\n",
    "  * Generate control flow between states\n",
    "  * Generate each state/scope:\n",
    "    * Generate additional code for array allocation/deallocation based on lifetime (state- or scope-wide)\n",
    "    * Generate data movement code for each memlet path\n",
    "    * Generate each node in a topological, hierarchical order (descending into scopes such as Maps)\n",
    "      * If a node is a scope entry, generate code for the scope as a subgraph\n",
    "      * If a node is a nested SDFG, generate code for that SDFG as a separate function and call it\n",
    "  \n",
    "In this tutorial, we will show how to customize the code generation procedure by creating a new map scheduler and generating custom code for it. First, we will import dace and some classes to make the rest of the code cleaner:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<link href=\"https://fonts.googleapis.com/icon?family=Material+Icons\" rel=\"stylesheet\">\n",
       "        <script>\n",
       "        require.config({\n",
       "            paths: {\n",
       "                \"math\": \"https://spcl.github.io/dace/webclient/external_lib/math.min\"\n",
       "            },\n",
       "            waitSeconds: 40\n",
       "          });\n",
       "        require( [\"math\"], x => window.math = x);\n",
       "        </script><script src=\"https://spcl.github.io/dace/webclient/renderer_dir/dagre.js\"></script>\n",
       "<script src=\"https://spcl.github.io/dace/webclient/renderer_dir/global_vars.js\"></script>\n",
       "<script src=\"https://spcl.github.io/dace/webclient/context_menu.js\"></script>\n",
       "<script src=\"https://spcl.github.io/dace/webclient/renderer_elements.js\"></script>\n",
       "<script src=\"https://spcl.github.io/dace/webclient/sdfg_utils.js\"></script>\n",
       "<script src=\"https://spcl.github.io/dace/webclient/renderer.js\"></script>\n",
       "<link href=\"https://spcl.github.io/dace/webclient/sdfv.css\" rel=\"stylesheet\">\n"
      ],
      "text/plain": [
       "<IPython.core.display.HTML object>"
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "import dace\n",
    "from dace import registry\n",
    "from dace.sdfg.scope import ScopeSubgraphView\n",
    "from dace.codegen.prettycode import CodeIOStream\n",
    "from dace.codegen.targets.target import TargetCodeGenerator\n",
    "from dace.codegen.targets.framecode import DaCeCodeGenerator\n",
    "from dace.codegen.targets.cpp import sym2cpp"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Next, we can define some simple program to work with:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "\n",
       "<div id=\"contents_9125830616792462010\" style=\"position: relative; resize: vertical; overflow: auto\"></div>\n",
       "<script>\n",
       "    var sdfg_9125830616792462010 = \"{\\n  \\\"type\\\": \\\"SDFG\\\",\\n  \\\"attributes\\\": {\\n    \\\"arg_types\\\": [],\\n    \\\"constants_prop\\\": {},\\n    \\\"_arrays\\\": {\\n      \\\"A\\\": {\\n        \\\"type\\\": \\\"Array\\\",\\n        \\\"attributes\\\": {\\n          \\\"allow_conflicts\\\": false,\\n          \\\"strides\\\": [\\n            \\\"30\\\",\\n            \\\"1\\\"\\n          ],\\n          \\\"total_size\\\": \\\"600\\\",\\n          \\\"offset\\\": [\\n            \\\"0\\\",\\n            \\\"0\\\"\\n          ],\\n          \\\"may_alias\\\": false,\\n          \\\"alignment\\\": 0,\\n          \\\"dtype\\\": \\\"float64\\\",\\n          \\\"shape\\\": [\\n            \\\"20\\\",\\n            \\\"30\\\"\\n          ],\\n          \\\"transient\\\": false,\\n          \\\"storage\\\": \\\"Default\\\",\\n          \\\"lifetime\\\": \\\"Scope\\\",\\n          \\\"location\\\": {},\\n          \\\"debuginfo\\\": null\\n        }\\n      }\\n    },\\n    \\\"symbols\\\": {},\\n    \\\"global_code\\\": {\\n      \\\"frame\\\": {\\n        \\\"string_data\\\": \\\"\\\",\\n        \\\"language\\\": \\\"CPP\\\"\\n      }\\n    },\\n    \\\"init_code\\\": {\\n      \\\"frame\\\": {\\n        \\\"string_data\\\": \\\"\\\",\\n        \\\"language\\\": \\\"CPP\\\"\\n      }\\n    },\\n    \\\"exit_code\\\": {\\n      \\\"frame\\\": {\\n        \\\"string_data\\\": \\\"\\\",\\n        \\\"language\\\": \\\"CPP\\\"\\n      }\\n    },\\n    \\\"orig_sdfg\\\": null,\\n    \\\"transformation_hist\\\": [],\\n    \\\"name\\\": \\\"simple\\\"\\n  },\\n  \\\"nodes\\\": [\\n    {\\n      \\\"type\\\": \\\"SDFGState\\\",\\n      \\\"label\\\": \\\"MapState\\\",\\n      \\\"id\\\": 0,\\n      \\\"collapsed\\\": false,\\n      \\\"scope_dict\\\": {\\n        \\\"-1\\\": [\\n          0,\\n          2,\\n          3\\n        ],\\n        \\\"0\\\": [\\n          1,\\n          4\\n        ]\\n      },\\n      \\\"nodes\\\": [\\n        {\\n          \\\"type\\\": \\\"MapEntry\\\",\\n          \\\"label\\\": \\\"simple_3[i=0:20:2, j=0:30]\\\",\\n          \\\"attributes\\\": {\\n            \\\"label\\\": \\\"simple_3\\\",\\n            \\\"params\\\": [\\n              \\\"i\\\",\\n              \\\"j\\\"\\n            ],\\n            \\\"range\\\": {\\n              \\\"type\\\": \\\"Range\\\",\\n              \\\"ranges\\\": [\\n                {\\n                  \\\"start\\\": \\\"0\\\",\\n                  \\\"end\\\": \\\"19\\\",\\n                  \\\"step\\\": {\\n                    \\\"main\\\": \\\"2\\\",\\n                    \\\"approx\\\": \\\"2\\\"\\n                  },\\n                  \\\"tile\\\": \\\"1\\\"\\n                },\\n                {\\n                  \\\"start\\\": \\\"0\\\",\\n                  \\\"end\\\": \\\"29\\\",\\n                  \\\"step\\\": {\\n                    \\\"main\\\": \\\"1\\\",\\n                    \\\"approx\\\": \\\"1\\\"\\n                  },\\n                  \\\"tile\\\": \\\"1\\\"\\n                }\\n              ]\\n            },\\n            \\\"schedule\\\": \\\"Default\\\",\\n            \\\"unroll\\\": false,\\n            \\\"collapse\\\": 1,\\n            \\\"debuginfo\\\": {\\n              \\\"type\\\": \\\"DebugInfo\\\",\\n              \\\"start_line\\\": 3,\\n              \\\"end_line\\\": 3,\\n              \\\"start_column\\\": 4,\\n              \\\"end_column\\\": 4,\\n              \\\"filename\\\": \\\"<ipython-input-2-66eabfc1015a>\\\"\\n            },\\n            \\\"is_collapsed\\\": false,\\n            \\\"instrument\\\": \\\"No_Instrumentation\\\",\\n            \\\"in_connectors\\\": {\\n              \\\"IN_1\\\": null\\n            },\\n            \\\"out_connectors\\\": {\\n              \\\"OUT_1\\\": null\\n            }\\n          },\\n          \\\"id\\\": 0,\\n          \\\"scope_entry\\\": null,\\n          \\\"scope_exit\\\": \\\"1\\\"\\n        },\\n        {\\n          \\\"type\\\": \\\"MapExit\\\",\\n          \\\"label\\\": \\\"simple_3[i=0:20:2, j=0:30]\\\",\\n          \\\"attributes\\\": {\\n            \\\"in_connectors\\\": {\\n              \\\"IN_1\\\": null\\n            },\\n            \\\"out_connectors\\\": {\\n              \\\"OUT_1\\\": null\\n            }\\n          },\\n          \\\"id\\\": 1,\\n          \\\"scope_entry\\\": \\\"0\\\",\\n          \\\"scope_exit\\\": \\\"1\\\"\\n        },\\n        {\\n          \\\"type\\\": \\\"AccessNode\\\",\\n          \\\"label\\\": \\\"A\\\",\\n          \\\"attributes\\\": {\\n            \\\"access\\\": \\\"ReadOnly\\\",\\n            \\\"setzero\\\": false,\\n            \\\"debuginfo\\\": {\\n              \\\"type\\\": \\\"DebugInfo\\\",\\n              \\\"start_line\\\": 3,\\n              \\\"end_line\\\": 3,\\n              \\\"start_column\\\": 4,\\n              \\\"end_column\\\": 4,\\n              \\\"filename\\\": \\\"<ipython-input-2-66eabfc1015a>\\\"\\n            },\\n            \\\"data\\\": \\\"A\\\",\\n            \\\"in_connectors\\\": {},\\n            \\\"out_connectors\\\": {}\\n          },\\n          \\\"id\\\": 2,\\n          \\\"scope_entry\\\": null,\\n          \\\"scope_exit\\\": null\\n        },\\n        {\\n          \\\"type\\\": \\\"AccessNode\\\",\\n          \\\"label\\\": \\\"A\\\",\\n          \\\"attributes\\\": {\\n            \\\"access\\\": \\\"WriteOnly\\\",\\n            \\\"setzero\\\": false,\\n            \\\"debuginfo\\\": {\\n              \\\"type\\\": \\\"DebugInfo\\\",\\n              \\\"start_line\\\": 3,\\n              \\\"end_line\\\": 3,\\n              \\\"start_column\\\": 4,\\n              \\\"end_column\\\": 4,\\n              \\\"filename\\\": \\\"<ipython-input-2-66eabfc1015a>\\\"\\n            },\\n            \\\"data\\\": \\\"A\\\",\\n            \\\"in_connectors\\\": {},\\n            \\\"out_connectors\\\": {}\\n          },\\n          \\\"id\\\": 3,\\n          \\\"scope_entry\\\": null,\\n          \\\"scope_exit\\\": null\\n        },\\n        {\\n          \\\"type\\\": \\\"Tasklet\\\",\\n          \\\"label\\\": \\\"augassign_4_8\\\",\\n          \\\"attributes\\\": {\\n            \\\"code\\\": {\\n              \\\"string_data\\\": \\\"__out = (__in1 + __in2)\\\",\\n              \\\"language\\\": \\\"Python\\\"\\n            },\\n            \\\"debuginfo\\\": {\\n              \\\"type\\\": \\\"DebugInfo\\\",\\n              \\\"start_line\\\": 4,\\n              \\\"end_line\\\": 4,\\n              \\\"start_column\\\": 19,\\n              \\\"end_column\\\": 19,\\n              \\\"filename\\\": \\\"<ipython-input-2-66eabfc1015a>\\\"\\n            },\\n            \\\"instrument\\\": \\\"No_Instrumentation\\\",\\n            \\\"label\\\": \\\"augassign_4_8\\\",\\n            \\\"location\\\": {},\\n            \\\"environments\\\": [],\\n            \\\"in_connectors\\\": {\\n              \\\"__in2\\\": null,\\n              \\\"__in1\\\": null\\n            },\\n            \\\"out_connectors\\\": {\\n              \\\"__out\\\": null\\n            }\\n          },\\n          \\\"id\\\": 4,\\n          \\\"scope_entry\\\": \\\"0\\\",\\n          \\\"scope_exit\\\": \\\"1\\\"\\n        }\\n      ],\\n      \\\"edges\\\": [\\n        {\\n          \\\"type\\\": \\\"MultiConnectorEdge\\\",\\n          \\\"attributes\\\": {\\n            \\\"data\\\": {\\n              \\\"type\\\": \\\"Memlet\\\",\\n              \\\"attributes\\\": {\\n                \\\"volume\\\": \\\"300\\\",\\n                \\\"dynamic\\\": false,\\n                \\\"subset\\\": {\\n                  \\\"type\\\": \\\"Range\\\",\\n                  \\\"ranges\\\": [\\n                    {\\n                      \\\"start\\\": \\\"0\\\",\\n                      \\\"end\\\": \\\"2*int_floor(19, 2)\\\",\\n                      \\\"step\\\": \\\"1\\\",\\n                      \\\"tile\\\": \\\"1\\\"\\n                    },\\n                    {\\n                      \\\"start\\\": \\\"0\\\",\\n                      \\\"end\\\": \\\"29\\\",\\n                      \\\"step\\\": \\\"1\\\",\\n                      \\\"tile\\\": \\\"1\\\"\\n                    }\\n                  ]\\n                },\\n                \\\"other_subset\\\": null,\\n                \\\"data\\\": \\\"A\\\",\\n                \\\"wcr\\\": null,\\n                \\\"debuginfo\\\": null,\\n                \\\"wcr_nonatomic\\\": false,\\n                \\\"allow_oob\\\": false,\\n                \\\"src_subset\\\": {\\n                  \\\"type\\\": \\\"Range\\\",\\n                  \\\"ranges\\\": [\\n                    {\\n                      \\\"start\\\": \\\"0\\\",\\n                      \\\"end\\\": \\\"2*int_floor(19, 2)\\\",\\n                      \\\"step\\\": \\\"1\\\",\\n                      \\\"tile\\\": \\\"1\\\"\\n                    },\\n                    {\\n                      \\\"start\\\": \\\"0\\\",\\n                      \\\"end\\\": \\\"29\\\",\\n                      \\\"step\\\": \\\"1\\\",\\n                      \\\"tile\\\": \\\"1\\\"\\n                    }\\n                  ]\\n                },\\n                \\\"dst_subset\\\": null,\\n                \\\"num_accesses\\\": \\\"300\\\"\\n              }\\n            }\\n          },\\n          \\\"src\\\": \\\"2\\\",\\n          \\\"dst\\\": \\\"0\\\",\\n          \\\"dst_connector\\\": \\\"IN_1\\\",\\n          \\\"src_connector\\\": null\\n        },\\n        {\\n          \\\"type\\\": \\\"MultiConnectorEdge\\\",\\n          \\\"attributes\\\": {\\n            \\\"data\\\": {\\n              \\\"type\\\": \\\"Memlet\\\",\\n              \\\"attributes\\\": {\\n                \\\"volume\\\": \\\"300\\\",\\n                \\\"dynamic\\\": false,\\n                \\\"subset\\\": {\\n                  \\\"type\\\": \\\"Range\\\",\\n                  \\\"ranges\\\": [\\n                    {\\n                      \\\"start\\\": \\\"0\\\",\\n                      \\\"end\\\": \\\"2*int_floor(19, 2)\\\",\\n                      \\\"step\\\": \\\"1\\\",\\n                      \\\"tile\\\": \\\"1\\\"\\n                    },\\n                    {\\n                      \\\"start\\\": \\\"0\\\",\\n                      \\\"end\\\": \\\"29\\\",\\n                      \\\"step\\\": \\\"1\\\",\\n                      \\\"tile\\\": \\\"1\\\"\\n                    }\\n                  ]\\n                },\\n                \\\"other_subset\\\": null,\\n                \\\"data\\\": \\\"A\\\",\\n                \\\"wcr\\\": null,\\n                \\\"debuginfo\\\": null,\\n                \\\"wcr_nonatomic\\\": false,\\n                \\\"allow_oob\\\": false,\\n                \\\"src_subset\\\": null,\\n                \\\"dst_subset\\\": {\\n                  \\\"type\\\": \\\"Range\\\",\\n                  \\\"ranges\\\": [\\n                    {\\n                      \\\"start\\\": \\\"0\\\",\\n                      \\\"end\\\": \\\"2*int_floor(19, 2)\\\",\\n                      \\\"step\\\": \\\"1\\\",\\n                      \\\"tile\\\": \\\"1\\\"\\n                    },\\n                    {\\n                      \\\"start\\\": \\\"0\\\",\\n                      \\\"end\\\": \\\"29\\\",\\n                      \\\"step\\\": \\\"1\\\",\\n                      \\\"tile\\\": \\\"1\\\"\\n                    }\\n                  ]\\n                },\\n                \\\"num_accesses\\\": \\\"300\\\"\\n              }\\n            }\\n          },\\n          \\\"src\\\": \\\"1\\\",\\n          \\\"dst\\\": \\\"3\\\",\\n          \\\"dst_connector\\\": null,\\n          \\\"src_connector\\\": \\\"OUT_1\\\"\\n        },\\n        {\\n          \\\"type\\\": \\\"MultiConnectorEdge\\\",\\n          \\\"attributes\\\": {\\n            \\\"data\\\": {\\n              \\\"type\\\": \\\"Memlet\\\",\\n              \\\"attributes\\\": {\\n                \\\"volume\\\": \\\"1\\\",\\n                \\\"dynamic\\\": false,\\n                \\\"subset\\\": {\\n                  \\\"type\\\": \\\"Range\\\",\\n                  \\\"ranges\\\": [\\n                    {\\n                      \\\"start\\\": \\\"i\\\",\\n                      \\\"end\\\": \\\"i\\\",\\n                      \\\"step\\\": \\\"1\\\",\\n                      \\\"tile\\\": \\\"1\\\"\\n                    },\\n                    {\\n                      \\\"start\\\": \\\"j\\\",\\n                      \\\"end\\\": \\\"j\\\",\\n                      \\\"step\\\": \\\"1\\\",\\n                      \\\"tile\\\": \\\"1\\\"\\n                    }\\n                  ]\\n                },\\n                \\\"other_subset\\\": null,\\n                \\\"data\\\": \\\"A\\\",\\n                \\\"wcr\\\": null,\\n                \\\"debuginfo\\\": null,\\n                \\\"wcr_nonatomic\\\": false,\\n                \\\"allow_oob\\\": false,\\n                \\\"src_subset\\\": {\\n                  \\\"type\\\": \\\"Range\\\",\\n                  \\\"ranges\\\": [\\n                    {\\n                      \\\"start\\\": \\\"i\\\",\\n                      \\\"end\\\": \\\"i\\\",\\n                      \\\"step\\\": \\\"1\\\",\\n                      \\\"tile\\\": \\\"1\\\"\\n                    },\\n                    {\\n                      \\\"start\\\": \\\"j\\\",\\n                      \\\"end\\\": \\\"j\\\",\\n                      \\\"step\\\": \\\"1\\\",\\n                      \\\"tile\\\": \\\"1\\\"\\n                    }\\n                  ]\\n                },\\n                \\\"dst_subset\\\": null,\\n                \\\"num_accesses\\\": \\\"1\\\"\\n              }\\n            }\\n          },\\n          \\\"src\\\": \\\"0\\\",\\n          \\\"dst\\\": \\\"4\\\",\\n          \\\"dst_connector\\\": \\\"__in1\\\",\\n          \\\"src_connector\\\": \\\"OUT_1\\\"\\n        },\\n        {\\n          \\\"type\\\": \\\"MultiConnectorEdge\\\",\\n          \\\"attributes\\\": {\\n            \\\"data\\\": {\\n              \\\"type\\\": \\\"Memlet\\\",\\n              \\\"attributes\\\": {\\n                \\\"volume\\\": \\\"1\\\",\\n                \\\"dynamic\\\": false,\\n                \\\"subset\\\": {\\n                  \\\"type\\\": \\\"Range\\\",\\n                  \\\"ranges\\\": [\\n                    {\\n                      \\\"start\\\": \\\"i\\\",\\n                      \\\"end\\\": \\\"i\\\",\\n                      \\\"step\\\": \\\"1\\\",\\n                      \\\"tile\\\": \\\"1\\\"\\n                    },\\n                    {\\n                      \\\"start\\\": \\\"j\\\",\\n                      \\\"end\\\": \\\"j\\\",\\n                      \\\"step\\\": \\\"1\\\",\\n                      \\\"tile\\\": \\\"1\\\"\\n                    }\\n                  ]\\n                },\\n                \\\"other_subset\\\": null,\\n                \\\"data\\\": \\\"A\\\",\\n                \\\"wcr\\\": null,\\n                \\\"debuginfo\\\": null,\\n                \\\"wcr_nonatomic\\\": false,\\n                \\\"allow_oob\\\": false,\\n                \\\"src_subset\\\": {\\n                  \\\"type\\\": \\\"Range\\\",\\n                  \\\"ranges\\\": [\\n                    {\\n                      \\\"start\\\": \\\"i\\\",\\n                      \\\"end\\\": \\\"i\\\",\\n                      \\\"step\\\": \\\"1\\\",\\n                      \\\"tile\\\": \\\"1\\\"\\n                    },\\n                    {\\n                      \\\"start\\\": \\\"j\\\",\\n                      \\\"end\\\": \\\"j\\\",\\n                      \\\"step\\\": \\\"1\\\",\\n                      \\\"tile\\\": \\\"1\\\"\\n                    }\\n                  ]\\n                },\\n                \\\"dst_subset\\\": null,\\n                \\\"num_accesses\\\": \\\"1\\\"\\n              }\\n            }\\n          },\\n          \\\"src\\\": \\\"0\\\",\\n          \\\"dst\\\": \\\"4\\\",\\n          \\\"dst_connector\\\": \\\"__in2\\\",\\n          \\\"src_connector\\\": \\\"OUT_1\\\"\\n        },\\n        {\\n          \\\"type\\\": \\\"MultiConnectorEdge\\\",\\n          \\\"attributes\\\": {\\n            \\\"data\\\": {\\n              \\\"type\\\": \\\"Memlet\\\",\\n              \\\"attributes\\\": {\\n                \\\"volume\\\": \\\"1\\\",\\n                \\\"dynamic\\\": false,\\n                \\\"subset\\\": {\\n                  \\\"type\\\": \\\"Range\\\",\\n                  \\\"ranges\\\": [\\n                    {\\n                      \\\"start\\\": \\\"i\\\",\\n                      \\\"end\\\": \\\"i\\\",\\n                      \\\"step\\\": \\\"1\\\",\\n                      \\\"tile\\\": \\\"1\\\"\\n                    },\\n                    {\\n                      \\\"start\\\": \\\"j\\\",\\n                      \\\"end\\\": \\\"j\\\",\\n                      \\\"step\\\": \\\"1\\\",\\n                      \\\"tile\\\": \\\"1\\\"\\n                    }\\n                  ]\\n                },\\n                \\\"other_subset\\\": null,\\n                \\\"data\\\": \\\"A\\\",\\n                \\\"wcr\\\": null,\\n                \\\"debuginfo\\\": null,\\n                \\\"wcr_nonatomic\\\": false,\\n                \\\"allow_oob\\\": false,\\n                \\\"src_subset\\\": null,\\n                \\\"dst_subset\\\": {\\n                  \\\"type\\\": \\\"Range\\\",\\n                  \\\"ranges\\\": [\\n                    {\\n                      \\\"start\\\": \\\"i\\\",\\n                      \\\"end\\\": \\\"i\\\",\\n                      \\\"step\\\": \\\"1\\\",\\n                      \\\"tile\\\": \\\"1\\\"\\n                    },\\n                    {\\n                      \\\"start\\\": \\\"j\\\",\\n                      \\\"end\\\": \\\"j\\\",\\n                      \\\"step\\\": \\\"1\\\",\\n                      \\\"tile\\\": \\\"1\\\"\\n                    }\\n                  ]\\n                },\\n                \\\"num_accesses\\\": \\\"1\\\"\\n              }\\n            }\\n          },\\n          \\\"src\\\": \\\"4\\\",\\n          \\\"dst\\\": \\\"1\\\",\\n          \\\"dst_connector\\\": \\\"IN_1\\\",\\n          \\\"src_connector\\\": \\\"__out\\\"\\n        }\\n      ],\\n      \\\"attributes\\\": {\\n        \\\"is_collapsed\\\": false,\\n        \\\"nosync\\\": false,\\n        \\\"instrument\\\": \\\"No_Instrumentation\\\",\\n        \\\"location\\\": {}\\n      }\\n    }\\n  ],\\n  \\\"edges\\\": [],\\n  \\\"sdfg_list_id\\\": 0\\n}\";\n",
       "    var renderer_9125830616792462010 = new SDFGRenderer(parse_sdfg(sdfg_9125830616792462010),\n",
       "        document.getElementById('contents_9125830616792462010'));\n",
       "</script>"
      ],
      "text/plain": [
       "<dace.sdfg.sdfg.SDFG at 0x7f9b0882d0d0>"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "@dace.program\n",
    "def simple(A: dace.float64[20, 30]):\n",
    "    for i, j in dace.map[0:20:2, 0:30]:\n",
    "        A[i, j] += A[i, j]\n",
    "\n",
    "# Preview SDFG\n",
    "sdfg = simple.to_sdfg()\n",
    "sdfg"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "If we observe the generated code, by default our map would be scheduled to an OpenMP multi-core loop:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<style>.output_html .hll { background-color: #ffffcc }\n",
       ".output_html  { background: #f8f8f8; }\n",
       ".output_html .c { color: #408080; font-style: italic } /* Comment */\n",
       ".output_html .err { border: 1px solid #FF0000 } /* Error */\n",
       ".output_html .k { color: #008000; font-weight: bold } /* Keyword */\n",
       ".output_html .o { color: #666666 } /* Operator */\n",
       ".output_html .ch { color: #408080; font-style: italic } /* Comment.Hashbang */\n",
       ".output_html .cm { color: #408080; font-style: italic } /* Comment.Multiline */\n",
       ".output_html .cp { color: #BC7A00 } /* Comment.Preproc */\n",
       ".output_html .cpf { color: #408080; font-style: italic } /* Comment.PreprocFile */\n",
       ".output_html .c1 { color: #408080; font-style: italic } /* Comment.Single */\n",
       ".output_html .cs { color: #408080; font-style: italic } /* Comment.Special */\n",
       ".output_html .gd { color: #A00000 } /* Generic.Deleted */\n",
       ".output_html .ge { font-style: italic } /* Generic.Emph */\n",
       ".output_html .gr { color: #FF0000 } /* Generic.Error */\n",
       ".output_html .gh { color: #000080; font-weight: bold } /* Generic.Heading */\n",
       ".output_html .gi { color: #00A000 } /* Generic.Inserted */\n",
       ".output_html .go { color: #888888 } /* Generic.Output */\n",
       ".output_html .gp { color: #000080; font-weight: bold } /* Generic.Prompt */\n",
       ".output_html .gs { font-weight: bold } /* Generic.Strong */\n",
       ".output_html .gu { color: #800080; font-weight: bold } /* Generic.Subheading */\n",
       ".output_html .gt { color: #0044DD } /* Generic.Traceback */\n",
       ".output_html .kc { color: #008000; font-weight: bold } /* Keyword.Constant */\n",
       ".output_html .kd { color: #008000; font-weight: bold } /* Keyword.Declaration */\n",
       ".output_html .kn { color: #008000; font-weight: bold } /* Keyword.Namespace */\n",
       ".output_html .kp { color: #008000 } /* Keyword.Pseudo */\n",
       ".output_html .kr { color: #008000; font-weight: bold } /* Keyword.Reserved */\n",
       ".output_html .kt { color: #B00040 } /* Keyword.Type */\n",
       ".output_html .m { color: #666666 } /* Literal.Number */\n",
       ".output_html .s { color: #BA2121 } /* Literal.String */\n",
       ".output_html .na { color: #7D9029 } /* Name.Attribute */\n",
       ".output_html .nb { color: #008000 } /* Name.Builtin */\n",
       ".output_html .nc { color: #0000FF; font-weight: bold } /* Name.Class */\n",
       ".output_html .no { color: #880000 } /* Name.Constant */\n",
       ".output_html .nd { color: #AA22FF } /* Name.Decorator */\n",
       ".output_html .ni { color: #999999; font-weight: bold } /* Name.Entity */\n",
       ".output_html .ne { color: #D2413A; font-weight: bold } /* Name.Exception */\n",
       ".output_html .nf { color: #0000FF } /* Name.Function */\n",
       ".output_html .nl { color: #A0A000 } /* Name.Label */\n",
       ".output_html .nn { color: #0000FF; font-weight: bold } /* Name.Namespace */\n",
       ".output_html .nt { color: #008000; font-weight: bold } /* Name.Tag */\n",
       ".output_html .nv { color: #19177C } /* Name.Variable */\n",
       ".output_html .ow { color: #AA22FF; font-weight: bold } /* Operator.Word */\n",
       ".output_html .w { color: #bbbbbb } /* Text.Whitespace */\n",
       ".output_html .mb { color: #666666 } /* Literal.Number.Bin */\n",
       ".output_html .mf { color: #666666 } /* Literal.Number.Float */\n",
       ".output_html .mh { color: #666666 } /* Literal.Number.Hex */\n",
       ".output_html .mi { color: #666666 } /* Literal.Number.Integer */\n",
       ".output_html .mo { color: #666666 } /* Literal.Number.Oct */\n",
       ".output_html .sa { color: #BA2121 } /* Literal.String.Affix */\n",
       ".output_html .sb { color: #BA2121 } /* Literal.String.Backtick */\n",
       ".output_html .sc { color: #BA2121 } /* Literal.String.Char */\n",
       ".output_html .dl { color: #BA2121 } /* Literal.String.Delimiter */\n",
       ".output_html .sd { color: #BA2121; font-style: italic } /* Literal.String.Doc */\n",
       ".output_html .s2 { color: #BA2121 } /* Literal.String.Double */\n",
       ".output_html .se { color: #BB6622; font-weight: bold } /* Literal.String.Escape */\n",
       ".output_html .sh { color: #BA2121 } /* Literal.String.Heredoc */\n",
       ".output_html .si { color: #BB6688; font-weight: bold } /* Literal.String.Interpol */\n",
       ".output_html .sx { color: #008000 } /* Literal.String.Other */\n",
       ".output_html .sr { color: #BB6688 } /* Literal.String.Regex */\n",
       ".output_html .s1 { color: #BA2121 } /* Literal.String.Single */\n",
       ".output_html .ss { color: #19177C } /* Literal.String.Symbol */\n",
       ".output_html .bp { color: #008000 } /* Name.Builtin.Pseudo */\n",
       ".output_html .fm { color: #0000FF } /* Name.Function.Magic */\n",
       ".output_html .vc { color: #19177C } /* Name.Variable.Class */\n",
       ".output_html .vg { color: #19177C } /* Name.Variable.Global */\n",
       ".output_html .vi { color: #19177C } /* Name.Variable.Instance */\n",
       ".output_html .vm { color: #19177C } /* Name.Variable.Magic */\n",
       ".output_html .il { color: #666666 } /* Literal.Number.Integer.Long */</style><div class=\"highlight\"><pre><span></span><span class=\"cm\">/* DaCe AUTO-GENERATED FILE. DO NOT MODIFY */</span>\n",
       "<span class=\"cp\">#include</span> <span class=\"cpf\">&lt;dace/dace.h&gt;</span><span class=\"cp\"></span>\n",
       "\n",
       "<span class=\"kt\">void</span> <span class=\"nf\">__program_simple_internal</span><span class=\"p\">(</span><span class=\"kt\">double</span> <span class=\"o\">*</span> <span class=\"n\">__restrict__</span> <span class=\"n\">A</span><span class=\"p\">)</span>\n",
       "<span class=\"p\">{</span>\n",
       "\n",
       "    <span class=\"p\">{</span>\n",
       "        \n",
       "        \n",
       "        <span class=\"p\">{</span>\n",
       "            <span class=\"cp\">#pragma omp parallel for</span>\n",
       "            <span class=\"k\">for</span> <span class=\"p\">(</span><span class=\"k\">auto</span> <span class=\"n\">i</span> <span class=\"o\">=</span> <span class=\"mi\">0</span><span class=\"p\">;</span> <span class=\"n\">i</span> <span class=\"o\">&lt;</span> <span class=\"mi\">20</span><span class=\"p\">;</span> <span class=\"n\">i</span> <span class=\"o\">+=</span> <span class=\"mi\">2</span><span class=\"p\">)</span> <span class=\"p\">{</span>\n",
       "                <span class=\"k\">for</span> <span class=\"p\">(</span><span class=\"k\">auto</span> <span class=\"n\">j</span> <span class=\"o\">=</span> <span class=\"mi\">0</span><span class=\"p\">;</span> <span class=\"n\">j</span> <span class=\"o\">&lt;</span> <span class=\"mi\">30</span><span class=\"p\">;</span> <span class=\"n\">j</span> <span class=\"o\">+=</span> <span class=\"mi\">1</span><span class=\"p\">)</span> <span class=\"p\">{</span>\n",
       "                    <span class=\"p\">{</span>\n",
       "                        <span class=\"kt\">double</span> <span class=\"n\">__in1</span> <span class=\"o\">=</span> <span class=\"n\">A</span><span class=\"p\">[((</span><span class=\"mi\">30</span> <span class=\"o\">*</span> <span class=\"n\">i</span><span class=\"p\">)</span> <span class=\"o\">+</span> <span class=\"n\">j</span><span class=\"p\">)];</span>\n",
       "                        <span class=\"kt\">double</span> <span class=\"n\">__in2</span> <span class=\"o\">=</span> <span class=\"n\">A</span><span class=\"p\">[((</span><span class=\"mi\">30</span> <span class=\"o\">*</span> <span class=\"n\">i</span><span class=\"p\">)</span> <span class=\"o\">+</span> <span class=\"n\">j</span><span class=\"p\">)];</span>\n",
       "                        <span class=\"kt\">double</span> <span class=\"n\">__out</span><span class=\"p\">;</span>\n",
       "\n",
       "                        <span class=\"c1\">///////////////////</span>\n",
       "                        <span class=\"c1\">// Tasklet code (augassign_4_8)</span>\n",
       "                        <span class=\"n\">__out</span> <span class=\"o\">=</span> <span class=\"p\">(</span><span class=\"n\">__in1</span> <span class=\"o\">+</span> <span class=\"n\">__in2</span><span class=\"p\">);</span>\n",
       "                        <span class=\"c1\">///////////////////</span>\n",
       "\n",
       "                        <span class=\"n\">A</span><span class=\"p\">[((</span><span class=\"mi\">30</span> <span class=\"o\">*</span> <span class=\"n\">i</span><span class=\"p\">)</span> <span class=\"o\">+</span> <span class=\"n\">j</span><span class=\"p\">)]</span> <span class=\"o\">=</span> <span class=\"n\">__out</span><span class=\"p\">;</span>\n",
       "                    <span class=\"p\">}</span>\n",
       "                <span class=\"p\">}</span>\n",
       "            <span class=\"p\">}</span>\n",
       "        <span class=\"p\">}</span>\n",
       "    <span class=\"p\">}</span>\n",
       "<span class=\"p\">}</span>\n",
       "\n",
       "<span class=\"n\">DACE_EXPORTED</span> <span class=\"kt\">void</span> <span class=\"nf\">__program_simple</span><span class=\"p\">(</span><span class=\"kt\">double</span> <span class=\"o\">*</span> <span class=\"n\">__restrict__</span> <span class=\"n\">A</span><span class=\"p\">)</span>\n",
       "<span class=\"p\">{</span>\n",
       "    <span class=\"n\">__program_simple_internal</span><span class=\"p\">(</span><span class=\"n\">A</span><span class=\"p\">);</span>\n",
       "<span class=\"p\">}</span>\n",
       "\n",
       "<span class=\"n\">DACE_EXPORTED</span> <span class=\"kt\">int</span> <span class=\"nf\">__dace_init_simple</span><span class=\"p\">(</span><span class=\"kt\">double</span> <span class=\"o\">*</span> <span class=\"n\">__restrict__</span> <span class=\"n\">A</span><span class=\"p\">)</span>\n",
       "<span class=\"p\">{</span>\n",
       "    <span class=\"kt\">int</span> <span class=\"n\">__result</span> <span class=\"o\">=</span> <span class=\"mi\">0</span><span class=\"p\">;</span>\n",
       "\n",
       "    <span class=\"k\">return</span> <span class=\"n\">__result</span><span class=\"p\">;</span>\n",
       "<span class=\"p\">}</span>\n",
       "\n",
       "<span class=\"n\">DACE_EXPORTED</span> <span class=\"kt\">void</span> <span class=\"nf\">__dace_exit_simple</span><span class=\"p\">(</span><span class=\"kt\">double</span> <span class=\"o\">*</span> <span class=\"n\">__restrict__</span> <span class=\"n\">A</span><span class=\"p\">)</span>\n",
       "<span class=\"p\">{</span>\n",
       "<span class=\"p\">}</span>\n",
       "</pre></div>\n"
      ],
      "text/latex": [
       "\\begin{Verbatim}[commandchars=\\\\\\{\\}]\n",
       "\\PY{c+cm}{/* DaCe AUTO\\PYZhy{}GENERATED FILE. DO NOT MODIFY */}\n",
       "\\PY{c+cp}{\\PYZsh{}}\\PY{c+cp}{include} \\PY{c+cpf}{\\PYZlt{}dace/dace.h\\PYZgt{}}\n",
       "\n",
       "\\PY{k+kt}{void} \\PY{n+nf}{\\PYZus{}\\PYZus{}program\\PYZus{}simple\\PYZus{}internal}\\PY{p}{(}\\PY{k+kt}{double} \\PY{o}{*} \\PY{n}{\\PYZus{}\\PYZus{}restrict\\PYZus{}\\PYZus{}} \\PY{n}{A}\\PY{p}{)}\n",
       "\\PY{p}{\\PYZob{}}\n",
       "\n",
       "    \\PY{p}{\\PYZob{}}\n",
       "        \n",
       "        \n",
       "        \\PY{p}{\\PYZob{}}\n",
       "            \\PY{c+cp}{\\PYZsh{}}\\PY{c+cp}{pragma omp parallel for}\n",
       "            \\PY{k}{for} \\PY{p}{(}\\PY{k}{auto} \\PY{n}{i} \\PY{o}{=} \\PY{l+m+mi}{0}\\PY{p}{;} \\PY{n}{i} \\PY{o}{\\PYZlt{}} \\PY{l+m+mi}{20}\\PY{p}{;} \\PY{n}{i} \\PY{o}{+}\\PY{o}{=} \\PY{l+m+mi}{2}\\PY{p}{)} \\PY{p}{\\PYZob{}}\n",
       "                \\PY{k}{for} \\PY{p}{(}\\PY{k}{auto} \\PY{n}{j} \\PY{o}{=} \\PY{l+m+mi}{0}\\PY{p}{;} \\PY{n}{j} \\PY{o}{\\PYZlt{}} \\PY{l+m+mi}{30}\\PY{p}{;} \\PY{n}{j} \\PY{o}{+}\\PY{o}{=} \\PY{l+m+mi}{1}\\PY{p}{)} \\PY{p}{\\PYZob{}}\n",
       "                    \\PY{p}{\\PYZob{}}\n",
       "                        \\PY{k+kt}{double} \\PY{n}{\\PYZus{}\\PYZus{}in1} \\PY{o}{=} \\PY{n}{A}\\PY{p}{[}\\PY{p}{(}\\PY{p}{(}\\PY{l+m+mi}{30} \\PY{o}{*} \\PY{n}{i}\\PY{p}{)} \\PY{o}{+} \\PY{n}{j}\\PY{p}{)}\\PY{p}{]}\\PY{p}{;}\n",
       "                        \\PY{k+kt}{double} \\PY{n}{\\PYZus{}\\PYZus{}in2} \\PY{o}{=} \\PY{n}{A}\\PY{p}{[}\\PY{p}{(}\\PY{p}{(}\\PY{l+m+mi}{30} \\PY{o}{*} \\PY{n}{i}\\PY{p}{)} \\PY{o}{+} \\PY{n}{j}\\PY{p}{)}\\PY{p}{]}\\PY{p}{;}\n",
       "                        \\PY{k+kt}{double} \\PY{n}{\\PYZus{}\\PYZus{}out}\\PY{p}{;}\n",
       "\n",
       "                        \\PY{c+c1}{///////////////////}\n",
       "                        \\PY{c+c1}{// Tasklet code (augassign\\PYZus{}4\\PYZus{}8)}\n",
       "                        \\PY{n}{\\PYZus{}\\PYZus{}out} \\PY{o}{=} \\PY{p}{(}\\PY{n}{\\PYZus{}\\PYZus{}in1} \\PY{o}{+} \\PY{n}{\\PYZus{}\\PYZus{}in2}\\PY{p}{)}\\PY{p}{;}\n",
       "                        \\PY{c+c1}{///////////////////}\n",
       "\n",
       "                        \\PY{n}{A}\\PY{p}{[}\\PY{p}{(}\\PY{p}{(}\\PY{l+m+mi}{30} \\PY{o}{*} \\PY{n}{i}\\PY{p}{)} \\PY{o}{+} \\PY{n}{j}\\PY{p}{)}\\PY{p}{]} \\PY{o}{=} \\PY{n}{\\PYZus{}\\PYZus{}out}\\PY{p}{;}\n",
       "                    \\PY{p}{\\PYZcb{}}\n",
       "                \\PY{p}{\\PYZcb{}}\n",
       "            \\PY{p}{\\PYZcb{}}\n",
       "        \\PY{p}{\\PYZcb{}}\n",
       "    \\PY{p}{\\PYZcb{}}\n",
       "\\PY{p}{\\PYZcb{}}\n",
       "\n",
       "\\PY{n}{DACE\\PYZus{}EXPORTED} \\PY{k+kt}{void} \\PY{n+nf}{\\PYZus{}\\PYZus{}program\\PYZus{}simple}\\PY{p}{(}\\PY{k+kt}{double} \\PY{o}{*} \\PY{n}{\\PYZus{}\\PYZus{}restrict\\PYZus{}\\PYZus{}} \\PY{n}{A}\\PY{p}{)}\n",
       "\\PY{p}{\\PYZob{}}\n",
       "    \\PY{n}{\\PYZus{}\\PYZus{}program\\PYZus{}simple\\PYZus{}internal}\\PY{p}{(}\\PY{n}{A}\\PY{p}{)}\\PY{p}{;}\n",
       "\\PY{p}{\\PYZcb{}}\n",
       "\n",
       "\\PY{n}{DACE\\PYZus{}EXPORTED} \\PY{k+kt}{int} \\PY{n+nf}{\\PYZus{}\\PYZus{}dace\\PYZus{}init\\PYZus{}simple}\\PY{p}{(}\\PY{k+kt}{double} \\PY{o}{*} \\PY{n}{\\PYZus{}\\PYZus{}restrict\\PYZus{}\\PYZus{}} \\PY{n}{A}\\PY{p}{)}\n",
       "\\PY{p}{\\PYZob{}}\n",
       "    \\PY{k+kt}{int} \\PY{n}{\\PYZus{}\\PYZus{}result} \\PY{o}{=} \\PY{l+m+mi}{0}\\PY{p}{;}\n",
       "\n",
       "    \\PY{k}{return} \\PY{n}{\\PYZus{}\\PYZus{}result}\\PY{p}{;}\n",
       "\\PY{p}{\\PYZcb{}}\n",
       "\n",
       "\\PY{n}{DACE\\PYZus{}EXPORTED} \\PY{k+kt}{void} \\PY{n+nf}{\\PYZus{}\\PYZus{}dace\\PYZus{}exit\\PYZus{}simple}\\PY{p}{(}\\PY{k+kt}{double} \\PY{o}{*} \\PY{n}{\\PYZus{}\\PYZus{}restrict\\PYZus{}\\PYZus{}} \\PY{n}{A}\\PY{p}{)}\n",
       "\\PY{p}{\\PYZob{}}\n",
       "\\PY{p}{\\PYZcb{}}\n",
       "\\end{Verbatim}\n"
      ],
      "text/plain": [
       "/* DaCe AUTO-GENERATED FILE. DO NOT MODIFY */\n",
       "#include <dace/dace.h>\n",
       "\n",
       "void __program_simple_internal(double * __restrict__ A)\n",
       "{\n",
       "\n",
       "    {\n",
       "        \n",
       "        \n",
       "        {\n",
       "            #pragma omp parallel for\n",
       "            for (auto i = 0; i < 20; i += 2) {\n",
       "                for (auto j = 0; j < 30; j += 1) {\n",
       "                    {\n",
       "                        double __in1 = A[((30 * i) + j)];\n",
       "                        double __in2 = A[((30 * i) + j)];\n",
       "                        double __out;\n",
       "\n",
       "                        ///////////////////\n",
       "                        // Tasklet code (augassign_4_8)\n",
       "                        __out = (__in1 + __in2);\n",
       "                        ///////////////////\n",
       "\n",
       "                        A[((30 * i) + j)] = __out;\n",
       "                    }\n",
       "                }\n",
       "            }\n",
       "        }\n",
       "    }\n",
       "}\n",
       "\n",
       "DACE_EXPORTED void __program_simple(double * __restrict__ A)\n",
       "{\n",
       "    __program_simple_internal(A);\n",
       "}\n",
       "\n",
       "DACE_EXPORTED int __dace_init_simple(double * __restrict__ A)\n",
       "{\n",
       "    int __result = 0;\n",
       "\n",
       "    return __result;\n",
       "}\n",
       "\n",
       "DACE_EXPORTED void __dace_exit_simple(double * __restrict__ A)\n",
       "{\n",
       "}\n"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "from IPython.display import Code\n",
    "Code(sdfg.generate_code()[0].clean_code, language='cpp')"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Custom Scope Generator\n",
    "\n",
    "To begin, we need to add our own enum entries for our new map schedule. Here are the current schedule types in our extensible enumeration:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[<ScheduleType.Default: 1>,\n",
       " <ScheduleType.Sequential: 2>,\n",
       " <ScheduleType.MPI: 3>,\n",
       " <ScheduleType.CPU_Multicore: 4>,\n",
       " <ScheduleType.GPU_Device: 5>,\n",
       " <ScheduleType.GPU_ThreadBlock: 6>,\n",
       " <ScheduleType.GPU_ThreadBlock_Dynamic: 7>,\n",
       " <ScheduleType.GPU_Persistent: 8>,\n",
       " <ScheduleType.FPGA_Device: 9>]"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "list(dace.ScheduleType)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Registering a new value is just a matter of calling `register`:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "[<ScheduleType.Default: 1>,\n",
       " <ScheduleType.Sequential: 2>,\n",
       " <ScheduleType.MPI: 3>,\n",
       " <ScheduleType.CPU_Multicore: 4>,\n",
       " <ScheduleType.GPU_Device: 5>,\n",
       " <ScheduleType.GPU_ThreadBlock: 6>,\n",
       " <ScheduleType.GPU_ThreadBlock_Dynamic: 7>,\n",
       " <ScheduleType.GPU_Persistent: 8>,\n",
       " <ScheduleType.FPGA_Device: 9>,\n",
       " <ScheduleType.LoopyLoop: 10>]"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "dace.ScheduleType.register('LoopyLoop')\n",
    "list(dace.ScheduleType)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "It can also be used directly from now on:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "<ScheduleType.LoopyLoop: 10>"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "dace.ScheduleType.LoopyLoop"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "One additional step for code generation is to tell the code generator which arrays and scopes lie inside by default:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [],
   "source": [
    "dace.SCOPEDEFAULT_SCHEDULE[dace.ScheduleType.LoopyLoop] = dace.ScheduleType.Sequential\n",
    "dace.SCOPEDEFAULT_STORAGE[dace.ScheduleType.LoopyLoop] = dace.StorageType.CPU_Heap"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Now we can register and create a matching code generator:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [],
   "source": [
    "@registry.autoregister_params(name='loopy')\n",
    "class MyCustomLoop(TargetCodeGenerator):\n",
    "    def __init__(self, frame_codegen: DaCeCodeGenerator, sdfg: dace.SDFG):\n",
    "        ################################################################\n",
    "        # Define some locals:\n",
    "        # Can be used to call back to the frame-code generator\n",
    "        self.frame = frame_codegen\n",
    "        # Can be used to dispatch other code generators for allocation/nodes\n",
    "        self.dispatcher = frame_codegen.dispatcher\n",
    "        \n",
    "        ################################################################\n",
    "        # Register handlers/hooks through dispatcher: Can be used for \n",
    "        # nodes, memory copy/allocation, scopes, states, and more.\n",
    "        \n",
    "        # In this case, register scopes\n",
    "        self.dispatcher.register_map_dispatcher(dace.ScheduleType.LoopyLoop, self)\n",
    "        \n",
    "        # You can similarly use register_{array,copy,node,state}_dispatcher\n",
    "        \n",
    "    # A scope dispatcher will trigger a method called generate_scope whenever \n",
    "    # an SDFG has a scope with that schedule\n",
    "    def generate_scope(self, sdfg: dace.SDFG, scope: ScopeSubgraphView,\n",
    "                       state_id: int, function_stream: CodeIOStream,\n",
    "                       callsite_stream: CodeIOStream):\n",
    "        # The parameters here are:\n",
    "        # sdfg: The SDFG we are currently generating.\n",
    "        # scope: The subgraph of the state containing only the scope (map contents)\n",
    "        #        we want to generate the code for.\n",
    "        # state_id: The state in the SDFG the subgraph is taken from (i.e., \n",
    "        #           `sdfg.node(state_id)` is the same as `scope.graph`)\n",
    "        # function_stream: A cursor to the global code (which can be used to define\n",
    "        #                  functions, hence the name).\n",
    "        # callsite_stream: A cursor to the current location in the code, most of\n",
    "        #                  the code is generated here.\n",
    "        \n",
    "        # We can get the map entry node from the scope graph\n",
    "        entry_node = scope.source_nodes()[0]\n",
    "        \n",
    "        # First, generate an opening brace (for instrumentation and dynamic map ranges)\n",
    "        callsite_stream.write('{', sdfg, state_id, entry_node)\n",
    "        \n",
    "        ################################################################\n",
    "        # Generate specific code: We will generate a reversed loop with a \n",
    "        # comment for each dimension of the map. For the sake of simplicity,\n",
    "        # dynamic map ranges are not supported.\n",
    "        \n",
    "        for param, rng in zip(entry_node.map.params, entry_node.map.range):\n",
    "            # We use the sym2cpp function from the cpp support functions\n",
    "            # to convert symbolic expressions to proper C++\n",
    "            begin, end, stride = (sym2cpp(r) for r in rng)\n",
    "            \n",
    "            # Every write is optionally (but recommended to be) tagged with\n",
    "            # 1-3 extra arguments, serving as line information to match\n",
    "            # SDFG, state, and graph nodes/edges to written code.\n",
    "            callsite_stream.write(f'''// Loopy-loop {param}\n",
    "            for (auto {param} = {end}; {param} >= {begin}; {param} -= {stride}) {{''',\n",
    "                                  sdfg, state_id, entry_node\n",
    "            )\n",
    "        \n",
    "            # NOTE: CodeIOStream will automatically take care of indentation for us.\n",
    "        \n",
    "        \n",
    "        # Now that the loops have been defined, use the dispatcher to invoke any\n",
    "        # code generator (including this one) that is registered to deal with\n",
    "        # the internal nodes in the subgraph. We skip the MapEntry node.\n",
    "        self.dispatcher.dispatch_subgraph(sdfg, scope, state_id,\n",
    "                                          function_stream, callsite_stream,\n",
    "                                          skip_entry_node=True)\n",
    "        \n",
    "        # NOTE: Since skip_exit_node above is set to False, closing braces will\n",
    "        #       be automatically generated"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "After the code generator has been registered, all that's left is to change the map schedule and generate new code:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<style>.output_html .hll { background-color: #ffffcc }\n",
       ".output_html  { background: #f8f8f8; }\n",
       ".output_html .c { color: #408080; font-style: italic } /* Comment */\n",
       ".output_html .err { border: 1px solid #FF0000 } /* Error */\n",
       ".output_html .k { color: #008000; font-weight: bold } /* Keyword */\n",
       ".output_html .o { color: #666666 } /* Operator */\n",
       ".output_html .ch { color: #408080; font-style: italic } /* Comment.Hashbang */\n",
       ".output_html .cm { color: #408080; font-style: italic } /* Comment.Multiline */\n",
       ".output_html .cp { color: #BC7A00 } /* Comment.Preproc */\n",
       ".output_html .cpf { color: #408080; font-style: italic } /* Comment.PreprocFile */\n",
       ".output_html .c1 { color: #408080; font-style: italic } /* Comment.Single */\n",
       ".output_html .cs { color: #408080; font-style: italic } /* Comment.Special */\n",
       ".output_html .gd { color: #A00000 } /* Generic.Deleted */\n",
       ".output_html .ge { font-style: italic } /* Generic.Emph */\n",
       ".output_html .gr { color: #FF0000 } /* Generic.Error */\n",
       ".output_html .gh { color: #000080; font-weight: bold } /* Generic.Heading */\n",
       ".output_html .gi { color: #00A000 } /* Generic.Inserted */\n",
       ".output_html .go { color: #888888 } /* Generic.Output */\n",
       ".output_html .gp { color: #000080; font-weight: bold } /* Generic.Prompt */\n",
       ".output_html .gs { font-weight: bold } /* Generic.Strong */\n",
       ".output_html .gu { color: #800080; font-weight: bold } /* Generic.Subheading */\n",
       ".output_html .gt { color: #0044DD } /* Generic.Traceback */\n",
       ".output_html .kc { color: #008000; font-weight: bold } /* Keyword.Constant */\n",
       ".output_html .kd { color: #008000; font-weight: bold } /* Keyword.Declaration */\n",
       ".output_html .kn { color: #008000; font-weight: bold } /* Keyword.Namespace */\n",
       ".output_html .kp { color: #008000 } /* Keyword.Pseudo */\n",
       ".output_html .kr { color: #008000; font-weight: bold } /* Keyword.Reserved */\n",
       ".output_html .kt { color: #B00040 } /* Keyword.Type */\n",
       ".output_html .m { color: #666666 } /* Literal.Number */\n",
       ".output_html .s { color: #BA2121 } /* Literal.String */\n",
       ".output_html .na { color: #7D9029 } /* Name.Attribute */\n",
       ".output_html .nb { color: #008000 } /* Name.Builtin */\n",
       ".output_html .nc { color: #0000FF; font-weight: bold } /* Name.Class */\n",
       ".output_html .no { color: #880000 } /* Name.Constant */\n",
       ".output_html .nd { color: #AA22FF } /* Name.Decorator */\n",
       ".output_html .ni { color: #999999; font-weight: bold } /* Name.Entity */\n",
       ".output_html .ne { color: #D2413A; font-weight: bold } /* Name.Exception */\n",
       ".output_html .nf { color: #0000FF } /* Name.Function */\n",
       ".output_html .nl { color: #A0A000 } /* Name.Label */\n",
       ".output_html .nn { color: #0000FF; font-weight: bold } /* Name.Namespace */\n",
       ".output_html .nt { color: #008000; font-weight: bold } /* Name.Tag */\n",
       ".output_html .nv { color: #19177C } /* Name.Variable */\n",
       ".output_html .ow { color: #AA22FF; font-weight: bold } /* Operator.Word */\n",
       ".output_html .w { color: #bbbbbb } /* Text.Whitespace */\n",
       ".output_html .mb { color: #666666 } /* Literal.Number.Bin */\n",
       ".output_html .mf { color: #666666 } /* Literal.Number.Float */\n",
       ".output_html .mh { color: #666666 } /* Literal.Number.Hex */\n",
       ".output_html .mi { color: #666666 } /* Literal.Number.Integer */\n",
       ".output_html .mo { color: #666666 } /* Literal.Number.Oct */\n",
       ".output_html .sa { color: #BA2121 } /* Literal.String.Affix */\n",
       ".output_html .sb { color: #BA2121 } /* Literal.String.Backtick */\n",
       ".output_html .sc { color: #BA2121 } /* Literal.String.Char */\n",
       ".output_html .dl { color: #BA2121 } /* Literal.String.Delimiter */\n",
       ".output_html .sd { color: #BA2121; font-style: italic } /* Literal.String.Doc */\n",
       ".output_html .s2 { color: #BA2121 } /* Literal.String.Double */\n",
       ".output_html .se { color: #BB6622; font-weight: bold } /* Literal.String.Escape */\n",
       ".output_html .sh { color: #BA2121 } /* Literal.String.Heredoc */\n",
       ".output_html .si { color: #BB6688; font-weight: bold } /* Literal.String.Interpol */\n",
       ".output_html .sx { color: #008000 } /* Literal.String.Other */\n",
       ".output_html .sr { color: #BB6688 } /* Literal.String.Regex */\n",
       ".output_html .s1 { color: #BA2121 } /* Literal.String.Single */\n",
       ".output_html .ss { color: #19177C } /* Literal.String.Symbol */\n",
       ".output_html .bp { color: #008000 } /* Name.Builtin.Pseudo */\n",
       ".output_html .fm { color: #0000FF } /* Name.Function.Magic */\n",
       ".output_html .vc { color: #19177C } /* Name.Variable.Class */\n",
       ".output_html .vg { color: #19177C } /* Name.Variable.Global */\n",
       ".output_html .vi { color: #19177C } /* Name.Variable.Instance */\n",
       ".output_html .vm { color: #19177C } /* Name.Variable.Magic */\n",
       ".output_html .il { color: #666666 } /* Literal.Number.Integer.Long */</style><div class=\"highlight\"><pre><span></span><span class=\"cm\">/* DaCe AUTO-GENERATED FILE. DO NOT MODIFY */</span>\n",
       "<span class=\"cp\">#include</span> <span class=\"cpf\">&lt;dace/dace.h&gt;</span><span class=\"cp\"></span>\n",
       "\n",
       "<span class=\"kt\">void</span> <span class=\"nf\">__program_simple_internal</span><span class=\"p\">(</span><span class=\"kt\">double</span> <span class=\"o\">*</span> <span class=\"n\">__restrict__</span> <span class=\"n\">A</span><span class=\"p\">)</span>\n",
       "<span class=\"p\">{</span>\n",
       "\n",
       "    <span class=\"p\">{</span>\n",
       "        \n",
       "        \n",
       "        <span class=\"c1\">// Loopy-loop i</span>\n",
       "        <span class=\"k\">for</span> <span class=\"p\">(</span><span class=\"k\">auto</span> <span class=\"n\">i</span> <span class=\"o\">=</span> <span class=\"mi\">19</span><span class=\"p\">;</span> <span class=\"n\">i</span> <span class=\"o\">&gt;=</span> <span class=\"mi\">0</span><span class=\"p\">;</span> <span class=\"n\">i</span> <span class=\"o\">-=</span> <span class=\"mi\">2</span><span class=\"p\">)</span> <span class=\"p\">{</span>\n",
       "            <span class=\"c1\">// Loopy-loop j</span>\n",
       "            <span class=\"k\">for</span> <span class=\"p\">(</span><span class=\"k\">auto</span> <span class=\"n\">j</span> <span class=\"o\">=</span> <span class=\"mi\">29</span><span class=\"p\">;</span> <span class=\"n\">j</span> <span class=\"o\">&gt;=</span> <span class=\"mi\">0</span><span class=\"p\">;</span> <span class=\"n\">j</span> <span class=\"o\">-=</span> <span class=\"mi\">1</span><span class=\"p\">)</span> <span class=\"p\">{</span>\n",
       "                <span class=\"p\">{</span>\n",
       "                    <span class=\"kt\">double</span> <span class=\"n\">__in1</span> <span class=\"o\">=</span> <span class=\"n\">A</span><span class=\"p\">[((</span><span class=\"mi\">30</span> <span class=\"o\">*</span> <span class=\"n\">i</span><span class=\"p\">)</span> <span class=\"o\">+</span> <span class=\"n\">j</span><span class=\"p\">)];</span>\n",
       "                    <span class=\"kt\">double</span> <span class=\"n\">__in2</span> <span class=\"o\">=</span> <span class=\"n\">A</span><span class=\"p\">[((</span><span class=\"mi\">30</span> <span class=\"o\">*</span> <span class=\"n\">i</span><span class=\"p\">)</span> <span class=\"o\">+</span> <span class=\"n\">j</span><span class=\"p\">)];</span>\n",
       "                    <span class=\"kt\">double</span> <span class=\"n\">__out</span><span class=\"p\">;</span>\n",
       "\n",
       "                    <span class=\"c1\">///////////////////</span>\n",
       "                    <span class=\"c1\">// Tasklet code (augassign_4_8)</span>\n",
       "                    <span class=\"n\">__out</span> <span class=\"o\">=</span> <span class=\"p\">(</span><span class=\"n\">__in1</span> <span class=\"o\">+</span> <span class=\"n\">__in2</span><span class=\"p\">);</span>\n",
       "                    <span class=\"c1\">///////////////////</span>\n",
       "\n",
       "                    <span class=\"n\">A</span><span class=\"p\">[((</span><span class=\"mi\">30</span> <span class=\"o\">*</span> <span class=\"n\">i</span><span class=\"p\">)</span> <span class=\"o\">+</span> <span class=\"n\">j</span><span class=\"p\">)]</span> <span class=\"o\">=</span> <span class=\"n\">__out</span><span class=\"p\">;</span>\n",
       "                <span class=\"p\">}</span>\n",
       "            <span class=\"p\">}</span>\n",
       "        <span class=\"p\">}</span>\n",
       "    <span class=\"p\">}</span>\n",
       "<span class=\"p\">}</span>\n",
       "<span class=\"p\">}</span>\n",
       "\n",
       "<span class=\"n\">DACE_EXPORTED</span> <span class=\"kt\">void</span> <span class=\"n\">__program_simple</span><span class=\"p\">(</span><span class=\"kt\">double</span> <span class=\"o\">*</span> <span class=\"n\">__restrict__</span> <span class=\"n\">A</span><span class=\"p\">)</span>\n",
       "<span class=\"p\">{</span>\n",
       "<span class=\"n\">__program_simple_internal</span><span class=\"p\">(</span><span class=\"n\">A</span><span class=\"p\">);</span>\n",
       "<span class=\"p\">}</span>\n",
       "\n",
       "<span class=\"n\">DACE_EXPORTED</span> <span class=\"kt\">int</span> <span class=\"n\">__dace_init_simple</span><span class=\"p\">(</span><span class=\"kt\">double</span> <span class=\"o\">*</span> <span class=\"n\">__restrict__</span> <span class=\"n\">A</span><span class=\"p\">)</span>\n",
       "<span class=\"p\">{</span>\n",
       "<span class=\"kt\">int</span> <span class=\"n\">__result</span> <span class=\"o\">=</span> <span class=\"mi\">0</span><span class=\"p\">;</span>\n",
       "\n",
       "<span class=\"k\">return</span> <span class=\"n\">__result</span><span class=\"p\">;</span>\n",
       "<span class=\"p\">}</span>\n",
       "\n",
       "<span class=\"n\">DACE_EXPORTED</span> <span class=\"kt\">void</span> <span class=\"n\">__dace_exit_simple</span><span class=\"p\">(</span><span class=\"kt\">double</span> <span class=\"o\">*</span> <span class=\"n\">__restrict__</span> <span class=\"n\">A</span><span class=\"p\">)</span>\n",
       "<span class=\"p\">{</span>\n",
       "<span class=\"p\">}</span>\n",
       "</pre></div>\n"
      ],
      "text/latex": [
       "\\begin{Verbatim}[commandchars=\\\\\\{\\}]\n",
       "\\PY{c+cm}{/* DaCe AUTO\\PYZhy{}GENERATED FILE. DO NOT MODIFY */}\n",
       "\\PY{c+cp}{\\PYZsh{}}\\PY{c+cp}{include} \\PY{c+cpf}{\\PYZlt{}dace/dace.h\\PYZgt{}}\n",
       "\n",
       "\\PY{k+kt}{void} \\PY{n+nf}{\\PYZus{}\\PYZus{}program\\PYZus{}simple\\PYZus{}internal}\\PY{p}{(}\\PY{k+kt}{double} \\PY{o}{*} \\PY{n}{\\PYZus{}\\PYZus{}restrict\\PYZus{}\\PYZus{}} \\PY{n}{A}\\PY{p}{)}\n",
       "\\PY{p}{\\PYZob{}}\n",
       "\n",
       "    \\PY{p}{\\PYZob{}}\n",
       "        \n",
       "        \n",
       "        \\PY{c+c1}{// Loopy\\PYZhy{}loop i}\n",
       "        \\PY{k}{for} \\PY{p}{(}\\PY{k}{auto} \\PY{n}{i} \\PY{o}{=} \\PY{l+m+mi}{19}\\PY{p}{;} \\PY{n}{i} \\PY{o}{\\PYZgt{}}\\PY{o}{=} \\PY{l+m+mi}{0}\\PY{p}{;} \\PY{n}{i} \\PY{o}{\\PYZhy{}}\\PY{o}{=} \\PY{l+m+mi}{2}\\PY{p}{)} \\PY{p}{\\PYZob{}}\n",
       "            \\PY{c+c1}{// Loopy\\PYZhy{}loop j}\n",
       "            \\PY{k}{for} \\PY{p}{(}\\PY{k}{auto} \\PY{n}{j} \\PY{o}{=} \\PY{l+m+mi}{29}\\PY{p}{;} \\PY{n}{j} \\PY{o}{\\PYZgt{}}\\PY{o}{=} \\PY{l+m+mi}{0}\\PY{p}{;} \\PY{n}{j} \\PY{o}{\\PYZhy{}}\\PY{o}{=} \\PY{l+m+mi}{1}\\PY{p}{)} \\PY{p}{\\PYZob{}}\n",
       "                \\PY{p}{\\PYZob{}}\n",
       "                    \\PY{k+kt}{double} \\PY{n}{\\PYZus{}\\PYZus{}in1} \\PY{o}{=} \\PY{n}{A}\\PY{p}{[}\\PY{p}{(}\\PY{p}{(}\\PY{l+m+mi}{30} \\PY{o}{*} \\PY{n}{i}\\PY{p}{)} \\PY{o}{+} \\PY{n}{j}\\PY{p}{)}\\PY{p}{]}\\PY{p}{;}\n",
       "                    \\PY{k+kt}{double} \\PY{n}{\\PYZus{}\\PYZus{}in2} \\PY{o}{=} \\PY{n}{A}\\PY{p}{[}\\PY{p}{(}\\PY{p}{(}\\PY{l+m+mi}{30} \\PY{o}{*} \\PY{n}{i}\\PY{p}{)} \\PY{o}{+} \\PY{n}{j}\\PY{p}{)}\\PY{p}{]}\\PY{p}{;}\n",
       "                    \\PY{k+kt}{double} \\PY{n}{\\PYZus{}\\PYZus{}out}\\PY{p}{;}\n",
       "\n",
       "                    \\PY{c+c1}{///////////////////}\n",
       "                    \\PY{c+c1}{// Tasklet code (augassign\\PYZus{}4\\PYZus{}8)}\n",
       "                    \\PY{n}{\\PYZus{}\\PYZus{}out} \\PY{o}{=} \\PY{p}{(}\\PY{n}{\\PYZus{}\\PYZus{}in1} \\PY{o}{+} \\PY{n}{\\PYZus{}\\PYZus{}in2}\\PY{p}{)}\\PY{p}{;}\n",
       "                    \\PY{c+c1}{///////////////////}\n",
       "\n",
       "                    \\PY{n}{A}\\PY{p}{[}\\PY{p}{(}\\PY{p}{(}\\PY{l+m+mi}{30} \\PY{o}{*} \\PY{n}{i}\\PY{p}{)} \\PY{o}{+} \\PY{n}{j}\\PY{p}{)}\\PY{p}{]} \\PY{o}{=} \\PY{n}{\\PYZus{}\\PYZus{}out}\\PY{p}{;}\n",
       "                \\PY{p}{\\PYZcb{}}\n",
       "            \\PY{p}{\\PYZcb{}}\n",
       "        \\PY{p}{\\PYZcb{}}\n",
       "    \\PY{p}{\\PYZcb{}}\n",
       "\\PY{p}{\\PYZcb{}}\n",
       "\\PY{p}{\\PYZcb{}}\n",
       "\n",
       "\\PY{n}{DACE\\PYZus{}EXPORTED} \\PY{k+kt}{void} \\PY{n}{\\PYZus{}\\PYZus{}program\\PYZus{}simple}\\PY{p}{(}\\PY{k+kt}{double} \\PY{o}{*} \\PY{n}{\\PYZus{}\\PYZus{}restrict\\PYZus{}\\PYZus{}} \\PY{n}{A}\\PY{p}{)}\n",
       "\\PY{p}{\\PYZob{}}\n",
       "\\PY{n}{\\PYZus{}\\PYZus{}program\\PYZus{}simple\\PYZus{}internal}\\PY{p}{(}\\PY{n}{A}\\PY{p}{)}\\PY{p}{;}\n",
       "\\PY{p}{\\PYZcb{}}\n",
       "\n",
       "\\PY{n}{DACE\\PYZus{}EXPORTED} \\PY{k+kt}{int} \\PY{n}{\\PYZus{}\\PYZus{}dace\\PYZus{}init\\PYZus{}simple}\\PY{p}{(}\\PY{k+kt}{double} \\PY{o}{*} \\PY{n}{\\PYZus{}\\PYZus{}restrict\\PYZus{}\\PYZus{}} \\PY{n}{A}\\PY{p}{)}\n",
       "\\PY{p}{\\PYZob{}}\n",
       "\\PY{k+kt}{int} \\PY{n}{\\PYZus{}\\PYZus{}result} \\PY{o}{=} \\PY{l+m+mi}{0}\\PY{p}{;}\n",
       "\n",
       "\\PY{k}{return} \\PY{n}{\\PYZus{}\\PYZus{}result}\\PY{p}{;}\n",
       "\\PY{p}{\\PYZcb{}}\n",
       "\n",
       "\\PY{n}{DACE\\PYZus{}EXPORTED} \\PY{k+kt}{void} \\PY{n}{\\PYZus{}\\PYZus{}dace\\PYZus{}exit\\PYZus{}simple}\\PY{p}{(}\\PY{k+kt}{double} \\PY{o}{*} \\PY{n}{\\PYZus{}\\PYZus{}restrict\\PYZus{}\\PYZus{}} \\PY{n}{A}\\PY{p}{)}\n",
       "\\PY{p}{\\PYZob{}}\n",
       "\\PY{p}{\\PYZcb{}}\n",
       "\\end{Verbatim}\n"
      ],
      "text/plain": [
       "/* DaCe AUTO-GENERATED FILE. DO NOT MODIFY */\n",
       "#include <dace/dace.h>\n",
       "\n",
       "void __program_simple_internal(double * __restrict__ A)\n",
       "{\n",
       "\n",
       "    {\n",
       "        \n",
       "        \n",
       "        // Loopy-loop i\n",
       "        for (auto i = 19; i >= 0; i -= 2) {\n",
       "            // Loopy-loop j\n",
       "            for (auto j = 29; j >= 0; j -= 1) {\n",
       "                {\n",
       "                    double __in1 = A[((30 * i) + j)];\n",
       "                    double __in2 = A[((30 * i) + j)];\n",
       "                    double __out;\n",
       "\n",
       "                    ///////////////////\n",
       "                    // Tasklet code (augassign_4_8)\n",
       "                    __out = (__in1 + __in2);\n",
       "                    ///////////////////\n",
       "\n",
       "                    A[((30 * i) + j)] = __out;\n",
       "                }\n",
       "            }\n",
       "        }\n",
       "    }\n",
       "}\n",
       "}\n",
       "\n",
       "DACE_EXPORTED void __program_simple(double * __restrict__ A)\n",
       "{\n",
       "__program_simple_internal(A);\n",
       "}\n",
       "\n",
       "DACE_EXPORTED int __dace_init_simple(double * __restrict__ A)\n",
       "{\n",
       "int __result = 0;\n",
       "\n",
       "return __result;\n",
       "}\n",
       "\n",
       "DACE_EXPORTED void __dace_exit_simple(double * __restrict__ A)\n",
       "{\n",
       "}\n"
      ]
     },
     "execution_count": 9,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# Change schedule\n",
    "for node, _ in sdfg.all_nodes_recursive():\n",
    "    if isinstance(node, dace.nodes.MapEntry):\n",
    "        node.schedule = dace.ScheduleType.LoopyLoop\n",
    "\n",
    "Code(sdfg.generate_code()[0].clean_code, language='cpp')"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "and the code is generated appropriately."
   ]
  }
 ],
 "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.6"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}