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/src/data/ellpack_page.cu
+++ b/src/data/ellpack_page.cu
@@ -286,7 +286,7 @@ void CopyDataToEllpack(const AdapterBatchT& batch, common::Span<FeatureType cons
                         cub::NullType(), batch.Size(), nullptr, false);
 #endif

-#elif defined (__HIP_PLATFORM_AMD__)
+#elif defined (__HIPCC__)

  rocprim::inclusive_scan(nullptr, temp_storage_bytes, key_value_index_iter, out, batch.Size(), TupleScanOp<Tuple>());