From b71b9d2159f95cf8df2bbdec1e170ecb3df0724d Mon Sep 17 00:00:00 2001
From: Po Yen Chen <PoYen.Chen@amd.com>
Date: Tue, 9 Apr 2024 13:49:31 +0000
Subject: [PATCH] Add equal<float> & equal<double>

---
 include/ck_tile/core/numeric/math.hpp | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/include/ck_tile/core/numeric/math.hpp b/include/ck_tile/core/numeric/math.hpp
index 6bb6fc49f6..41ecdd2b75 100644
--- a/include/ck_tile/core/numeric/math.hpp
+++ b/include/ck_tile/core/numeric/math.hpp
@@ -343,6 +343,24 @@ struct equal<void, void>
 
 __host__ __device__ equal()->equal<void, void>;
 
+template <>
+struct equal<float, float>
+{
+    CK_TILE_HOST_DEVICE constexpr bool operator()(float lhs, float rhs) const
+    {
+        return bit_cast<uint32_t>(lhs) == bit_cast<uint32_t>(rhs);
+    }
+};
+
+template <>
+struct equal<double, double>
+{
+    CK_TILE_HOST_DEVICE constexpr bool operator()(double lhs, double rhs) const
+    {
+        return bit_cast<uint64_t>(lhs) == bit_cast<uint64_t>(rhs);
+    }
+};
+
 template <typename Left = void, typename Right = Left>
 struct less
 {