Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

restore stream logic

Browse Source
This commit is contained in:
Hendrik Groove 2024-10-21 01:39:43 +02:00
parent b3ee7a59c7
commit 1b6c6baf76
1 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/data/array_interface.cu
View File

@ -11,6 +11,33 @@
namespace xgboost {
void ArrayInterfaceHandler::SyncCudaStream(std::int64_t stream) {
#if defined(XGBOOST_USE_HIP)
if (stream == 1 || stream == 2) {
// Default streams, no synchronization needed
return;
}
hipEvent_t event;
hipError_t err = hipEventCreate(&event);
if (err != hipSuccess) {
LOG(WARNING) << "Failed to create HIP event: " << hipGetErrorString(err);
return;
}
err = hipEventRecord(event, reinterpret_cast<hipStream_t>(stream));
if (err != hipSuccess) {
LOG(WARNING) << "Failed to record HIP event: " << hipGetErrorString(err);
hipEventDestroy(event);
return;
}
err = hipStreamWaitEvent(hipStreamPerThread, event, 0);
if (err != hipSuccess) {
LOG(WARNING) << "Failed to wait for HIP event: " << hipGetErrorString(err);
}
hipEventDestroy(event);
#else
switch (stream) {
case 0:
/**
@ -31,6 +58,7 @@ void ArrayInterfaceHandler::SyncCudaStream(std::int64_t stream) {
dh::DefaultStream().Wait(e);
}
}
#endif
}
bool ArrayInterfaceHandler::IsCudaPtr(void const* ptr) {