zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-02 05:14:40 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add mx.finfo and use it when making causal mask (#1726)

Browse Source 
* finfo

* fixes

* docs
This commit is contained in:
Awni Hannun
2024-12-19 14:52:41 -08:00
committed by GitHub
parent e03f0372b1
commit c3628eea49
9 changed files with 154 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/utils_tests.cpp
View File

@@ -43,3 +43,15 @@ TEST_CASE("test normalize axis") {
CHECK_THROWS(normalize_axis_index(3, 3));
CHECK_THROWS(normalize_axis_index(-4, 3));
}
TEST_CASE("test finfo") {
CHECK_EQ(finfo(float32).dtype, float32);
CHECK_EQ(finfo(complex64).dtype, float32);
CHECK_EQ(finfo(float16).dtype, float16);
CHECK_EQ(finfo(float32).min, std::numeric_limits<float>::lowest());
CHECK_EQ(finfo(float32).max, std::numeric_limits<float>::max());
CHECK_EQ(finfo(complex64).min, std::numeric_limits<float>::lowest());
CHECK_EQ(finfo(complex64).max, std::numeric_limits<float>::max());
CHECK_EQ(finfo(float16).min, -65504);
CHECK_EQ(finfo(float16).max, 65504);
}