code: def setUpModule():
      global prev_xla_flags
      prev_xla_flags = os.getenv("XLA_FLAGS")
      flags_str = prev_xla_flags or ""
      # Don't override user-specified device count, or other XLA flags.
      if "xla_force_host_platform_device_count" not in flags_str:
          os.environ["XLA_FLAGS"] = (
              flags_str + " --xla_force_host_platform_device_count=8"
          )
      # Clear any cached backends so new CPU backend will pick up the env var.
      xla_bridge.get_backend.cache_clear()
  
imports: import os
  from unittest import SkipTest
  from absl.testing import absltest
  import jax
  import jax.numpy
  from jax import lax
  from jax._src import test_util
  from jax._src import xla_bridge
  from jax import config
label_desc: Use image node_id 1 for calling a global function with example usage: setUpModule() without return types