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

Cette révision appartient à :
Wenkai Du a révisé 2020-05-12 23:54:07 +00:00
1 Parent 66a9f11910
révision f1058b6353
1 fichier modifié
+4 -2
+4 -2
Voir le fichier
@@ -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++;
}