typo

benchmark/benchmark.ipynb

@@ -28,20 +28,7 @@
     "    output = type(n).fill_(0)\n",
     "    index = torch.LongTensor(n).fill_(0)\n",
     "    input = type(n).fill_(1)\n",
-    "    return output, index, input\n",
-    "\n",
-    "n = 1000000\n",
-    "num_runs = 100"
-   ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": 14,
-   "metadata": {},
-   "outputs": [],
-   "source": [
-    "types = [torch.FloatTensor, torch.DoubleTensor, torch.ByteTensor, torch.CharTensor,\n",
-    "         torch.ShortTensor, torch.IntTensor, torch.LongTensor]"
+    "    return output, index, input"
    ]
   },
   {
@@ -76,6 +63,19 @@
     "    return runtimes"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": 14,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "types = [torch.FloatTensor, torch.DoubleTensor, torch.ByteTensor, torch.CharTensor,\n",
+    "         torch.ShortTensor, torch.IntTensor, torch.LongTensor]\n",
+    "\n",
+    "n = 1000000\n",
+    "num_runs = 100"
+   ]
+  },
   {
    "cell_type": "markdown",
    "metadata": {},