Merge pull request #204 from wenkaidu/launch_flags

Set flags when calling hipExtLaunchMultiKernelMultiDevice in hip-clang

[ROCm/rccl commit: 94d16c0f0a]
Этот коммит содержится в:
Wenkai Du
2020-05-08 11:20:50 -07:00
коммит произвёл GitHub
родитель 65ad48404b 069322d05a
Коммит 0d642448d7
+4
Просмотреть файл
@@ -74,7 +74,11 @@ ncclResult_t ncclLaunchCooperativeKernelMultiDevice(hipLaunchParams *paramsList,
if (cgMode & 0x01) {
CUDACHECK(hipExtLaunchMultiKernelMultiDevice(paramsList, numDevices,
// These flags are to reduce the latency of using this API
#if __HIP__
hipCooperativeLaunchMultiDeviceNoPreSync|hipCooperativeLaunchMultiDeviceNoPostSync));
#else
0));
#endif
return ncclSuccess;
}
int savedDev;