zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-12-16 01:49:05 +08:00
Code Issues Actions 1 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

5
mlx/utils.h
View File

@@ -47,6 +47,7 @@ struct PrintFormatter {
inline void print(std::ostream& os, float16_t val);
inline void print(std::ostream& os, bfloat16_t val);
inline void print(std::ostream& os, float val);
inline void print(std::ostream& os, double val);
inline void print(std::ostream& os, complex64_t val);
bool capitalize_bool{false};
@@ -61,8 +62,8 @@ void abort_with_exception(const std::exception& error);
struct finfo {
explicit finfo(Dtype dtype);
Dtype dtype;
float min;
float max;
double min;
double max;
};
/** The type from promoting the arrays' types with one another. */