def test_kura(self): py_cf = py_coupling.Kuramoto() cu_cf = cu_kura_cfe_pre() out = numpy.zeros((self.n_thread, ), 'f') xj, xi = numpy.random.rand(2, self.n_thread).astype('f') @self.jit_and_run(out, xi, xj) def kernel(out, xi, xj): t = cuda.blockDim.x * cuda.blockIdx.x + cuda.threadIdx.x out[t] = cu_cf(xi[t], xj[t]) numpy.testing.assert_allclose(out, py_cf.pre(xi, xj), 1e-5, 1e-6)
def gen_sim(a): dt = 0.1 conn = connectivity.Connectivity() conn.weights = weights conn.tract_lengths = idelays * dt conn.speed = 1.0 sim = simulator.Simulator( coupling=py_coupling.Kuramoto(a=a), connectivity=conn, model=models.Kuramoto(omega=100 * 2 * numpy.pi / 1e3), monitors=monitors.Raw(), integrator=integrators.EulerDeterministic(dt=dt)) sim.configure() sim.history[:] = 0.1 return sim
def test_kuramoto_coupling(self): k = coupling.Kuramoto() assert k.a == 1 self._apply_coupling(k)
def test_kuramoto_coupling(self): k = coupling.Kuramoto() self.assertEqual(k.a, 1) self._apply_coupling(k)