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

2020-05-12 23:53:19 +00:00
parent 66a9f11910
commit f1058b6353
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++;
    }