[llvm-branch-commits] [llvm] [Frontend][OpenMP] Add functions for checking construct type (PR #87258)

[Sergio Afonso via llvm-branch-commits]

================
@@ -37,3 +37,29 @@ TEST(Composition, GetCompoundConstruct) {
   Directive C6 = getCompoundConstruct({OMPD_parallel_for, OMPD_simd});
   ASSERT_EQ(C6, OMPD_parallel_for_simd);
 }
+
+TEST(Composition, IsLeafConstruct) {
+  ASSERT_TRUE(isLeafConstruct(OMPD_loop));
+  ASSERT_TRUE(isLeafConstruct(OMPD_teams));
+  ASSERT_FALSE(isLeafConstruct(OMPD_for_simd));
----------------
skatrak wrote:

Nit: Can you add one test case for a combined construct? e.g. `ASSERT_FALSE(isLeafConstruct(OMPD_parallel_for));`.

https://github.com/llvm/llvm-project/pull/87258