code: def test_iter_map_simplify_unit_loop_order():
      """Test itermap simplify"""
      x = tvm.tir.Var("x", "int64")
      y = tvm.tir.Var("y", "int64")
      z = tvm.tir.Var("z", "int64")
  
      # trivial iterators can be found at any when comparing via scale
      # ensure order unchange
      assert_iter_map_simplify(
          {x + y + z: x + y + z},
          var_dom([(x, 1), (y, 1), (z, 1)]),
          simplify_trivial_iterators=False,
      )
  
      # Even with simplifcation, it should follow the original order
      assert_iter_map_simplify(
          {x + y + (z // 4) * 4 + z % 4: z + x + y},
          var_dom([(x, 1), (y, 1), (z, 32)]),
          simplify_trivial_iterators=False,
      )
  
      assert_iter_map_simplify(
          {y + 64 - x % 2 * 64: y + 64 - x % 2 * 64},
          var_dom([(x, 6), (y, 64)]),
          simplify_trivial_iterators=False,
      )
  
      # When we have iterators that have same scale but one of them come
      # with unit extent, we should prioritize unit extent
      assert_iter_map_simplify(
          {x // 128 + y + z: y + z},
          var_dom([(x, 128), (y, 128), (z, 1)]),
          simplify_trivial_iterators=False,
      )
  
imports: import tvm
  import tvm.testing
  from tvm.tir import floordiv, floormod
label_desc: Use image node_id 27 for calling a global function with example usage: test_iter_map_simplify_unit_loop_order() without return types