tutorials/tutorial-4/xdsl-tutorial-4.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 2,
   "id": "71cf4ed4-38c4-4038-b949-5e24253dc572",
   "metadata": {},
   "outputs": [],
   "source": [
    "from xdsl.dialects.experimental import aie\n",
    "from xdsl.dialects.builtin import Region, IndexType, ModuleOp, i32, IntegerAttr, ArrayAttr, i64, StringAttr\n",
    "from xdsl.dialects.arith import Constant\n",
    "from xdsl.dialects.memref import MemRefType\n",
    "from xdsl.dialects import builtin, arith, memref, func\n",
    "from xdsl.builder import Builder\n",
    "from xdsl.traits import SymbolTable\n",
    "\n",
    "def I32Attr(value):\n",
    "    return IntegerAttr.from_int_and_width(value, i32)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "id": "f146e44c-e14b-4ba5-90b3-1306fa9aa88e",
   "metadata": {},
   "outputs": [],
   "source": [
    "def tutorial4():\n",
    "    col = IntegerAttr.from_int_and_width(1, i32)\n",
    "    row = IntegerAttr.from_int_and_width(4, i32)\n",
    "    tile14 = aie.TileOp(col, row)\n",
    "\n",
    "    col = IntegerAttr.from_int_and_width(3, i32)\n",
    "    row = IntegerAttr.from_int_and_width(4, i32)\n",
    "    tile34 = aie.TileOp(col, row)\n",
    "\n",
    "    elem_number = IntegerAttr.from_int_and_width(1, i32)\n",
    "    object_fifo = aie.createObjectFifo(elem_number, tile14, tile34, i32, [256], \"of\")\n",
    "\n",
    "    lock34_8 = aie.LockOp(I32Attr(8), I32Attr(0), tile34, StringAttr(\"lock_a34_8\"))\n",
    "\n",
    "    arith.Constant.from_int_and_width\n",
    "\n",
    "    @Builder.region\n",
    "    def region0(builder: Builder):\n",
    "        builder.insert(tile14)\n",
    "        builder.insert(tile34)\n",
    "        builder.insert(object_fifo)\n",
    "        builder.insert(lock34_8)\n",
    "        \n",
    "    device_val = IntegerAttr.from_int_and_width(0, i32)\n",
    "    device = aie.DeviceOp(device_val, region0)\n",
    "\n",
    "    acquire_fifo = aie.ObjectFifoAcquireOp(I32Attr(aie.PRODUCE_PORT), I32Attr(1), object_fifo.sym_name, device)\n",
    "    \n",
    "    @Builder.region\n",
    "    def core14_region(builder: Builder):\n",
    "        builder.insert(acquire_fifo)\n",
    "\n",
    "    stackSize = IntegerAttr.from_int_and_width(1, i32)\n",
    "    core14 = aie.CoreOp(stackSize, tile14, core14_region)\n",
    "\n",
    "\n",
    "    region0.block.add_op(core14)\n",
    "\n",
    "    subview = aie.ObjectFIFOSubviewAccessOp(I32Attr(0), acquire_fifo)\n",
    "    core14.region.block.add_op(subview)\n",
    "\n",
    "    module = builtin.ModuleOp([device])\n",
    "    value_14 = Constant.from_int_and_width(14, i32)\n",
    "    idx_3 = Constant.from_int_and_width(3, IndexType())\n",
    "    memref_store = memref.Store.get(value_14, subview, idx_3)\n",
    "    release_fifo = aie.ObjectFIFOReleaseOp(I32Attr(0), I32Attr(1), object_fifo.sym_name)\n",
    "\n",
    "    core14.region.block.add_op(value_14)\n",
    "    core14.region.block.add_op(idx_3)\n",
    "    core14.region.block.add_op(memref_store)\n",
    "    core14.region.block.add_op(release_fifo)\n",
    "    core14.region.block.add_op(aie.EndOp())\n",
    "\n",
    "    acquire_fifo = aie.ObjectFifoAcquireOp(I32Attr(aie.CONSUME_PORT), I32Attr(1), object_fifo.sym_name, device)\n",
    "    \n",
    "    @Builder.region\n",
    "    def core34_region(builder: Builder):\n",
    "        uselock34_8_acquire = aie.UseLockOp(I32Attr(0), I32Attr(aie.LOCK_ACQUIRE), I32Attr(1), lock34_8)\n",
    "        builder.insert(uselock34_8_acquire)\n",
    "        builder.insert(acquire_fifo)\n",
    "\n",
    "    core34 = aie.CoreOp(stackSize, tile34, core34_region)\n",
    "\n",
    "    subview = aie.ObjectFIFOSubviewAccessOp(I32Attr(0), acquire_fifo)\n",
    "    idx_1 = Constant.from_int_and_width(3, IndexType())\n",
    "    d1 = memref.Load.get(subview, idx_1)\n",
    "    c1 = Constant.from_int_and_width(100, i32)\n",
    "    d2 = arith.Addi(d1, c1)\n",
    "    idx_2 = Constant.from_int_and_width(5, IndexType())\n",
    "    store = memref.Store.get(d2, subview, idx_2)\n",
    "    release_fifo = aie.ObjectFIFOReleaseOp(I32Attr(aie.CONSUME_PORT), I32Attr(1), object_fifo.sym_name)\n",
    "    uselock34_8_release = aie.UseLockOp(I32Attr(1), I32Attr(aie.LOCK_RELEASE), I32Attr(1), lock34_8)\n",
    "\n",
    "    core34.region.block.add_op(subview)\n",
    "    core34.region.block.add_op(idx_1)\n",
    "    core34.region.block.add_op(d1)\n",
    "    core34.region.block.add_op(c1)\n",
    "    core34.region.block.add_op(d2)\n",
    "    core34.region.block.add_op(idx_2)\n",
    "    core34.region.block.add_op(store)\n",
    "    core34.region.block.add_op(release_fifo)\n",
    "    core34.region.block.add_op(uselock34_8_release)\n",
    "    core34.region.block.add_op(aie.EndOp())\n",
    "\n",
    "    region0.block.add_op(core34)\n",
    "\n",
    "    print(module)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "id": "fbaf4fc9-d1d5-4376-8719-9712dfbbb892",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "builtin.module {\n",
      "  AIE.device(xcvc1902) {\n",
      "    %0 = \"AIE.tile\"() {\"col\" = 1 : i32, \"row\" = 4 : i32} : () -> index\n",
      "    %1 = \"AIE.tile\"() {\"col\" = 3 : i32, \"row\" = 4 : i32} : () -> index\n",
      "    AIE.objectFifo @of(%0, {%1}, 1 : i32) : !AIE.objectFifo<memref<256xi32>>\n",
      "    %2 = \"AIE.lock\"(%1) {\"lockID\" = 8 : i32, \"init\" = 0 : i32, \"sym_name\" = \"lock_a34_8\"} : (index) -> index\n",
      "    %3 = AIE.core (%0) {\n",
      "      %4 = AIE.objectFifo.acquire @of (Produce, 1) : !AIE.objectFifoSubview<memref<256xi32>>\n",
      "      %5 = AIE.objectFifo.subview.access %4[0] : !AIE.objectFifoSubview<memref<256xi32>> -> memref<256xi32>\n",
      "      %6 = arith.constant 14 : i32\n",
      "      %7 = arith.constant 3 : index\n",
      "      memref.store %6, %5[%7] : memref<256xi32>\n",
      "      AIE.objectFifo.release @of (Produce, 1)\n",
      "      \"AIE.end\"() : () -> ()\n",
      "    }\n",
      "    %8 = AIE.core (%1) {\n",
      "      AIE.useLock(%2, \"Acquire\", 0)\n",
      "      %9 = AIE.objectFifo.acquire @of (Consume, 1) : !AIE.objectFifoSubview<memref<256xi32>>\n",
      "      %10 = AIE.objectFifo.subview.access %9[0] : !AIE.objectFifoSubview<memref<256xi32>> -> memref<256xi32>\n",
      "      %11 = arith.constant 3 : index\n",
      "      %12 = memref.load %10[%11] : memref<256xi32>\n",
      "      %13 = arith.constant 100 : i32\n",
      "      %14 = arith.addi %12, %13 : i32\n",
      "      %15 = arith.constant 5 : index\n",
      "      memref.store %14, %10[%15] : memref<256xi32>\n",
      "      AIE.objectFifo.release @of (Consume, 1)\n",
      "      AIE.useLock(%2, \"Release\", 1)\n",
      "      \"AIE.end\"() : () -> ()\n",
      "    }\n",
      "  }\n",
      "}\n"
     ]
    }
   ],
   "source": [
    "tutorial4()"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "env_xdsl",
   "language": "python",
   "name": "env_xdsl"
  },
  "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.10.6"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}