Paritioner is working

src/include/migraphx/op/if_op.hpp

@@ -70,13 +70,13 @@ struct if_op
         module_ref mod = cond ? mods[0] : mods[1];
         std::unordered_map<std::string, argument> params;
 
-        std::set<std::string> pnames;
+        std::vector<std::string> pnames;
         for(const_module_ref smod : mods)
         {
             auto names = smod->get_parameter_names();
-            pnames.insert(names.begin(), names.end());
+            std::cout << smod->name() << std::endl;
+            pnames.insert(pnames.end(), names.begin(), names.end());
         }
-
         assert(pnames.size() < args.size());
         std::transform(pnames.begin(),
                        pnames.end(),

test/multi_target/multitarget_test.cpp

@@ -531,10 +531,13 @@ TEST_CASE(multitarget_compile_nested_if_then_else_partition)
             "target_" + std::to_string(tid) + "_" + std::to_string(counter_map[tid]++);
         auto* test_mod = prog.create_module(mod_name);
         std::vector<float> data(ds.elements(), -1);
-        auto l1   = test_mod->add_literal(migraphx::literal(ds, data));
-        auto ins1 = test_mod->add_instruction(migraphx::make_op("add"), inputs[0], l1);
-        auto ins2 = test_mod->add_instruction(migraphx::make_op("mul"), ins1, inputs[1]);
-        auto ins3 = test_mod->add_instruction(migraphx::make_op("sub"), ins2, inputs[2]);
+        auto l1               = test_mod->add_literal(migraphx::literal(ds, data));
+        auto test_mod_param_0 = test_mod->add_parameter(mod_name + "_param_0", ds);
+        auto test_mod_param_1 = test_mod->add_parameter(mod_name + "_param_1", ds);
+        auto test_mod_param_2 = test_mod->add_parameter(mod_name + "_param_2", ds);
+        auto ins1 = test_mod->add_instruction(migraphx::make_op("add"), test_mod_param_0, l1);
+        auto ins2 = test_mod->add_instruction(migraphx::make_op("mul"), ins1, test_mod_param_1);
+        auto ins3 = test_mod->add_instruction(migraphx::make_op("sub"), ins2, test_mod_param_2);
         test_mod->add_return({ins3});
         tass.insert(tass.begin(), std::make_pair(ins1, tid));
         tass.insert(tass.begin(), std::make_pair(ins2, tid));