rccl-prim-test: add flags when calling hipExtLaunchMultiKernelMultiDevice in hip-clang

[ROCm/rccl commit: f1058b6353]
2020-05-12 23:53:19 +00:00
parent 5b01a901b7
commit ced9958094
1 changed files with 4 additions and 2 deletions
@@ -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++;
    }