From f1058b6353d44bbc951b21cbca1c41b689ca851d Mon Sep 17 00:00:00 2001 From: Wenkai Du Date: Tue, 12 May 2020 23:53:19 +0000 Subject: [PATCH] rccl-prim-test: add flags when calling hipExtLaunchMultiKernelMultiDevice in hip-clang --- tools/rccl-prim-test/rccl_prim_test.cpp | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tools/rccl-prim-test/rccl_prim_test.cpp b/tools/rccl-prim-test/rccl_prim_test.cpp index d328dbf5c8..0a0620308f 100644 --- a/tools/rccl-prim-test/rccl_prim_test.cpp +++ b/tools/rccl-prim-test/rccl_prim_test.cpp @@ -556,7 +556,8 @@ int main(int argc,char* argv[]) launchParamsList[i].stream = stream[i]; launchParamsList[i].args = args + i*3; } - hipExtLaunchMultiKernelMultiDevice(launchParamsList, nGpu, 0); + hipExtLaunchMultiKernelMultiDevice(launchParamsList, nGpu, + hipCooperativeLaunchMultiDeviceNoPreSync|hipCooperativeLaunchMultiDeviceNoPostSync); opCount++; } @@ -582,7 +583,8 @@ int main(int argc,char* argv[]) launchParamsList[i].stream = stream[i]; launchParamsList[i].args = args + i*3; } - hipExtLaunchMultiKernelMultiDevice(launchParamsList, nGpu, 0); + hipExtLaunchMultiKernelMultiDevice(launchParamsList, nGpu, + hipCooperativeLaunchMultiDeviceNoPreSync|hipCooperativeLaunchMultiDeviceNoPostSync); opCount++; }