zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-11-04 10:38:10 +08:00
Code Issues Actions 1 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
3 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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(