def test_hyperbolic_coupling(self):
k = coupling.HyperbolicTangent()
self.assertEqual(k.a, 1)
self.assertEqual(k.b, 1)
self.assertEqual(k.midpoint, 0)
self.assertEqual(k.sigma, 1)
self._apply_coupling(k)
def test_hyperbolic_coupling(self):
k = coupling.HyperbolicTangent()
assert k.a == 1
assert k.b == 1
assert k.midpoint == 0
assert k.sigma == 1
self._apply_coupling(k)
def test_tanh(self):
py_cf = py_coupling.HyperbolicTangent()
cu_cf = cu_tanh_cfe_pre(py_cf.a[0], py_cf.b[0], py_cf.midpoint[0], py_cf.sigma[0])
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))
