Fix rocshmem_team_split_strided API

This commit is contained in:
avinashkethineedi
2024-12-21 18:16:42 +00:00
parent 7fd66663c5
commit cb8b9094b4
2 ha cambiato i file con 3 aggiunte e 2 eliminazioni
+2 -1
Vedi File
@@ -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) {