code: def assert_dce(
      x: PyTree,
      msg: str,
      *,
      on_error: Literal[
          "default", "raise", "breakpoint", "nan"
      ] = "default",
  ) -> PyTree:
      """Asserts that a particular array (or PyTree of arrays) is DCE'd."""
  
      if currently_jitting():
          pred = jnp.invert(
              False
          )  # Prevent the trace-time error-raising from running.
          return error_if(x, pred, msg, on_error=on_error)
      else:
          # Don't run if not JIT'ing, as without the compiler nothing will be DCE'd.
          return x
  
imports: import functools
  import inspect
  import warnings
  from collections.abc import Sequence
  from typing import Literal, Union
  import jax
  import jax._src.traceback_util
  import jax.core
  import jax.lax
  import jax.numpy
  import jax.tree_util
  import numpy
  from jaxtyping import Array, ArrayLike, Bool, Int, PyTree
  from ._ad import filter_custom_jvp
  from ._config import EQX_ON_ERROR, EQX_ON_ERROR_BREAKPOINT_FRAMES
  from ._doc_utils import doc_remove_args
  from ._filters import combine, is_array, partition
  from ._jit import filter_jit
  from ._misc import currently_jitting
  from ._unvmap import unvmap_any, unvmap_max
label_desc: Use image node_id 7 for calling a global function with example usage: assert_dce(x, msg) and returns: error_if, x