Ejemplos de set_n_cpu_devices en Python

Ejemplo n.º 1

Archivo: variants_test.py Proyecto: stjordanis/chex

def setUpModule():
    # Parse flags (for `pytest`).
    if not FLAGS.is_parsed():
        FLAGS(sys.argv, known_only=True)
    fake.set_n_cpu_devices(FLAGS.chex_n_cpu_devices)
    asserts.assert_devices_available(FLAGS.chex_n_cpu_devices,
                                     'cpu',
                                     backend='cpu')

Ejemplo n.º 2

Archivo: fake_set_n_cpu_devices_test.py Proyecto: stompchicken/chex

    def test_set_n_cpu_devices(self):
        # Should not initialize backends.
        fake.set_n_cpu_devices(4)

        # Hence, this one does not fail.
        fake.set_n_cpu_devices(6)

        # This assert initializes backends.
        asserts.assert_devices_available(6, 'cpu', backend='cpu')

        # Which means that next call must fail.
        with self.assertRaisesRegex(
                RuntimeError, 'Attempted to set 8 devices, but 6 CPUs.+'):
            fake.set_n_cpu_devices(8)

Ejemplo n.º 3

Archivo: fake_set_n_cpu_devices_test.py Proyecto: graingert/chex

  def test_set_n_cpu_devices(self):
    try:
      # Should not initialize backends.
      fake.set_n_cpu_devices(4)
    except RuntimeError as set_cpu_exception:
      raise unittest.SkipTest(
          "set_n_cpu_devices: backend's already been initialized. "
          'Run this test in isolation from others.') from set_cpu_exception

    # Hence, this one does not fail.
    fake.set_n_cpu_devices(6)

    # This assert initializes backends.
    asserts.assert_devices_available(6, 'cpu', backend='cpu')

    # Which means that next call must fail.
    with self.assertRaisesRegex(RuntimeError,
                                'Attempted to set 8 devices, but 6 CPUs.+'):
      fake.set_n_cpu_devices(8)

Ejemplo n.º 4

def setUpModule():
    fake.set_n_cpu_devices()
    asserts.assert_devices_available(FLAGS['chex_n_cpu_devices'].value,
                                     'cpu',
                                     backend='cpu')

Ejemplo n.º 5

Archivo: fake_test.py Proyecto: stompchicken/chex

def setUpModule():
    fake.set_n_cpu_devices()

Ejemplo n.º 6

Archivo: variants_test.py Proyecto: misalraj/chex

def setUpModule():
    # Has the same effect as `fake.set_n_cpu_devices(FLAGS.chex_n_cpu_devices)`.
    fake.set_n_cpu_devices()
    asserts.assert_devices_available(FLAGS.chex_n_cpu_devices,
                                     'cpu',
                                     backend='cpu')