AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #207 from wenkaidu/hip-clang

Browse Source 
rccl-prim-test: add flags when calling hipExtLaunchMultiKernelMultiDe…

[ROCm/rccl commit: ca4987e5fb]
This commit is contained in:
Wenkai Du
2020-05-14 18:31:56 -07:00
committed by GitHub
parent 5b01a901b7 ced9958094
commit da05861904
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/tools/rccl-prim-test/rccl_prim_test.cpp
+4 -2
View File
@@ -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++;
}