mirror of
https://github.com/ml-explore/mlx.git
synced 2025-10-20 01:18:12 +08:00
Fp64 on the CPU (#1843)
* add fp64 data type * clean build * update docs * fix bug
This commit is contained in:
@@ -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`
|
||||
|
||||
|
Reference in New Issue
Block a user