zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-27 00:08:09 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Refactor code examples to use 'gelu' (#2592)

Browse Source 
Updated code examples to use 'gelu' directly instead of 'nn.gelu'.
This commit is contained in:
Umberto Mignozzetti
2025-09-15 09:47:02 -07:00
committed by GitHub
parent ee18e1cbf0
commit 36cad99a11
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
docs/src/usage/compile.rst
View File

@@ -130,8 +130,8 @@ Now make an array, and benchmark both functions:
.. code-block:: python .. code-block:: python
x = mx.random.uniform(shape=(32, 1000, 4096)) x = mx.random.uniform(shape=(32, 1000, 4096))
timeit(nn.gelu, x) timeit(gelu, x)
timeit(mx.compile(nn.gelu), x) timeit(mx.compile(gelu), x)
On an M1 Max the times are 15.5 and 3.1 milliseconds. The compiled ``gelu`` is On an M1 Max the times are 15.5 and 3.1 milliseconds. The compiled ``gelu`` is
five times faster. five times faster.