Merge 2274bfb into 4f67d63

pymc-devs · Jul 22, 2019 · 3ecf80e · 3ecf80e
2 parents 4f67d63 + 2274bfb
commit 3ecf80e
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/pymc3/step_methods/smc.py b/pymc3/step_methods/smc.py
@@ -196,7 +196,15 @@ def sample_smc(
 
     while beta < 1:
 
-        likelihoods = np.array([likelihood_logp(sample) for sample in posterior]).squeeze()
+        if step.parallel and cores > 1:
+            pool = mp.Pool(processes=cores)
+            results = pool.starmap(
+                likelihood_logp,
+                [(sample,) for sample in posterior],
+            )
+        else:
+            results = [likelihood_logp(sample) for sample in posterior]
+        likelihoods = np.array(results).squeeze()
         beta, old_beta, weights, sj = calc_beta(beta, likelihoods, step.threshold)
         model.marginal_likelihood *= sj
         # resample based on plausibility weights (selection)