[OpenMP] Fix potential division by 0

This patch fixes potential division by 0 in case hwloc does not recognize cores (or architecture has no cores). Patch by Andrey Churbanov Differential Revision: https://reviews.llvm.org/D90954
llvm · Nov 6, 2020 · 0820319 · 0820319
1 parent c955989
commit 0820319
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/openmp/runtime/src/kmp_affinity.cpp b/openmp/runtime/src/kmp_affinity.cpp
@@ -589,6 +589,8 @@ static int __kmp_affinity_create_hwloc_map(AddrUnsPair **address2os,
     else
       __kmp_nThreadsPerCore = 1; // no CORE found
     __kmp_ncores = __kmp_xproc / __kmp_nThreadsPerCore;
+    if (nCoresPerPkg == 0)
+      nCoresPerPkg = 1; // to prevent possible division by 0
     nPackages = (__kmp_xproc + nCoresPerPkg - 1) / nCoresPerPkg;
     if (__kmp_affinity_verbose) {
       KMP_INFORM(AffNotCapableUseLocCpuidL11, "KMP_AFFINITY");