code: def test_splitted_square_sum_with_predicate():
      s = tir.ScheduleState(
          splitted_square_sum_with_predicate, debug_mask="all"
      )
      # pylint: disable=protected-access
      assert s._get_cached_flags(_get_block(s, "B")) == CachedFlags(
          affine_binding=True,
          region_cover=True,
          stage_pipeline=True,
      )
  
imports: import sys
  import pytest
  import tvm
  import tvm.testing
  from tvm import tir
  from tvm.script import tir
  from tvm.tir.schedule.state import CachedFlags
  from tvm.tir.stmt_functor import post_order_visit
label_desc: Use image node_id 41 for calling a global function with example usage: test_splitted_square_sum_with_predicate() without return types