code: def _make_aval(obj: object) -> jax_core.AbstractValue:
      if isinstance(obj, MemoryRef):
          return obj.get_aval()
      if isinstance(obj, SemaphoreType):
          return obj.get_aval()
      raise ValueError(
          f"No registered conversion for {type(obj)}. "
          "Only VMEM and SemaphoreType are supported."
      )
  
imports: from __future__ import annotations
  from collections.abc import Sequence
  import dataclasses
  import enum
  import functools
  from typing import Any
  from jax._src import core
  from jax._src import dtypes
  from jax._src import state
  from jax._src import tree_util
  from jax._src import util
  import jax.numpy
  from jax._src.pallas import core
label_desc: Use image node_id 2 for calling a global function with example usage: _make_aval(obj) and returns: obj, obj