Merge pull request #2338 from NervanaSystems/external_pr_2307

Changed python setup multiprocessing usage

Merge pull request #2338 from NervanaSystems/external_pr_2307
Changed python setup multiprocessing usage
7000d7d6 · Sang Ik Lee · GitHub · 18d0993e · 23b0c28b · 7000d7d6
Unverified Commit 7000d7d6 authored Jan 17, 2019 by Sang Ik Lee Committed by GitHub Jan 17, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

setup.py python/setup.py +2 -1

No files found.
--- a/python/setup.py
+++ b/python/setup.py
@@ -106,7 +106,8 @@ def parallelCCompile(
        self._compile(obj, src, ext, cc_args, extra_postargs, pp_opts)

    # convert to list, imap is evaluated on-demand
-    list(multiprocessing.pool.ThreadPool().imap(_single_compile, objects))
+    pool = multiprocessing.pool.ThreadPool()
+    list(pool.imap(_single_compile, objects))
    return objects