Add device argsort. (#6749)

This is part of https://github.com/dmlc/xgboost/pull/6747 .
2021-03-16 16:05:22 +08:00
parent 325bc93e16
commit 1a73a28511
3 changed files with 97 additions and 1 deletions
--- a/include/xgboost/span.h
+++ b/include/xgboost/span.h
@@ -95,7 +95,7 @@ namespace common {
 #define KERNEL_CHECK(cond)                                                     \
  (XGBOOST_EXPECT((cond), true)                                                \
       ? static_cast<void>(0)                                                  \
-       : __assert_fail(__ASSERT_STR_HELPER(e), __FILE__, __LINE__,             \
+       : __assert_fail(__ASSERT_STR_HELPER((cond)), __FILE__, __LINE__,        \
                       __PRETTY_FUNCTION__))

 #endif  // defined(_MSC_VER)