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

sync up May15 2023

Browse Source
This commit is contained in:
amdsc21
2023-05-15 18:59:18 +02:00
parent b066accad6 7375bd058b
commit 8cad8c693c
37 changed files with 628 additions and 398 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -1273,14 +1273,12 @@ class CUDAStream {
hipStream_t stream_;
public:
CUDAStream() {
dh::safe_cuda(hipStreamCreateWithFlags(&stream_, hipStreamNonBlocking));
}
~CUDAStream() {
dh::safe_cuda(hipStreamDestroy(stream_));
}
CUDAStream() { dh::safe_cuda(hipStreamCreateWithFlags(&stream_, hipStreamNonBlocking)); }
~CUDAStream() { dh::safe_cuda(hipStreamDestroy(stream_)); }
[[nodiscard]] CUDAStreamView View() const { return CUDAStreamView{stream_}; }
[[nodiscard]] hipStream_t Handle() const { return stream_; }
CUDAStreamView View() const { return CUDAStreamView{stream_}; }
void Sync() { this->View().Sync(); }
};