From e2e6b6e71f3fac708cde936c2bab944d92806d2b Mon Sep 17 00:00:00 2001
From: Hendrik Groove <hendrik.groove@dw.com>
Date: Mon, 21 Oct 2024 00:06:21 +0200
Subject: [PATCH] more logging

---
 src/common/device_helpers.hip.h | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/common/device_helpers.hip.h b/src/common/device_helpers.hip.h
index d54be2c16..5e5ceb13d 100644
--- a/src/common/device_helpers.hip.h
+++ b/src/common/device_helpers.hip.h
@@ -991,6 +991,11 @@ auto Reduce(Policy policy, InputIt first, InputIt second, Init init, Func reduce
 
     std::cerr << "Processing batch: offset=" << offset << ", batch_size=" << batch_size << std::endl;
 
+    hipError_t err = hipStreamQuery(stream);
+    if (err != hipSuccess && err != hipErrorNotReady) {
+        std::cerr << "Invalid stream: " << hipGetErrorString(err) << std::endl;
+    }
+
     try {
       auto ret = thrust::reduce(policy, begin_it, end_it, init, reduce_op);
       aggregate = reduce_op(aggregate, ret);