use __HIPCC__ for device code

2024-01-24 11:57:58 -08:00
parent fd3ad29dc4
commit 74677e4e9d
22 changed files with 69 additions and 68 deletions
--- a/tests/cpp/metric/test_rank_metric.cc
+++ b/tests/cpp/metric/test_rank_metric.cc
@@ -20,7 +20,7 @@
 namespace xgboost {
 namespace metric {

-#if !defined(__CUDACC__) && !defined(__HIP_PLATFORM_AMD__)
+#if !defined(__CUDACC__) && !defined(__HIPCC__)
 TEST(Metric, AMS) {
  auto ctx = MakeCUDACtx(GPUIDX);
  EXPECT_ANY_THROW(Metric::Create("ams", &ctx));