def test_runner_multyvac():
defn = model_definition(10, [bb, nich, niw(3)])
Y = toy_dataset(defn)
view = numpy_dataview(Y)
kc = runner.default_kernel_config(defn)
prng = rng()
latents = [model.initialize(defn, view, prng)
for _ in xrange(2)]
runners = [runner.runner(defn, view, latent, kc) for latent in latents]
r = parallel.runner(runners, backend='multyvac', layer='perf', core='f2')
r.run(r=prng, niters=1000)
r.run(r=prng, niters=1000)
def test_runner_multiprocessing():
defn = model_definition(10, [bb, nich, niw(3)])
Y = toy_dataset(defn)
view = numpy_dataview(Y)
kc = runner.default_kernel_config(defn)
prng = rng()
latents = [model.initialize(defn, view, prng)
for _ in xrange(mp.cpu_count())]
runners = [runner.runner(defn, view, latent, kc) for latent in latents]
r = parallel.runner(runners)
# check it is restartable
r.run(r=prng, niters=10)
r.run(r=prng, niters=10)
