AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Set flags when calling hipExtLaunchMultiKernelMultiDevice in hip-clang

Browse Source
This commit is contained in:
Wenkai Du
2020-05-08 15:53:51 +00:00
parent 33c23fdcda
commit 24ea2ef6dd
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/enqueue.cc
+4
View File
@@ -74,7 +74,11 @@ ncclResult_t ncclLaunchCooperativeKernelMultiDevice(hipLaunchParams *paramsList,
if (cgMode & 0x01) { if (cgMode & 0x01) {
CUDACHECK(hipExtLaunchMultiKernelMultiDevice(paramsList, numDevices, CUDACHECK(hipExtLaunchMultiKernelMultiDevice(paramsList, numDevices,
// These flags are to reduce the latency of using this API // These flags are to reduce the latency of using this API
#if __HIP__
hipCooperativeLaunchMultiDeviceNoPreSync|hipCooperativeLaunchMultiDeviceNoPostSync));
#else
0)); 0));
#endif
return ncclSuccess; return ncclSuccess;
} }
int savedDev; int savedDev;