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 000000000000..54cbf99b5974
--- /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 000000000000..6cb671d8ac74
--- /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;
+}
--
GitLab