spirv-opt: Add more tests to control dependence (#4410)
In particular, adds tests for DoesBlockExist, ForEachBlockLabel, and
WhileEachBlockLabel.
diff --git a/test/opt/control_dependence.cpp b/test/opt/control_dependence.cpp
index 02793f1..94618d7 100644
--- a/test/opt/control_dependence.cpp
+++ b/test/opt/control_dependence.cpp
@@ -122,6 +122,43 @@
ControlDependenceAnalysis cdg;
cdg.ComputeControlDependenceGraph(cfg, pdom);
+ // Test DoesBlockExist.
+ for (uint32_t id = 10; id <= 19; id++) {
+ EXPECT_TRUE(cdg.DoesBlockExist(id));
+ }
+ EXPECT_TRUE(
+ cdg.DoesBlockExist(ControlDependenceAnalysis::kPseudoEntryBlock));
+ // Check blocks before/after valid range.
+ EXPECT_FALSE(cdg.DoesBlockExist(5));
+ EXPECT_FALSE(cdg.DoesBlockExist(25));
+ EXPECT_FALSE(cdg.DoesBlockExist(UINT32_MAX));
+
+ // Test ForEachBlockLabel.
+ std::set<uint32_t> block_labels;
+ cdg.ForEachBlockLabel([&block_labels](uint32_t id) {
+ bool inserted = block_labels.insert(id).second;
+ EXPECT_TRUE(inserted); // Should have no duplicates.
+ });
+ EXPECT_THAT(block_labels, testing::ElementsAre(0, 10, 11, 12, 13, 14, 15,
+ 16, 17, 18, 19));
+
+ {
+ // Test WhileEachBlockLabel.
+ uint32_t iters = 0;
+ EXPECT_TRUE(cdg.WhileEachBlockLabel([&iters](uint32_t) {
+ ++iters;
+ return true;
+ }));
+ EXPECT_EQ((uint32_t)block_labels.size(), iters);
+ iters = 0;
+ EXPECT_FALSE(cdg.WhileEachBlockLabel([&iters](uint32_t) {
+ ++iters;
+ return false;
+ }));
+ EXPECT_EQ(1, iters);
+ }
+
+ // Test IsDependent.
EXPECT_TRUE(cdg.IsDependent(12, 11));
EXPECT_TRUE(cdg.IsDependent(13, 11));
EXPECT_TRUE(cdg.IsDependent(15, 14));
@@ -137,6 +174,7 @@
EXPECT_FALSE(cdg.IsDependent(19, 14));
EXPECT_FALSE(cdg.IsDependent(12, 0));
+ // Test GetDependenceSources/Targets.
std::vector<ControlDependence> edges;
GatherEdges(cdg, edges);
EXPECT_THAT(edges,
