From 32ae49ab929557dc826757e144725199a8b5325f Mon Sep 17 00:00:00 2001 From: Hui Liu <96135754+amdsc21@users.noreply.github.com> Date: Fri, 27 Oct 2023 13:00:49 -0700 Subject: [PATCH] temp hack for multi GPUs --- src/data/array_interface.cu | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/data/array_interface.cu b/src/data/array_interface.cu index 492c24200..b0004c300 100644 --- a/src/data/array_interface.cu +++ b/src/data/array_interface.cu @@ -20,7 +20,10 @@ void ArrayInterfaceHandler::SyncCudaStream(std::int64_t stream) { * case where 0 might be given should either use None, 1, or 2 instead for * clarity. */ + /* ignored for HIP */ +#if !defined(XGBOOST_USE_HIP) LOG(FATAL) << "Invalid stream ID in array interface: " << stream; +#endif case 1: // default legacy stream break;