commit 72a8498e0881a5480c66d950ffa77cd44c3b65fa
author Jonathan Peyton <jonathan.l.peyton@intel.com> Thu Jun 16 20:14:54 2016 +0000
committer Jonathan Peyton <jonathan.l.peyton@intel.com> Thu Jun 16 20:14:54 2016 +0000
tree b567916fe387a8d278a03f4f7acf939802a89279
parent 4ba3b0cda9050053e327a067d06281df9ef6a5bf

Fixed missing memory cleanup in __kmp_affinity_create_hwloc_map()

Cleanup: fixed missing memory cleanup in couple of corner cases. Fixes possible
memory leak in some corner cases

Patch by Andrey Churbanov

Differential Revision: http://reviews.llvm.org/D21355

llvm-svn: 272946

openmp/runtime/src/kmp_affinity.cpp

diff --git a/openmp/runtime/src/kmp_affinity.cpp b/openmp/runtime/src/kmp_affinity.cpp
index b6e77a6..6f4c7c1 100644
--- a/openmp/runtime/src/kmp_affinity.cpp
+++ b/openmp/runtime/src/kmp_affinity.cpp
@@ -414,6 +414,7 @@
             KMP_INFORM(Topology, "KMP_AFFINITY", nPackages, nCoresPerPkg,
               __kmp_nThreadsPerCore, __kmp_ncores);
         }
+        KMP_CPU_FREE(oldMask);
         return 0;
     }
 
@@ -562,6 +563,7 @@
     }
 
     if (__kmp_affinity_type == affinity_none) {
+        __kmp_free(retval);
         KMP_CPU_FREE(oldMask);
         return 0;
     }