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

add back per-thread

Browse Source
This commit is contained in:
Hui Liu 2023-10-24 15:17:19 -07:00
parent 3752b06550
commit cd28b9f997
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/common/device_helpers.hip.h
View File

@ -1099,7 +1099,13 @@ inline void CUDAEvent::Record(CUDAStreamView stream) { // NOLINT
dh::safe_cuda(hipEventRecord(event_, hipStream_t{stream}));
}
inline CUDAStreamView DefaultStream() { return CUDAStreamView{hipStreamDefault}; }
inline CUDAStreamView DefaultStream() {
#ifdef HIP_API_PER_THREAD_DEFAULT_STREAM
return CUDAStreamView{hipStreamPerThread};
#else
return CUDAStreamView{hipStreamDefault};
#endif
}
class CUDAStream {
hipStream_t stream_;