From e6295c645f85b7f0ffc64183a24feea3c0574c3c Mon Sep 17 00:00:00 2001
From: Shraiysh Vaishay <Shraiysh.Vaishay@amd.com>
Date: Mon, 2 May 2022 10:41:46 +0530
Subject: [PATCH] [flang] Added tests for taskwait and taskyield translation

Reviewed By: kiranchandramohan

Differential Revision: https://reviews.llvm.org/D124229

Co-authored-by: Sourabh Singh Tomar <SourabhSingh.Tomar@amd.com>
---
 flang/test/Lower/OpenMP/taskwait.f90  | 12 ++++++++++++
 flang/test/Lower/OpenMP/taskyield.f90 | 12 ++++++++++++
 2 files changed, 24 insertions(+)
 create mode 100644 flang/test/Lower/OpenMP/taskwait.f90
 create mode 100644 flang/test/Lower/OpenMP/taskyield.f90

diff --git a/flang/test/Lower/OpenMP/taskwait.f90 b/flang/test/Lower/OpenMP/taskwait.f90
new file mode 100644
index 000000000000..30e6422d3381
--- /dev/null
+++ b/flang/test/Lower/OpenMP/taskwait.f90
@@ -0,0 +1,12 @@
+!RUN: %flang_fc1 -emit-fir -fopenmp %s -o - | FileCheck %s --check-prefixes="FIRDialect,OMPDialect"
+!RUN: %flang_fc1 -emit-fir -fopenmp %s -o - | fir-opt --fir-to-llvm-ir | FileCheck %s --check-prefixes="OMPDialect"
+
+!FIRDialect-LABEL: @_QPomp_taskwait
+subroutine omp_taskwait
+  !OMPDialect: omp.taskwait
+  !$omp taskwait
+  !FIRDialect: fir.call @_QPfoo() : () -> ()
+  call foo()
+  !OMPDialect: omp.taskwait
+  !$omp taskwait
+end subroutine omp_taskwait
diff --git a/flang/test/Lower/OpenMP/taskyield.f90 b/flang/test/Lower/OpenMP/taskyield.f90
new file mode 100644
index 000000000000..07d0c124e16a
--- /dev/null
+++ b/flang/test/Lower/OpenMP/taskyield.f90
@@ -0,0 +1,12 @@
+!RUN: %flang_fc1 -emit-fir -fopenmp %s -o - | FileCheck %s --check-prefixes="FIRDialect,OMPDialect"
+!RUN: %flang_fc1 -emit-fir -fopenmp %s -o - | fir-opt --fir-to-llvm-ir | FileCheck %s --check-prefixes="OMPDialect"
+
+!FIRDialect-LABEL: @_QPomp_taskyield
+subroutine omp_taskyield
+  !OMPDialect: omp.taskyield
+  !$omp taskyield
+  !FIRDialect: fir.call @_QPfoo() : () -> ()
+  call foo()
+  !OMPDialect: omp.taskyield
+  !$omp taskyield
+end subroutine omp_taskyield