Implement hipStreamAddCallback

Change-Id: Ib851e4d86ba9c8406ca37b88162ea483ccbc9d36
This commit is contained in:
Maneesh Gupta
2017-12-19 16:06:14 +05:30
والد 1026e58ad9
کامیت cebb070d30
4فایلهای تغییر یافته به همراه114 افزوده شده و 38 حذف شده
+20
مشاهده پرونده
@@ -622,6 +622,24 @@ private: // Data
};
//----
// Internal structure for stream callback handler
class ihipStreamCallback_t {
public:
ihipStreamCallback_t(hipStream_t stream, hipStreamCallback_t callback, void *userData) :
_stream(stream),
_callback(callback),
_userData(userData)
{
_ready = false;
};
hipStream_t _stream;
hipStreamCallback_t _callback;
void* _userData;
std::mutex _mtx;
bool _ready;
};
//----
// Internal event structure:
@@ -931,6 +949,8 @@ ihipCtx_t * ihipGetPrimaryCtx(unsigned deviceIndex);
hipStream_t ihipSyncAndResolveStream(hipStream_t);
hipError_t ihipStreamSynchronize(hipStream_t stream);
void ihipStreamCallbackHandler(ihipStreamCallback_t *cb);
// Stream printf functions:
inline std::ostream& operator<<(std::ostream& os, const ihipStream_t& s)