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

try to change stream

Browse Source
This commit is contained in:
Hendrik Groove 2024-10-21 00:25:18 +02:00
parent 6fcffef7dc
commit 0d600b4535
1 changed files with 0 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
src/data/array_interface.cu
View File

@ -11,33 +11,6 @@
namespace xgboost { namespace xgboost {
void ArrayInterfaceHandler::SyncCudaStream(std::int64_t stream) { 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) { switch (stream) {
case 0: case 0:
/** /**
@ -58,7 +31,6 @@ void ArrayInterfaceHandler::SyncCudaStream(std::int64_t stream) {
dh::DefaultStream().Wait(e); dh::DefaultStream().Wait(e);
} }
} }
#endif
} }
bool ArrayInterfaceHandler::IsCudaPtr(void const* ptr) { bool ArrayInterfaceHandler::IsCudaPtr(void const* ptr) {