Merge 07c3a4a into 032cf20

Author: AlexanderKalistratov

dpnp/backend/extensions/common/ext/common.hpp

@@ -70,20 +70,6 @@ struct AtomicOp
     }
 };
 
-template <typename T>
-struct Less
-{
-    bool operator()(const T &lhs, const T &rhs) const
-    {
-        if constexpr (type_utils::is_complex_v<T>) {
-            return dpctl::tensor::math_utils::less_complex(lhs, rhs);
-        }
-        else {
-            return std::less{}(lhs, rhs);
-        }
-    }
-};
-
 template <typename T>
 struct IsNan
 {
@@ -106,6 +92,21 @@ struct IsNan
     }
 };
 
+template <typename T>
+struct Less
+{
+    bool operator()(const T &lhs, const T &rhs) const
+    {
+        if constexpr (type_utils::is_complex_v<T>) {
+            return IsNan<T>::isnan(rhs) ||
+                   dpctl::tensor::math_utils::less_complex(lhs, rhs);
+        }
+        else {
+            return IsNan<T>::isnan(rhs) || std::less{}(lhs, rhs);
+        }
+    }
+};
+
 template <typename T, bool hasValueType>
 struct value_type_of_impl;
 

dpnp/backend/extensions/statistics/CMakeLists.txt

@@ -30,6 +30,7 @@ set(_module_src
     ${CMAKE_CURRENT_SOURCE_DIR}/histogram.cpp
     ${CMAKE_CURRENT_SOURCE_DIR}/histogramdd.cpp
     ${CMAKE_CURRENT_SOURCE_DIR}/histogram_common.cpp
+    ${CMAKE_CURRENT_SOURCE_DIR}/kth_element1d.cpp
     ${CMAKE_CURRENT_SOURCE_DIR}/sliding_dot_product1d.cpp
     ${CMAKE_CURRENT_SOURCE_DIR}/sliding_window1d.cpp
     ${CMAKE_CURRENT_SOURCE_DIR}/statistics_py.cpp