Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix macro

Browse Source
This commit is contained in:
amdsc21 2023-03-25 05:17:39 +01:00
parent 22525c002a
commit e74b3bbf3c
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/metric/rank_metric.cu
View File

@ -24,6 +24,12 @@
#include "xgboost/logging.h" // for CHECK #include "xgboost/logging.h" // for CHECK
#include "xgboost/metric.h" #include "xgboost/metric.h"
#if defined(XGBOOST_USE_HIP)
#include <hipcub/hipcub.hpp>
namespace cub = hipcub;
#endif
namespace xgboost::metric { namespace xgboost::metric {
// tag the this file, used by force static link later. // tag the this file, used by force static link later.
DMLC_REGISTRY_FILE_TAG(rank_metric_gpu); DMLC_REGISTRY_FILE_TAG(rank_metric_gpu);