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++; }