Sun, Peng
9b6b29a535
Change-Id: I87ff9b3f1203d0909f998c96c839f7b321fc3f09
[ROCm/hip commit: 80a99350a3]
39 lines
1.2 KiB
C++
39 lines
1.2 KiB
C++
#include "hip/hcc_detail/grid_launch_GGL.hpp"
|
|
|
|
// Internal header, do not percolate upwards.
|
|
#include "hip_hcc_internal.h"
|
|
#include "hc.hpp"
|
|
#include "trace_helper.h"
|
|
|
|
namespace hip_impl
|
|
{
|
|
hc::accelerator_view lock_stream_hip_(
|
|
hipStream_t& stream, void*& locked_stream)
|
|
{ // This allocated but does not take ownership of locked_stream. If it is
|
|
// not deleted elsewhere it will leak.
|
|
using L = decltype(stream->lockopen_preKernelCommand());
|
|
|
|
HIP_INIT();
|
|
|
|
stream = ihipSyncAndResolveStream(stream);
|
|
locked_stream = new L{stream->lockopen_preKernelCommand()};
|
|
return (*static_cast<L*>(locked_stream))->_av;
|
|
}
|
|
|
|
void unlock_stream_hip_(
|
|
hipStream_t stream,
|
|
void* locked_stream,
|
|
const char* kernel_name,
|
|
hc::accelerator_view* acc_v)
|
|
{ // Precondition: acc_v is the accelerator_view associated with stream
|
|
// which is guarded by locked_stream;
|
|
// locked_stream is deletable.
|
|
using L = decltype(stream->lockopen_preKernelCommand());
|
|
|
|
stream->lockclose_postKernelCommand(kernel_name, acc_v);
|
|
|
|
delete static_cast<L*>(locked_stream);
|
|
locked_stream = nullptr;
|
|
}
|
|
}
|