From 3ee3bea683ca69fc90980512326b76c18990b186 Mon Sep 17 00:00:00 2001
From: amdsc21 <96135754+amdsc21@users.noreply.github.com>
Date: Sat, 25 Mar 2023 22:37:37 +0100
Subject: [PATCH] fix warp header

---
 src/predictor/gpu_predictor.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/predictor/gpu_predictor.cu b/src/predictor/gpu_predictor.cu
index 0ab587693..c5f80fa25 100644
--- a/src/predictor/gpu_predictor.cu
+++ b/src/predictor/gpu_predictor.cu
@@ -1,7 +1,6 @@
 /**
  * Copyright 2017-2023 by XGBoost Contributors
  */
-#include <amd_warp_primitives.h>
 #include <GPUTreeShap/gpu_treeshap.h>
 #include <thrust/copy.h>
 #include <thrust/device_ptr.h>
@@ -19,6 +18,7 @@
 #if defined(XGBOOST_USE_CUDA)
 #include "../common/device_helpers.cuh"
 #elif defined(XGBOOST_USE_HIP)
+#include <amd_warp_primitives.h>
 #include "../common/device_helpers.hip.h"
 #endif