Skip to content

Commit 3ab3441

Browse files
committed
Typo
1 parent 14923ea commit 3ab3441

File tree

1 file changed

+15
-17
lines changed

1 file changed

+15
-17
lines changed

docs/source/04-benchmark.ipynb

Lines changed: 15 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -2,18 +2,9 @@
22
"cells": [
33
{
44
"cell_type": "code",
5-
"execution_count": 12,
5+
"execution_count": 1,
66
"metadata": {},
7-
"outputs": [
8-
{
9-
"name": "stdout",
10-
"output_type": "stream",
11-
"text": [
12-
"The autoreload extension is already loaded. To reload it, use:\n",
13-
" %reload_ext autoreload\n"
14-
]
15-
}
16-
],
7+
"outputs": [],
178
"source": [
189
"# Copyright (c) 2024 Graphcore Ltd. All rights reserved.\n",
1910
"\n",
@@ -39,24 +30,31 @@
3930
"\n",
4031
"The `gfloat` library is designed for readability over performance, and the reference code for computations is the (slow) scalar code e.g. `round_float`.\n",
4132
"\n",
42-
"There are vectorized implementations (e.g. `round_ndarray`, and when combined with JAX, these can go reasonably fast).\n",
33+
"There are vectorized implementations (e.g. `round_ndarray`), and when combined with JAX, these can go reasonably fast.\n",
4334
"\n",
4435
"Let's see how long it takes to encoed some values to FP8..."
4536
]
4637
},
4738
{
4839
"cell_type": "code",
49-
"execution_count": null,
40+
"execution_count": 2,
5041
"metadata": {},
5142
"outputs": [
43+
{
44+
"name": "stderr",
45+
"output_type": "stream",
46+
"text": [
47+
"An NVIDIA GPU may be present on this machine, but a CUDA-enabled jaxlib is not installed. Falling back to cpu.\n"
48+
]
49+
},
5250
{
5351
"name": "stdout",
5452
"output_type": "stream",
5553
"text": [
56-
"GFloat scalar : 6996.75 nsec (50 runs at size 10000)\n",
57-
"GFloat vectorized, numpy arrays: 75.04 nsec (50 runs at size 1000000)\n",
58-
"GFloat vectorized, JAX JIT : 3.18 nsec (1000 runs at size 1000000)\n",
59-
"ML_dtypes : 3.13 nsec (1000 runs at size 1000000)\n"
54+
"GFloat scalar : 6666.04 nsec (50 runs at size 10000)\n",
55+
"GFloat vectorized, numpy arrays: 57.84 nsec (50 runs at size 1000000)\n",
56+
"GFloat vectorized, JAX JIT : 3.17 nsec (1000 runs at size 1000000)\n",
57+
"ML_dtypes : 2.92 nsec (1000 runs at size 1000000)\n"
6058
]
6159
}
6260
],

0 commit comments

Comments
 (0)