zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-24 09:21:16 +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)
if(NOT MLX_VERSION)
set(MLX_VERSION 0.7.0)
set(MLX_VERSION 0.8.0)
endif()
# --------------------- Processor tests -------------------------

12
python/src/ops.cpp
View File

@ -1261,7 +1261,7 @@ void init_ops(nb::module_& m) {
"start"_a,
"stop"_a,
"num"_a = 50,
"dtype"_a = float32,
"dtype"_a.none() = float32,
"stream"_a = nb::none(),
nb::sig(
"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,
"dtype"_a = float32,
"dtype"_a.none() = float32,
nb::kw_only(),
"stream"_a = nb::none(),
nb::sig(
@ -1442,7 +1442,7 @@ void init_ops(nb::module_& m) {
}
},
"shape"_a,
"dtype"_a = float32,
"dtype"_a.none() = float32,
nb::kw_only(),
"stream"_a = nb::none(),
nb::sig(
@ -1487,7 +1487,7 @@ void init_ops(nb::module_& m) {
"n"_a,
"m"_a = nb::none(),
"k"_a = 0,
"dtype"_a = float32,
"dtype"_a.none() = float32,
nb::kw_only(),
"stream"_a = nb::none(),
nb::sig(
@ -1511,7 +1511,7 @@ void init_ops(nb::module_& m) {
return identity(n, dtype.value_or(float32), s);
},
"n"_a,
"dtype"_a = float32,
"dtype"_a.none() = float32,
nb::kw_only(),
"stream"_a = nb::none(),
nb::sig(
@ -1539,7 +1539,7 @@ void init_ops(nb::module_& m) {
"n"_a,
"m"_a = nb::none(),
"k"_a = 0,
"dtype"_a = float32,
"dtype"_a.none() = float32,
nb::kw_only(),
"stream"_a = nb::none(),
nb::sig(

2
setup.py
View File

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