[Openmp-commits] [openmp] r292678 - Merging r292349:

[Jonathan Peyton via Openmp-commits]

Author: jlpeyton
Date: Fri Jan 20 17:32:49 2017
New Revision: 292678

URL: http://llvm.org/viewvc/llvm-project?rev=292678&view=rev
Log:
Merging r292349:
------------------------------------------------------------------------
r292349 | jlpeyton | 2017-01-18 01:02:21 -0600 (Wed, 18 Jan 2017) | 6 lines

Fix memory error in case of reinit using kmp_set_defaults() for lock code.

The lock tables were being reallocated if kmp_set_defaults() was called.
In the env_init code it says that the user should be able to switch between
different KMP_CONSISTENCY_CHECK values which is what this change enables.

------------------------------------------------------------------------

Added:
    openmp/branches/release_40/runtime/test/api/kmp_set_defaults_lock_bug.c
Modified:
    openmp/branches/release_40/runtime/src/kmp_lock.cpp

Modified: openmp/branches/release_40/runtime/src/kmp_lock.cpp
URL: http://llvm.org/viewvc/llvm-project/openmp/branches/release_40/runtime/src/kmp_lock.cpp?rev=292678&r1=292677&r2=292678&view=diff
==============================================================================
--- openmp/branches/release_40/runtime/src/kmp_lock.cpp (original)
+++ openmp/branches/release_40/runtime/src/kmp_lock.cpp Fri Jan 20 17:32:49 2017
@@ -3573,6 +3573,12 @@ __kmp_init_dynamic_user_locks()
         __kmp_indirect_unset = indirect_unset;
         __kmp_indirect_test  = indirect_test;
     }
+    // If the user locks have already been initialized, then return.
+    // Allow the switch between different KMP_CONSISTENCY_CHECK values,
+    // but do not allocate new lock tables if they have already been
+    // allocated.
+    if (__kmp_init_user_locks)
+        return;
 
     // Initialize lock index table
     __kmp_i_lock_table.size = KMP_I_LOCK_CHUNK;

Added: openmp/branches/release_40/runtime/test/api/kmp_set_defaults_lock_bug.c
URL: http://llvm.org/viewvc/llvm-project/openmp/branches/release_40/runtime/test/api/kmp_set_defaults_lock_bug.c?rev=292678&view=auto
==============================================================================
--- openmp/branches/release_40/runtime/test/api/kmp_set_defaults_lock_bug.c (added)
+++ openmp/branches/release_40/runtime/test/api/kmp_set_defaults_lock_bug.c Fri Jan 20 17:32:49 2017
@@ -0,0 +1,53 @@
+// RUN: %libomp-compile-and-run
+#include <stdio.h>
+#include "omp_testsuite.h"
+/* The bug occurs if the lock table is reallocated after
+   kmp_set_defaults() is called.  If the table is reallocated,
+   then the lock will not point to a valid lock object after the
+   kmp_set_defaults() call.*/
+omp_lock_t lock;
+
+int test_kmp_set_defaults_lock_bug()
+{
+  /* checks that omp_get_num_threads is equal to the number of
+     threads */
+  int nthreads_lib;
+  int nthreads = 0;
+
+  nthreads_lib = -1;
+
+  #pragma omp parallel
+  {
+    omp_set_lock(&lock);
+    nthreads++;
+    omp_unset_lock(&lock);
+    #pragma omp single
+    {
+      nthreads_lib = omp_get_num_threads ();
+    }  /* end of single */
+  } /* end of parallel */
+  kmp_set_defaults("OMP_NUM_THREADS");
+  #pragma omp parallel
+  {
+    omp_set_lock(&lock);
+    nthreads++;
+    omp_unset_lock(&lock);
+  } /* end of parallel */
+
+  return (nthreads == 2*nthreads_lib);
+}
+
+int main()
+{
+  int i;
+  int num_failed=0;
+  omp_init_lock(&lock);
+
+  for(i = 0; i < REPETITIONS; i++) {
+    if(!test_kmp_set_defaults_lock_bug()) {
+      num_failed++;
+    }
+  }
+  omp_destroy_lock(&lock);
+  return num_failed;
+}