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

Bump extension MLX version (#1350)

Browse Source 
* Bump extension MLX version

* fix some docs nits
This commit is contained in:
Awni Hannun 2024-08-23 12:38:34 -07:00 committed by GitHub
parent 684e11c664
commit 3b4d5484c7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
4 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
docs/src/python/ops.rst
View File

@ -44,6 +44,7 @@ Operations
convolve convolve
conv1d conv1d
conv2d conv2d
conv3d
conv_general conv_general
cos cos
cosh cosh

2
examples/extensions/pyproject.toml
View File

@ -2,7 +2,7 @@
requires = [ requires = [
"setuptools>=42", "setuptools>=42",
"cmake>=3.24", "cmake>=3.24",
"mlx>=0.16.2", "mlx>=0.17.0",
"nanobind==2.1.0", "nanobind==2.1.0",
] ]
build-backend = "setuptools.build_meta" build-backend = "setuptools.build_meta"

2
examples/extensions/requirements.txt
View File

@ -1,4 +1,4 @@
setuptools>=42 setuptools>=42
cmake>=3.24 cmake>=3.24
mlx>=0.16.2 mlx>=0.17.0
nanobind==2.1.0 nanobind==2.1.0

5
python/src/fast.cpp
View File

@ -102,11 +102,12 @@ void init_fast(nb::module_& parent_module) {
implementation which rotates consecutive dimensions. implementation which rotates consecutive dimensions.
base (float, optional): The base used to compute angular frequency for base (float, optional): The base used to compute angular frequency for
each dimension in the positional encodings. Exactly one of ``base`` and each dimension in the positional encodings. Exactly one of ``base`` and
``freqs`` must be ``None``. ``freqs`` must be ``None``.
scale (float): The scale used to scale the positions. scale (float): The scale used to scale the positions.
offset (int): The position offset to start at. offset (int): The position offset to start at.
freqs (array, optional): Optional frequencies to use with RoPE. freqs (array, optional): Optional frequencies to use with RoPE.
If set, the ``base`` parameter must be ``None``. ``Default: None``. If set, the ``base`` parameter must be ``None``. Default: ``None``.
Returns: Returns:
array: The output array. array: The output array.
)pbdoc"); )pbdoc");