From 88f5ad524a152711b7345b6bee2e06c5af0e88bc Mon Sep 17 00:00:00 2001
From: Jakub Jelinek <jakub@redhat.com>
Date: Tue, 12 Oct 2021 09:32:28 +0200
Subject: [PATCH] openmp: Add testsuite coverage for
 omp_{get_max,set_num}_threads and omp_{s,g}et_teams_thread_limit

This adds (C/C++ only) testsuite coverage for these new OpenMP 5.1 APIs.

2021-10-12  Jakub Jelinek  <jakub@redhat.com>

	* testsuite/libgomp.c-c++-common/icv-3.c: New test.
	* testsuite/libgomp.c-c++-common/icv-4.c: New test.
---
 .../testsuite/libgomp.c-c++-common/icv-3.c    | 54 +++++++++++++++++++
 .../testsuite/libgomp.c-c++-common/icv-4.c    | 40 ++++++++++++++
 2 files changed, 94 insertions(+)
 create mode 100644 libgomp/testsuite/libgomp.c-c++-common/icv-3.c
 create mode 100644 libgomp/testsuite/libgomp.c-c++-common/icv-4.c

diff --git a/libgomp/testsuite/libgomp.c-c++-common/icv-3.c b/libgomp/testsuite/libgomp.c-c++-common/icv-3.c
new file mode 100644
index 00000000000..54cbf99b597
--- /dev/null
+++ b/libgomp/testsuite/libgomp.c-c++-common/icv-3.c
@@ -0,0 +1,54 @@
+#include <omp.h>
+#include <stdlib.h>
+
+int
+main ()
+{
+  if (getenv ("OMP_NUM_TEAMS") == NULL
+      && omp_get_max_teams () != 0)
+    abort ();
+  omp_set_num_teams (7);
+  if (omp_get_max_teams () != 7)
+    abort ();
+  if (getenv ("OMP_TEAMS_THREAD_LIMIT") == NULL
+      && omp_get_teams_thread_limit () != 0)
+    abort ();
+  omp_set_teams_thread_limit (15);
+  if (omp_get_teams_thread_limit () != 15)
+    abort ();
+  #pragma omp teams
+  {
+    if (omp_get_max_teams () != 7
+	|| omp_get_teams_thread_limit () != 15
+	|| omp_get_num_teams () < 1
+	|| omp_get_num_teams () > 7
+	|| omp_get_team_num () < 0
+	|| omp_get_team_num () >= omp_get_num_teams ()
+	|| omp_get_thread_limit () < 1
+	|| omp_get_thread_limit () > 15)
+      abort ();
+  }
+  #pragma omp teams num_teams(5) thread_limit (13)
+  {
+    if (omp_get_max_teams () != 7
+	|| omp_get_teams_thread_limit () != 15
+	|| omp_get_num_teams () != 5
+	|| omp_get_team_num () < 0
+	|| omp_get_team_num () >= omp_get_num_teams ()
+	|| omp_get_thread_limit () < 1
+	|| omp_get_thread_limit () > 13)
+      abort ();
+  }
+  #pragma omp teams num_teams(8) thread_limit (16)
+  {
+    if (omp_get_max_teams () != 7
+	|| omp_get_teams_thread_limit () != 15
+	|| omp_get_num_teams () != 8
+	|| omp_get_team_num () < 0
+	|| omp_get_team_num () >= omp_get_num_teams ()
+	|| omp_get_thread_limit () < 1
+	|| omp_get_thread_limit () > 16)
+      abort ();
+  }
+  return 0;
+}
diff --git a/libgomp/testsuite/libgomp.c-c++-common/icv-4.c b/libgomp/testsuite/libgomp.c-c++-common/icv-4.c
new file mode 100644
index 00000000000..6cb671d8ac7
--- /dev/null
+++ b/libgomp/testsuite/libgomp.c-c++-common/icv-4.c
@@ -0,0 +1,40 @@
+/* { dg-set-target-env-var OMP_NUM_TEAMS "6" } */
+/* { dg-set-target-env-var OMP_TEAMS_THREAD_LIMIT "12" } */
+
+#include <omp.h>
+#include <stdlib.h>
+#include <string.h>
+
+int
+main ()
+{
+  if (getenv ("OMP_NUM_TEAMS") != NULL
+      && strcmp (getenv ("OMP_NUM_TEAMS"), "6") == 0)
+    {
+      if (omp_get_max_teams () != 6)
+	abort ();
+    }
+  else
+    omp_set_num_teams (6);
+  if (getenv ("OMP_TEAMS_THREAD_LIMIT") == NULL
+      && strcmp (getenv ("OMP_TEAMS_THREAD_LIMIT"), "12") == 0)
+    {
+      if (omp_get_teams_thread_limit () != 12)
+	abort ();
+    }
+  else
+    omp_set_teams_thread_limit (12);
+  #pragma omp teams
+  {
+    if (omp_get_max_teams () != 6
+	|| omp_get_teams_thread_limit () != 12
+	|| omp_get_num_teams () < 1
+	|| omp_get_num_teams () > 6
+	|| omp_get_team_num () < 0
+	|| omp_get_team_num () >= omp_get_num_teams ()
+	|| omp_get_thread_limit () < 1
+	|| omp_get_thread_limit () > 12)
+      abort ();
+  }
+  return 0;
+}