From b5a8fc30d662410bebd868126c2c9539234f3eea Mon Sep 17 00:00:00 2001
From: Matthew Michel <matthew.michel@intel.com>
Date: Tue, 4 Jun 2024 09:05:42 -0500
Subject: [PATCH] Add dpct::argmax and dpct::argmin to functional.h

Signed-off-by: Matthew Michel <matthew.michel@intel.com>
---
 .../include/dpct/dpl_extras/functional.h      | 34 +++++++++++++++++++
 1 file changed, 34 insertions(+)

diff --git a/clang/runtime/dpct-rt/include/dpct/dpl_extras/functional.h b/clang/runtime/dpct-rt/include/dpct/dpl_extras/functional.h
index bab82814c210..72799cc2bdad 100644
--- a/clang/runtime/dpct-rt/include/dpct/dpl_extras/functional.h
+++ b/clang/runtime/dpct-rt/include/dpct/dpl_extras/functional.h
@@ -58,6 +58,40 @@ template <typename _Op> struct mark_functor_const {
   }
 };
 
+// Forward declare key_value_pair to avoid creating cyclic dependency between
+// iterators.h and functional.h.
+template <typename _KeyTp, typename _ValueTp> class key_value_pair;
+
+// Returns the smaller of two key_value_pair objects based on their value
+// member. If value elements compare equal, then the pair with the lower key is
+// returned.
+struct argmin {
+  template <typename _ValueTp, typename _KeyTp>
+  key_value_pair<_KeyTp, _ValueTp>
+  operator()(const key_value_pair<_KeyTp, _ValueTp> &lhs,
+             const key_value_pair<_KeyTp, _ValueTp> &rhs) const {
+    return (lhs.value < rhs.value) ||
+                   (lhs.value == rhs.value && lhs.key < rhs.key)
+               ? lhs
+               : rhs;
+  }
+};
+
+// Returns the larger of two key_value_pair objects based on their value member.
+// If value elements compare equal, then the pair with the lower key is
+// returned.
+struct argmax {
+  template <typename _ValueTp, typename _KeyTp>
+  key_value_pair<_KeyTp, _ValueTp>
+  operator()(const key_value_pair<_KeyTp, _ValueTp> &lhs,
+             const key_value_pair<_KeyTp, _ValueTp> &rhs) const {
+    return (lhs.value > rhs.value) ||
+                   (lhs.value == rhs.value && lhs.key < rhs.key)
+               ? lhs
+               : rhs;
+  }
+};
+
 namespace internal {
 
 template <class _ExecPolicy, class _T>