This is an archive of the discontinued LLVM Phabricator instance.

Differential D142819

[OpenMP][NVPTX] Added `__tgt_rtl_launch_kernel` in old CUDA plugin
ClosedPublic

Authored by tianshilei1992 on Jan 28 2023, 3:40 PM.

Details

Reviewers
jdoerfert
ye-luo
jhuber6
Commits
rGad95b0e97750: [OpenMP][NVPTX] Added `__tgt_rtl_launch_kernel` in old CUDA plugin
Summary

Fix #60248.

Diff Detail

Event Timeline

tianshilei1992 created this revision.Jan 28 2023, 3:40 PM
Herald added a project: Restricted Project. · View Herald TranscriptJan 28 2023, 3:40 PM
Herald added subscribers: mattd, gchakrabarti, asavonic and 2 others. · View Herald Transcript
tianshilei1992 requested review of this revision.Jan 28 2023, 3:40 PM
Herald added a project: Restricted Project. · View Herald TranscriptJan 28 2023, 3:40 PM
Herald added subscribers: openmp-commits, sstefan1. · View Herald Transcript
Harbormaster completed remote builds in B210564: Diff 493036.Jan 28 2023, 3:43 PM
tianshilei1992 added a reviewer: jhuber6.Jan 28 2023, 3:47 PM
jhuber6 accepted this revision.Jan 28 2023, 3:48 PM
This revision is now accepted and ready to land.Jan 28 2023, 3:48 PM
Closed by commit rGad95b0e97750: [OpenMP][NVPTX] Added `__tgt_rtl_launch_kernel` in old CUDA plugin (authored by tianshilei1992). · Explain WhyJan 28 2023, 3:56 PM
This revision was automatically updated to reflect the committed changes.
tianshilei1992 added a commit: rGad95b0e97750: [OpenMP][NVPTX] Added `__tgt_rtl_launch_kernel` in old CUDA plugin.

Revision Contents

PathSize
openmp/
libomptarget/
plugins/
cuda/
src/
15 lines

Diff 493036

openmp/libomptarget/plugins/cuda/src/rtl.cpp

Show First 20 Lines • Show All 1,895 Lines • ▼ Show 20 Linesint32_t __tgt_rtl_init_device_info(int32_t DeviceId,
assert(DeviceInfoPtr && "device_info_ptr is nullptr"); assert(DeviceInfoPtr && "device_info_ptr is nullptr");
if (DeviceRTL.setContext(DeviceId) != OFFLOAD_SUCCESS) if (DeviceRTL.setContext(DeviceId) != OFFLOAD_SUCCESS)
return OFFLOAD_FAIL; return OFFLOAD_FAIL;
return DeviceRTL.initDeviceInfo(DeviceId, DeviceInfoPtr, ErrStr); return DeviceRTL.initDeviceInfo(DeviceId, DeviceInfoPtr, ErrStr);
} }
int32_t __tgt_rtl_launch_kernel(int32_t DeviceId, void *TgtEntryPtr,
void **TgtArgs, ptrdiff_t *TgtOffsets,
KernelArgsTy *KernelArgs,
__tgt_async_info *AsyncInfo) {
assert(DeviceRTL.isValidDeviceId(DeviceId) && "device_id is invalid");
if (DeviceRTL.setContext(DeviceId) != OFFLOAD_SUCCESS)
return OFFLOAD_FAIL;
return DeviceRTL.runTargetTeamRegion(
DeviceId, TgtEntryPtr, TgtArgs, TgtOffsets, KernelArgs->NumArgs,
KernelArgs->NumTeams[0], KernelArgs->ThreadLimit[0],
KernelArgs->Tripcount, AsyncInfo);
}
#ifdef __cplusplus #ifdef __cplusplus
} }
#endif #endif