Fonctions_Nathan.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 30,
   "metadata": {},
   "outputs": [],
   "source": [
    "\n",
    "import numpy as np\n",
    "\n",
    "G=6\n",
    "S=3\n",
    "T=3\n",
    "C=1\n",
    "\n",
    "\n",
    "\n",
    "\n",
    "# task type 0: POTRF 1:SYRK 2:TRSM 3: GEMMS\n",
    "# all have 4 values, but the last one is used for GEMMS only (-;\n",
    "# a task is a 4-uple and everything will be stored in dictionnaries later\n",
    "# succalap is the set of successors in the alap schedule, ie the predecessors in the original graph\n",
    "\n",
    "\n",
    "\n",
    "def succASAP(x,n):\n",
    "    tasktype=x[0]\n",
    "    i=x[1]\n",
    "    j=x[2]\n",
    "    k=x[3]\n",
    "    listsucc=[]\n",
    "    if tasktype==0:\n",
    "        if i<n:\n",
    "            for j in range(i+1,n+1,1):\n",
    "                y=(2,i,j,0)\n",
    "                listsucc.append(y)\n",
    "        else:\n",
    "            y=(4,0,0,0)\n",
    "            listsucc.append(y)  \n",
    "    \n",
    "    \n",
    "    if tasktype==1:\n",
    "        if j<i-1:\n",
    "            y=(1,i,j+1,0)\n",
    "            listsucc.append(y)\n",
    "        else:\n",
    "            y=(0,i,0,0)\n",
    "            listsucc.append(y)\n",
    "\n",
    "           \n",
    "            \n",
    "    if tasktype==2:\n",
    "        if i<=n-1:\n",
    "            for k in range(i+1,j):\n",
    "                y=(3,k,j,i)\n",
    "                listsucc.append(y)\n",
    "            for k in range(j+1,n+1):\n",
    "                y=(3,j,k,i)\n",
    "                listsucc.append(y)\n",
    "            y=(1,j,i,0)\n",
    "            listsucc.append(y)\n",
    "\n",
    "\n",
    "    if tasktype==3:\n",
    "        if k<i-1:\n",
    "            y=(3,i,j,k+1)\n",
    "            listsucc.append(y)\n",
    "        else:\n",
    "            y=(2,i,j,0)\n",
    "            listsucc.append(y)\n",
    "\n",
    "\n",
    "    return listsucc\n",
    "\n",
    "\n",
    "# duration is just the duration of a task\n",
    "\n",
    "def duration(x):\n",
    "    tasktype=x[0]\n",
    "    if tasktype==0:\n",
    "        duration=C\n",
    "    if tasktype==1:\n",
    "        duration=S        \n",
    "    if tasktype==2:\n",
    "        duration=T\n",
    "    if tasktype==3:\n",
    "        duration=G\n",
    "    if tasktype==4:\n",
    "        duration=0\n",
    "    return duration\n",
    "\n",
    "durations = [C, S, T, G]\n",
    "\n",
    "\n",
    "def CPAndWorkBelow(x,n):\n",
    "    ReadyTasks=[]\n",
    "    ReadyTasks.append(x)\n",
    "    Seen=[]\n",
    "    ToVisit=[]\n",
    "    ToVisit.append(x)\n",
    "    TotalWork=duration(x)\n",
    "    CPl=0\n",
    "    while len(ToVisit)>0:\n",
    "        for t in ToVisit:\n",
    "            for succ in succASAP(t,n):\n",
    "                if succ not in Seen:\n",
    "                    TotalWork=TotalWork+duration(succ)\n",
    "                    Seen.append(succ)\n",
    "                    ToVisit.append(succ)\n",
    "            ToVisit.remove(t)\n",
    "            \n",
    "    tasktype=x[0]\n",
    "    if tasktype==0:\n",
    "        CPl=C+(n-x[1])*(T+S+C)\n",
    "    if tasktype==1:\n",
    "        CPl=(x[1]-x[2])*S+C+(n-x[1])*(T+S+C)       \n",
    "    if tasktype==2:\n",
    "        CPl=(x[2]-x[1]-1)*(T+G)+(n-x[2]+1)*(T+S+C)\n",
    "    if tasktype==3:\n",
    "        CPl=(x[1]-x[3])*G+(x[2]-x[1]-1)*(T + G)+(n-x[2] +1)*(T +S +C)    \n",
    "    \n",
    "    \n",
    "    \n",
    "    return(CPl,TotalWork)\n",
    "\n",
    "            \n",
    "\n",
    "    \n",
    "    \n",
    "    \n",
    "    \n",
    "    \n",
    "    \n",
    "    \n",
    "    \n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 31,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "(50, 512)\n",
      "512.0 512 512\n",
      "50 50\n"
     ]
    }
   ],
   "source": [
    "n=8\n",
    "R0=CPAndWorkBelow((0,1,0,0),n)  \n",
    "print(R0)\n",
    "CPl,TotalWork=CPAndWorkBelow((0,1,0,0),n) \n",
    "#CPl,TotalWork=R0[0],R0[1]\n",
    "Wtot=n*(n-1)*(n-2)/6*G+n*(n-1)/2*(S+T)+n*C\n",
    "R0=CPAndWorkBelow((0,2,0,0),n+1)       \n",
    "CPl1,TotalWork1=R0[0],R0[1]\n",
    "\n",
    "print(Wtot,TotalWork,TotalWork1)         \n",
    "print(CPl,CPl1)         \n",
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "anaconda-cloud": {},
  "kernelspec": {
   "display_name": "Python [conda env:py36]",
   "language": "python",
   "name": "conda-env-py36-py"
  },
  "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.6.6"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}