6-speedup-efficiency-api.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>nid</th>\n",
       "      <th>name</th>\n",
       "      <th>125.time</th>\n",
       "      <th>216.time</th>\n",
       "      <th>512.time</th>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>node</th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "      <th></th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>{'name': 'CalcHydroConstraintForElems', 'type': 'region'}</th>\n",
       "      <td>23</td>\n",
       "      <td>CalcHydroConstraintForElems</td>\n",
       "      <td>0.895616</td>\n",
       "      <td>0.994473</td>\n",
       "      <td>0.871081</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>{'name': 'CalcSoundSpeedForElems', 'type': 'region'}</th>\n",
       "      <td>43</td>\n",
       "      <td>CalcSoundSpeedForElems</td>\n",
       "      <td>0.761886</td>\n",
       "      <td>1.062826</td>\n",
       "      <td>0.899906</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>{'name': 'CalcCourantConstraintForElems', 'type': 'region'}</th>\n",
       "      <td>20</td>\n",
       "      <td>CalcCourantConstraintForElems</td>\n",
       "      <td>0.791160</td>\n",
       "      <td>0.990797</td>\n",
       "      <td>0.923470</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "                                                    nid  \\\n",
       "node                                                      \n",
       "{'name': 'CalcHydroConstraintForElems', 'type':...   23   \n",
       "{'name': 'CalcSoundSpeedForElems', 'type': 'reg...   43   \n",
       "{'name': 'CalcCourantConstraintForElems', 'type...   20   \n",
       "\n",
       "                                                                             name  \\\n",
       "node                                                                                \n",
       "{'name': 'CalcHydroConstraintForElems', 'type':...    CalcHydroConstraintForElems   \n",
       "{'name': 'CalcSoundSpeedForElems', 'type': 'reg...         CalcSoundSpeedForElems   \n",
       "{'name': 'CalcCourantConstraintForElems', 'type...  CalcCourantConstraintForElems   \n",
       "\n",
       "                                                    125.time  216.time  \\\n",
       "node                                                                     \n",
       "{'name': 'CalcHydroConstraintForElems', 'type':...  0.895616  0.994473   \n",
       "{'name': 'CalcSoundSpeedForElems', 'type': 'reg...  0.761886  1.062826   \n",
       "{'name': 'CalcCourantConstraintForElems', 'type...  0.791160  0.990797   \n",
       "\n",
       "                                                    512.time  \n",
       "node                                                          \n",
       "{'name': 'CalcHydroConstraintForElems', 'type':...  0.871081  \n",
       "{'name': 'CalcSoundSpeedForElems', 'type': 'reg...  0.899906  \n",
       "{'name': 'CalcCourantConstraintForElems', 'type...  0.923470  "
      ]
     },
     "metadata": {},
     "output_type": "display_data"
    }
   ],
   "source": [
    "import hatchet as ht\n",
    "import glob\n",
    "import re\n",
    "\n",
    "profiles = glob.glob(\n",
    "    \"/home/ocankur/research/research-code/hatchet-aditya-tomar/hatchet-tutorial/data/lulesh-scaling/chopper-example/*\"\n",
    ")\n",
    "\n",
    "gfs_eff_simple = ht.GraphFrame.construct_from(profiles)\n",
    "gfs_eff_simple_flt = []\n",
    "\n",
    "profile_gf = dict(zip(profiles, gfs_eff_simple))\n",
    "for profile, gf in profile_gf.items():\n",
    "    num_pes = re.match(\"(.*)-(\\d+)(.*)\", profile).group(2)\n",
    "    gf.update_metadata(int(num_pes))\n",
    "    gfs_eff_simple_flt.append(gf.filter(lambda x: not x['name'].startswith('MPI')))\n",
    "    \n",
    "lulesh_simple_weak = ht.Chopper.speedup_efficiency(gfs_eff_simple_flt, metrics=[\"time\"], weak=True, efficiency=True)\n",
    "display(lulesh_simple_weak.head(3))"
   ]
  }
 ],
 "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.7"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}