code: def test_compute_cycles_annotation(SRAM, FLASH, TwoConv2DTE):
      device_config = cs.EthosuDeviceConfig("ethos-u55-256")
      options = infra.make_options(
          cascade_region=SRAM,
          max_proposals=64,
          stripe_factors=4,
          max_plan_size=10,
          max_open_plans=8,
          max_closed_plans=32,
          always_copy_size=1024,
          disable_pareto_plans=False,
          disable_pareto_proposals=False,
          enable_striping=False,
      )
      sch, te_graph, const_dict = TwoConv2DTE
      cs.cascade(
          sch,
          te_graph,
          const_dict,
          options,
          SRAM,
          FLASH,
          [SRAM],
          device_config,
      )
  
      # Stages that should have compute cycle annotations
      # [copy, copy, conv2d, copy, conv2d]
      stages = [6, 8, 9, 18, 19]
      # Expected hints for each operation
      compute_cycles_hints = [4096, 5120, 1440, 2560, 3072]
  
      for stage, compute_cycles_hint in zip(
          stages, compute_cycles_hints
      ):
          op = sch.stages[stage]
          op_iter_vars = op.leaf_iter_vars[0]
          op_attrs = op.iter_var_attrs[op_iter_vars]
          assert op_attrs.pragma_keys[0] == "compute_cycles_hint"
          assert op_attrs.pragma_values[0] == compute_cycles_hint
  
imports: import pytest
  import tvm.contrib.ethosu.cascader
  from .None import infra
label_desc: Use image node_id 2 for calling a global function with example usage: test_compute_cycles_annotation(SRAM, FLASH, TwoConv2DTE) without return types