[flang] [llvm] [mlir] [flang][OpenMP][MLIR] Add MLIR op for loop directive (PR #113911)
Sergio Afonso via llvm-commits
llvm-commits at lists.llvm.org
Thu Oct 31 08:26:00 PDT 2024
================
@@ -2749,3 +2749,43 @@ func.func @omp_target_private(%map1: memref<?xi32>, %map2: memref<?xi32>, %priv_
return
}
+
+// CHECK-LABEL: omp_loop
+func.func @omp_loop(%lb : index, %ub : index, %step : index) {
+ // CHECK: omp.loop {
+ omp.loop {
+ // CHECK: omp.loop_nest {{.*}} {
+ omp.loop_nest (%iv) : index = (%lb) to (%ub) step (%step) {
+ // CHECK: omp.yield
+ omp.yield
+ // CHECK: }
+ }
+ // CHECK: }
+ }
+
+ // CHECK: omp.loop bind(teams) {
+ omp.loop bind(teams) {
+ omp.loop_nest (%iv) : index = (%lb) to (%ub) step (%step) {
+ omp.yield
+ }
+ // CHECK: }
----------------
skatrak wrote:
Nit: We probably don't need to check this here, same below.
```suggestion
```
https://github.com/llvm/llvm-project/pull/113911
More information about the llvm-commits
mailing list