From ba65d4412e9a4d33cdbf464273f5ea35e312acdb Mon Sep 17 00:00:00 2001 From: Jason Henline Date: Wed, 31 Aug 2016 00:11:14 +0000 Subject: [PATCH] [StreamExecutor] Add Stream::blockHostUntilDone Summary: Add the type-safe wrapper to the platform-specific implementation. Reviewers: jlebar Subscribers: jprice, parallel_libs-commits Differential Revision: https://reviews.llvm.org/D24063 llvm-svn: 280182 --- parallel-libs/streamexecutor/include/streamexecutor/Stream.h | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/parallel-libs/streamexecutor/include/streamexecutor/Stream.h b/parallel-libs/streamexecutor/include/streamexecutor/Stream.h index 2937c58..d1c82f9 100644 --- a/parallel-libs/streamexecutor/include/streamexecutor/Stream.h +++ b/parallel-libs/streamexecutor/include/streamexecutor/Stream.h @@ -78,7 +78,16 @@ public: return make_error(*ErrorMessage); else return Error::success(); - }; + } + + // Blocks the calling host thread until all work enqueued on this Stream + // completes. + // + // Returns the result of getStatus() after the Stream work completes. + Error blockHostUntilDone() { + setError(PDevice->blockHostUntilDone(ThePlatformStream.get())); + return getStatus(); + } /// Entrains onto the stream of operations a kernel launch with the given /// arguments. -- 2.7.4