mirror of
https://github.com/ml-explore/mlx.git
synced 2025-07-19 07:31:26 +08:00
dfdb284e16
All functions that take an optional dtype should
* have a default dtype visible in the generated docs (accomplished via `"dtype"_a = std::optional{float32}`)
* behave identical when `dtype=None` or no dtype is passed
This important when passing kw args down from a numpy function like:
```
def f(x, dtype=None):
mx.random.uniform(dtype=dtype)
# ...
```
NumPy functions behave like this.
It also fixes a minor bug in `tri`: #378
Closes #378
|
||
|---|---|---|
| .. | ||
| array.cpp | ||
| CMakeLists.txt | ||
| device.cpp | ||
| fft.cpp | ||
| indexing.cpp | ||
| indexing.h | ||
| linalg.cpp | ||
| load.cpp | ||
| load.h | ||
| metal.cpp | ||
| mlx.cpp | ||
| ops.cpp | ||
| random.cpp | ||
| stream.cpp | ||
| transforms.cpp | ||
| utils.h | ||