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

Bump (#869)

Browse Source 
* bump

* fix none in a few ops
This commit is contained in:
Awni Hannun 2024-03-21 13:56:56 -07:00 committed by GitHub
parent 2225374060
commit 44390bd3d0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
CMakeLists.txt
View File

@ -18,7 +18,7 @@ option(MLX_BUILD_METAL "Build metal backend" ON)
option(BUILD_SHARED_LIBS "Build mlx as a shared library" OFF) option(BUILD_SHARED_LIBS "Build mlx as a shared library" OFF)
if(NOT MLX_VERSION) if(NOT MLX_VERSION)
set(MLX_VERSION 0.7.0) set(MLX_VERSION 0.8.0)
endif() endif()
# --------------------- Processor tests ------------------------- # --------------------- Processor tests -------------------------

12
python/src/ops.cpp
View File

@ -1261,7 +1261,7 @@ void init_ops(nb::module_& m) {
"start"_a, "start"_a,
"stop"_a, "stop"_a,
"num"_a = 50, "num"_a = 50,
"dtype"_a = float32, "dtype"_a.none() = float32,
"stream"_a = nb::none(), "stream"_a = nb::none(),
nb::sig( nb::sig(
"def linspace(start, stop, num: Optional[int] = 50, dtype: Optional[Dtype] = float32, stream: Union[None, Stream, Device] = None) -> array"), "def linspace(start, stop, num: Optional[int] = 50, dtype: Optional[Dtype] = float32, stream: Union[None, Stream, Device] = None) -> array"),
@ -1396,7 +1396,7 @@ void init_ops(nb::module_& m) {
} }
}, },
"shape"_a, "shape"_a,
"dtype"_a = float32, "dtype"_a.none() = float32,
nb::kw_only(), nb::kw_only(),
"stream"_a = nb::none(), "stream"_a = nb::none(),
nb::sig( nb::sig(
@ -1442,7 +1442,7 @@ void init_ops(nb::module_& m) {
} }
}, },
"shape"_a, "shape"_a,
"dtype"_a = float32, "dtype"_a.none() = float32,
nb::kw_only(), nb::kw_only(),
"stream"_a = nb::none(), "stream"_a = nb::none(),
nb::sig( nb::sig(
@ -1487,7 +1487,7 @@ void init_ops(nb::module_& m) {
"n"_a, "n"_a,
"m"_a = nb::none(), "m"_a = nb::none(),
"k"_a = 0, "k"_a = 0,
"dtype"_a = float32, "dtype"_a.none() = float32,
nb::kw_only(), nb::kw_only(),
"stream"_a = nb::none(), "stream"_a = nb::none(),
nb::sig( nb::sig(
@ -1511,7 +1511,7 @@ void init_ops(nb::module_& m) {
return identity(n, dtype.value_or(float32), s); return identity(n, dtype.value_or(float32), s);
}, },
"n"_a, "n"_a,
"dtype"_a = float32, "dtype"_a.none() = float32,
nb::kw_only(), nb::kw_only(),
"stream"_a = nb::none(), "stream"_a = nb::none(),
nb::sig( nb::sig(
@ -1539,7 +1539,7 @@ void init_ops(nb::module_& m) {
"n"_a, "n"_a,
"m"_a = nb::none(), "m"_a = nb::none(),
"k"_a = 0, "k"_a = 0,
"dtype"_a = float32, "dtype"_a.none() = float32,
nb::kw_only(), nb::kw_only(),
"stream"_a = nb::none(), "stream"_a = nb::none(),
nb::sig( nb::sig(

2
setup.py
View File

@ -154,7 +154,7 @@ if __name__ == "__main__":
setup( setup(
name="mlx", name="mlx",
version=get_version("0.7.0"), version=get_version("0.8.0"),
author="MLX Contributors", author="MLX Contributors",
author_email="mlx@group.apple.com", author_email="mlx@group.apple.com",
description="A framework for machine learning on Apple silicon.", description="A framework for machine learning on Apple silicon.",