Fix rocshmem_team_split_strided API

Этот коммит содержится в:
avinashkethineedi
2024-12-21 18:16:42 +00:00
родитель 7fd66663c5
Коммит cb8b9094b4
2 изменённых файлов: 3 добавлений и 2 удалений
+2 -1
Просмотреть файл
@@ -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) {
+1 -1
Просмотреть файл
@@ -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) ||