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

Add isnan (#423)

Browse Source
This commit is contained in:
Ayush Shridhar
2024-01-12 14:16:48 -05:00
committed by GitHub
parent 29081204d1
commit 1416e7b664
7 changed files with 72 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
tests/ops_tests.cpp
View File

@@ -479,6 +479,32 @@ TEST_CASE("test comparison ops") {
}
}
TEST_CASE("test is nan") {
array x(1.0f);
CHECK_FALSE(isnan(x).item<bool>());
array y(NAN);
CHECK(isnan(y).item<bool>());
array z = identity(7);
CHECK_FALSE(all(isnan(z)).item<bool>());
array w = array({1.0f, NAN, 2.0f});
CHECK_FALSE(all(isnan(w)).item<bool>());
array a(1.0f, bfloat16);
CHECK_FALSE(isnan(a).item<bool>());
array b(1.0f, float16);
CHECK_FALSE(isnan(b).item<bool>());
array c(NAN, bfloat16);
CHECK(isnan(c).item<bool>());
array d(NAN, float16);
CHECK(isnan(d).item<bool>());
}
TEST_CASE("test all close") {
array x(1.0f);
array y(1.0f);