code: def _dma_wait_pp_eqn(
      eqn: jax_core.JaxprEqn,
      context: jax_core.JaxprPpContext,
      settings: jax_core.JaxprPpSettings,
  ):
      del settings
      invars = eqn.invars
      tree = eqn.params["tree"]
      sem, sem_indexers, ref, indexers = tree_util.tree_unflatten(
          tree, invars
      )
      return pp.concat(
          [
              pp.text("dma_wait"),
              pp.text(" "),
              sp.pp_ref_indexers(context, ref, indexers),
              pp.text(" "),
              sp.pp_ref_indexers(context, sem, sem_indexers),
          ]
      )
  
imports: from __future__ import annotations
  import contextlib
  import dataclasses
  import enum
  from typing import Any, Callable
  import jax
  from jax._src import api_util
  from jax._src import core
  from jax._src import effects
  from jax._src import linear_util
  from jax._src import pretty_printer
  from jax._src import state
  from jax._src import tree_util
  from jax._src import util
  from jax._src.state import indexing
  from jax._src.state import primitives
  from jax._src.interpreters import mlir
  from jax._src.interpreters import partial_eval
  from jax._src.pallas.mosaic import core
  import jax.numpy
label_desc: Use image node_id 20 for calling a global function with example usage: _dma_wait_pp_eqn(eqn, context, settings) and returns: pp