[Mlir-commits] [mlir] 05d85ec - [OpenMP][MLIR] Add test missed by #102341

[Akash Banerjee]

Author: Akash Banerjee
Date: 2024-08-12T16:35:24+01:00
New Revision: 05d85ecad707573cde0258cbab579ed764c61e51

URL: https://github.com/llvm/llvm-project/commit/05d85ecad707573cde0258cbab579ed764c61e51
DIFF: https://github.com/llvm/llvm-project/commit/05d85ecad707573cde0258cbab579ed764c61e51.diff

LOG: [OpenMP][MLIR] Add test missed by #102341

Add small test that I missed adding to #102341.

Added: 
    

Modified: 
    mlir/test/Dialect/OpenMP/ops.mlir

Removed: 
    


################################################################################
diff  --git a/mlir/test/Dialect/OpenMP/ops.mlir b/mlir/test/Dialect/OpenMP/ops.mlir
index 988e0096850bd..9ac97e069addd 100644
--- a/mlir/test/Dialect/OpenMP/ops.mlir
+++ b/mlir/test/Dialect/OpenMP/ops.mlir
@@ -116,6 +116,27 @@ func.func @omp_parallel(%data_var : memref<i32>, %if_cond : i1, %num_threads : i
     omp.terminator
   } {omp.composite}
 
+  // CHECK: omp.distribute
+  omp.distribute {
+    // CHECK-NEXT: omp.parallel
+    omp.parallel {
+      // CHECK-NEXT: omp.wsloop
+      omp.wsloop {
+        // CHECK-NEXT: omp.simd
+        omp.simd{
+          // CHECK-NEXT: omp.loop_nest
+          omp.loop_nest (%iv) : index = (%idx) to (%idx) step (%idx) {
+            omp.yield
+          }
+          omp.terminator
+        } {omp.composite}
+        omp.terminator
+      } {omp.composite}
+      omp.terminator
+    } {omp.composite}
+    omp.terminator
+  } {omp.composite}
+
   return
 }