code: def test_runtime_broadcasting():
      check_blockwise_runtime_broadcasting("JAX")
  
file_path: pytensor/tests/link/jax/test_blockwise.py
imports: import numpy
  import pytest
  from pytensor import config
  from pytensor.graph import FunctionGraph
  from pytensor.tensor import tensor
  from pytensor.tensor.blockwise import Blockwise
  from pytensor.tensor.math import Dot, matmul
  from tests.link.jax.test_basic import compare_jax_and_py
  from tests.tensor.test_blockwise import check_blockwise_runtime_broadcasting
label_desc: Use image node_id 1 for calling a global function with example usage: test_runtime_broadcasting() without return types