code: def _verify_block(
      block_name, dim_name, block, dim, should_divide=True
  ):
      if block > dim:
          raise ValueError(
              f"{block_name}={block} should be smaller or equal to {dim_name}={dim}"
          )
      if should_divide and dim % block != 0:
          raise ValueError(
              f"{dim_name}={dim} should be divisible by {block_name}={block}"
          )
  
imports: from __future__ import annotations
  import dataclasses
  import functools
  from typing import Any, NamedTuple
  import jax
  from jax import lax
  from jax.experimental import pallas
  from jax.experimental.pallas import tpu
  import jax.numpy
label_desc: Use image node_id 20 for calling a global function with example usage: _verify_block(block_name, dim_name, block, dim, should_divide) without return types