summaryrefslogtreecommitdiff
path: root/src/stream_manager.h
diff options
context:
space:
mode:
authorDeval Shah <[email protected]>2018-11-09 21:14:55 -0800
committerDeval Shah <[email protected]>2018-11-09 21:14:55 -0800
commit36496d4ab2ff1088979e0f3137c54592652c37da (patch)
treeca449a4dc68abd5355764ae1bc50ca6ea5888fec /src/stream_manager.h
parent0265d747b06c18d0a1ee00fb1641032201425c97 (diff)
parent0e8d4f190940681601bdf7f95a5686baa0b6463b (diff)
resolving merge conflict
Diffstat (limited to 'src/stream_manager.h')
-rw-r--r--src/stream_manager.h12
1 files changed, 11 insertions, 1 deletions
diff --git a/src/stream_manager.h b/src/stream_manager.h
index d3a804f..91d1b36 100644
--- a/src/stream_manager.h
+++ b/src/stream_manager.h
@@ -51,7 +51,8 @@ enum stream_operation_type {
stream_memcpy_to_symbol,
stream_memcpy_from_symbol,
stream_kernel_launch,
- stream_event
+ stream_event,
+ stream_wait_event
};
class stream_operation {
@@ -101,6 +102,14 @@ public:
m_stream=stream;
m_done=false;
}
+ stream_operation( struct CUstream_st *stream, class CUevent_st *e, unsigned int flags )
+ {
+ m_kernel=NULL;
+ m_type=stream_wait_event;
+ m_event=e;
+ m_stream=stream;
+ m_done=false;
+ }
stream_operation( const void *host_address_src, size_t device_address_dst, size_t cnt, struct CUstream_st *stream )
{
m_kernel=NULL;
@@ -246,6 +255,7 @@ public:
bool empty();
void print( FILE *fp);
void push( stream_operation op );
+ void pushCudaStreamWaitEventToAllStreams( CUevent_st *e, unsigned int flags );
bool operation(bool * sim);
void stop_all_running_kernels();
private: