diff options
| author | Tim Rogers <[email protected]> | 2018-11-16 13:24:28 -0500 |
|---|---|---|
| committer | GitHub <[email protected]> | 2018-11-16 13:24:28 -0500 |
| commit | 52df8562368a64a578ed1278680621a36ba98a64 (patch) | |
| tree | 3949a1f5600e5e42f7c3365909fb1c6a39ca7187 /src/stream_manager.h | |
| parent | 0265d747b06c18d0a1ee00fb1641032201425c97 (diff) | |
| parent | f9a626f93d891b553c11bb86172b41ccc1deb569 (diff) | |
Merge pull request #81 from deval281shah/dev
Merging changes for cuDNN, Pytorch and checkpoint support
Diffstat (limited to 'src/stream_manager.h')
| -rw-r--r-- | src/stream_manager.h | 12 |
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: |