Merge pull request #19 from avinashkethineedi/teams_split_API

[ROCm/rocshmem commit: c5902afe28]
This commit is contained in:
Avinash Kethineedi
2024-12-23 20:42:09 +05:30
zatwierdzone przez GitHub
2 zmienionych plików z 3 dodań i 2 usunięć
+2 -1
Wyświetl plik
@@ -229,7 +229,8 @@ __host__ int rocshmem_team_split_strided(
auto num_user_teams{backend->team_tracker.get_num_user_teams()};
auto max_num_teams{backend->team_tracker.get_max_num_teams()};
if (num_user_teams >= max_num_teams - 1) {
abort();
/* Exceeded maximum number of teams */
return -1;
}
if (parent_team == ROCSHMEM_TEAM_INVALID) {
@@ -570,7 +570,7 @@ bool Tester::peLaunchesKernel() {
* Some test types are active on both sides.
*/
is_launcher = is_launcher || (_type == TeamReductionTestType) ||
(_type == TeamBroadcastTestType) ||
(_type == TeamBroadcastTestType) || (_type == TeamCtxInfraTestType) ||
(_type == AllToAllTestType) || (_type == FCollectTestType) ||
(_type == PingPongTestType) || (_type == BarrierAllTestType) ||
(_type == SyncTestType) || (_type == SyncAllTestType) ||