zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-19 16:48:10 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fp64 on the CPU (#1843)

Browse Source 
* add fp64 data type

* clean build

* update docs

* fix bug
This commit is contained in:
Awni Hannun
2025-02-07 15:52:22 -08:00
committed by GitHub
parent 1a1b2108ec
commit 1c0c118f7c
32 changed files with 438 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/src/array.cpp
View File

@@ -128,6 +128,7 @@ void init_array(nb::module_& m) {
m.attr("int64") = nb::cast(mx::int64);
m.attr("float16") = nb::cast(mx::float16);
m.attr("float32") = nb::cast(mx::float32);
m.attr("float64") = nb::cast(mx::float64);
m.attr("bfloat16") = nb::cast(mx::bfloat16);
m.attr("complex64") = nb::cast(mx::complex64);
nb::enum_<mx::Dtype::Category>(
@@ -163,6 +164,7 @@ void init_array(nb::module_& m) {
* :ref:`float16 <data_types>`
* :ref:`bfloat16 <data_types>`
* :ref:`float32 <data_types>`
* :ref:`float64 <data_types>`
* :attr:`~mlx.core.complexfloating`