add back per-thread

2023-10-24 15:17:19 -07:00
parent 3752b06550
commit cd28b9f997
1 changed files with 7 additions and 1 deletions
--- a/src/common/device_helpers.hip.h
+++ b/src/common/device_helpers.hip.h
@@ -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_;