zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-03 01:06:43 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Do not pass integers to isnan (#1664)

Browse Source
This commit is contained in:
Cheng 2024-12-08 11:26:23 +09:00 committed by GitHub
parent 9635cffdc8
commit 6ae5423b4a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mlx/backend/common/ops.h
View File

@ -500,8 +500,13 @@ struct Equal {
struct NaNEqual { struct NaNEqual {
template <typename T> template <typename T>
bool operator()(T x, T y) { bool operator()(T x, T y) {
if constexpr (std::is_integral_v<T>) {
// isnan always returns false for integers, and MSVC refuses to compile.
return x == y;
} else {
return x == y || (std::isnan(x) && std::isnan(y)); return x == y || (std::isnan(x) && std::isnan(y));
} }
}
}; };
struct Greater { struct Greater {