def test_sigmoidal_coupling(self):
k = coupling.Sigmoidal()
self.assertEqual(k.cmin, -1.0)
self.assertEqual(k.cmax, 1.0)
self.assertEqual(k.midpoint, 0.0)
self.assertEqual(k.sigma, 230.)
self.assertEqual(k.a, 1.0)
def test_sigmoidal_coupling(self):
k = coupling.Sigmoidal()
assert k.cmin == -1.0
assert k.cmax == 1.0
assert k.midpoint == 0.0
assert k.sigma == 230.
assert k.a == 1.0
self._apply_coupling(k)
def test_sigm(self):
py_cf = py_coupling.Sigmoidal()
cu_cf = cu_sigm_cfe_post(py_cf.cmin[0], py_cf.cmax[0], py_cf.midpoint[0], py_cf.a[0], py_cf.sigma[0])
out = numpy.zeros((self.n_thread,), 'f')
gx = numpy.random.rand(self.n_thread).astype('f')
@self.jit_and_run(out, gx)
def kernel(out, gx):
t = cuda.blockDim.x * cuda.blockIdx.x + cuda.threadIdx.x
out[t] = cu_cf(gx[t])
numpy.testing.assert_allclose(out, py_cf.post(gx), 1e-4, 1e-5)
