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 Packages Projects Releases Wiki Activity

fix pinv (#2110)

Browse Source
This commit is contained in:
Awni Hannun
2025-04-23 13:08:28 -07:00
committed by GitHub
parent 38c1e720c2
commit fbc89e3ced
7 changed files with 34 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
mlx/utils.cpp
View File

@@ -283,9 +283,10 @@ int get_var(const char* name, int default_value) {
} // namespace env
template <typename T>
void set_finfo_limits(double& min, double& max) {
void set_finfo_limits(double& min, double& max, double& eps) {
min = numeric_limits<T>::lowest();
max = numeric_limits<T>::max();
eps = numeric_limits<T>::epsilon();
}
finfo::finfo(Dtype dtype) : dtype(dtype) {
@@ -295,16 +296,16 @@ finfo::finfo(Dtype dtype) : dtype(dtype) {
throw std::invalid_argument(msg.str());
}
if (dtype == float32) {
set_finfo_limits<float>(min, max);
set_finfo_limits<float>(min, max, eps);
} else if (dtype == float16) {
set_finfo_limits<float16_t>(min, max);
set_finfo_limits<float16_t>(min, max, eps);
} else if (dtype == bfloat16) {
set_finfo_limits<bfloat16_t>(min, max);
set_finfo_limits<bfloat16_t>(min, max, eps);
} else if (dtype == float64) {
set_finfo_limits<double>(min, max);
set_finfo_limits<double>(min, max, eps);
} else if (dtype == complex64) {
this->dtype = float32;
set_finfo_limits<float>(min, max);
set_finfo_limits<float>(min, max, eps);
}
}