fix random

This commit is contained in:
Hui Liu
2024-01-21 12:48:41 -08:00
parent 9759e28e6a
commit 1e0ccf7b87
7 changed files with 19 additions and 5 deletions

View File

@@ -13,7 +13,6 @@
#include <thrust/system_error.h> // for system_error
#elif defined(XGBOOST_USE_RCCL)
#include "../common/cuda_to_hip.h"
#include "../common/device_helpers.hip.h"
#include <hip/hip_runtime_api.h> // for cudaPeekAtLastError
#include <dlfcn.h> // for dlclose, dlsym, dlopen
#include <rccl.h>

View File

@@ -9,7 +9,15 @@
#include <nccl.h>
#elif defined(XGBOOST_USE_RCCL)
#include "../common/cuda_to_hip.h"
#include "../common/device_helpers.cuh"
#ifndef __HIP_PLATFORM_AMD__
#define __HIP_PLATFORM_AMD__
#endif
#ifndef THRUST_DEVICE_SYSTEM
#define THRUST_DEVICE_SYSTEM THRUST_DEVICE_SYSTEM_HIP
#endif
#include <hip/hip_runtime_api.h>
#include <rccl.h>
#endif