Add more tests

test/schedule_test.cpp

@@ -274,6 +274,52 @@ TEST_CASE(zero_merge2)
     EXPECT(check_conflicts(p, onep1, onep2));
 }
 
+TEST_CASE(zero_merge3)
+{
+    schedule_target t{};
+    migraphx::program p;
+    auto one    = p.add_literal(1);
+    auto onep1  = p.add_instruction(unary_op{}, one);
+    auto onep2  = p.add_instruction(unary_op{}, one);
+    auto id = p.add_instruction(migraphx::op::identity{}, onep1, onep2);
+    auto final = p.add_instruction(unary_op{}, id);
+    p.compile(t);
+    EXPECT(not t.has_stream(one));
+    EXPECT(t.get_stream(onep1) != t.get_stream(onep2));
+    // No stream assignment
+    EXPECT(not t.has_stream(id));
+    // There is no wait
+    EXPECT(get_wait_for(id).empty());
+    // Stream assignment for final op
+    EXPECT(t.get_stream(final) == 0);
+    EXPECT(get_wait_for(final) == get_wait_for(t.get_stream(final), {t.get_stream(onep1), t.get_stream(onep2)}));
+    EXPECT(check_conflicts(p, onep1, onep2));
+}
+
+TEST_CASE(zero_merge4)
+{
+    schedule_target t{};
+    migraphx::program p;
+    auto one    = p.add_literal(1);
+    auto onep1  = p.add_instruction(unary_op{}, one);
+    auto onep2  = p.add_instruction(unary_op{}, one);
+    auto id = p.add_instruction(migraphx::op::identity{},
+                                    p.add_instruction(migraphx::op::identity{}, onep1),
+                                    p.add_instruction(migraphx::op::identity{}, onep2));
+    auto final = p.add_instruction(unary_op{}, id);
+    p.compile(t);
+    EXPECT(not t.has_stream(one));
+    EXPECT(t.get_stream(onep1) != t.get_stream(onep2));
+    // No stream assignment
+    EXPECT(not t.has_stream(id));
+    // There is no wait
+    EXPECT(get_wait_for(id).empty());
+    // Stream assignment for final op
+    EXPECT(t.get_stream(final) == 0);
+    EXPECT(get_wait_for(final) == get_wait_for(t.get_stream(final), {t.get_stream(onep1), t.get_stream(onep2)}));
+    EXPECT(check_conflicts(p, onep1, onep2));
+}
+
 TEST_CASE(double_entry)
 {
     schedule_target t{};