TEST: fix one-sided tests always running (openucx#836)

* TEST: fix one-sided tests always running * TEST: sort list of collectives --------- Co-authored-by: ferrol aderholdt <[email protected]>
nsarka · Oct 24, 2023 · c95097e · c95097e
1 parent 64f3c5c
commit c95097e
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/test/mpi/main.cc b/test/mpi/main.cc
@@ -7,6 +7,7 @@
 
 #include <getopt.h>
 #include <sstream>
+#include <algorithm>
 #include <chrono>
 #include "test_mpi.h"
 
@@ -602,7 +603,17 @@ int main(int argc, char *argv[])
     }
 
     if (has_onesided) {
-        test->set_colls(onesided_colls);
+        std::vector<ucc_coll_type_t>           os_colls(onesided_colls.size());
+        std::vector<ucc_coll_type_t>::iterator it_start;
+
+        std::sort(colls.begin(), colls.end());
+        std::sort(onesided_colls.begin(), onesided_colls.end());
+
+        it_start = std::set_intersection(
+            colls.begin(), colls.end(), onesided_colls.begin(),
+            onesided_colls.end(), os_colls.begin());
+        os_colls.resize(it_start - os_colls.begin());
+        test->set_colls(os_colls);
         for (auto inpl : inplace) {
             for (auto pers : persistent) {
                 test->set_triggered(false);