procyberian/gcc
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

libgomp/plugin/plugin-nvptx.c: Fix device used for stream creation

Browse source 
libgomp/ChangeLog:

	* plugin/plugin-nvptx.c (GOMP_OFFLOAD_interop): Set context for
	stream creation to use the specified device.
This commit is contained in:
Tobias Burnus 2025-03-24 16:08:20 +01:00
parent c6e7d9ca72
commit 1c5a375c21
1 changed files with 16 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
libgomp/plugin/plugin-nvptx.c
View file

@ -2483,12 +2483,26 @@ GOMP_OFFLOAD_interop (struct interop_obj_t *obj, int ord,
break;
}
obj->device_data = ptx_devices[ord];
struct ptx_device *ptx_dev = obj->device_data = ptx_devices[ord];
if (targetsync)
{
CUstream stream = NULL;
CUDA_CALL_ASSERT (cuStreamCreate, &stream, CU_STREAM_DEFAULT);
CUdevice cur_ctx_dev;
CUresult res = CUDA_CALL_NOCHECK (cuCtxGetDevice, &cur_ctx_dev);
if (res != CUDA_SUCCESS && res != CUDA_ERROR_INVALID_CONTEXT)
GOMP_PLUGIN_fatal ("cuCtxGetDevice error: %s", cuda_error (res));
if (res != CUDA_ERROR_INVALID_CONTEXT && ptx_dev->dev == cur_ctx_dev)
CUDA_CALL_ASSERT (cuStreamCreate, &stream, CU_STREAM_DEFAULT);
else
{
CUcontext old_ctx;
assert (ptx_dev->ctx);
CUDA_CALL_ASSERT (cuCtxPushCurrent, ptx_dev->ctx);
CUDA_CALL_ASSERT (cuStreamCreate, &stream, CU_STREAM_DEFAULT);
if (res != CUDA_ERROR_INVALID_CONTEXT)
CUDA_CALL_ASSERT (cuCtxPopCurrent, &old_ctx);
}
obj->stream = stream;
}
}