[HIP] Refactor cooperative APIs

[ROCm/hip commit: 4844fbdf0a]
This commit is contained in:
Aryan Salmanpour
2020-03-06 18:30:12 -05:00
rodzic 624b6cd45e
commit c25dd0ca3d
2 zmienionych plików z 205 dodań i 129 usunięć
+3 -1
Wyświetl plik
@@ -1619,7 +1619,9 @@ void ihipPrintKernelLaunch(const char* kernelName, const grid_launch_parm* lp,
// Allows runtime to track some information about the stream.
hipStream_t ihipPreLaunchKernel(hipStream_t stream, dim3 grid, dim3 block, grid_launch_parm* lp,
const char* kernelNameStr, bool lockAcquired) {
stream = ihipSyncAndResolveStream(stream, lockAcquired);
if (stream == nullptr || stream != stream->getCtx()->_defaultStream){
stream = ihipSyncAndResolveStream(stream, lockAcquired);
}
lp->grid_dim.x = grid.x;
lp->grid_dim.y = grid.y;
lp->grid_dim.z = grid.z;