Merge pull request #1623 from huynhnguyen/master

fix issue with advi when running on GPU
pymc-devs · Jan 2, 2017 · 0bc5f8f · 0bc5f8f
2 parents cad8ec2 + 5376771
commit 0bc5f8f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pymc3/variational/advi.py b/pymc3/variational/advi.py
@@ -182,7 +182,7 @@ def _calc_elbo(vars, model, n_mcsamples, random_seed):
 
     uw = tt.vector('uw')
     uw.tag.test_value = np.concatenate([inarray.tag.test_value,
-                                        inarray.tag.test_value])
+                                        inarray.tag.test_value]).astype(theano.config.floatX)
 
     elbo = _elbo_t(logp, uw, inarray, n_mcsamples, random_seed)