use __HIPCC__ for device code

2024-01-24 12:32:51 -08:00
parent 7dc152450e
commit dc7ee041cc
22 changed files with 73 additions and 72 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));