parameters: (program, cur_rank)
return_statement: return [task_node.task_node()], task_id_to_rank
docstring: Origin scheduler for fleet executor, supports non-pp mode
      :param program: The origin program.
      :param cur_rank: Current rank (can be got from fleet.worker_index()).
      :return:
          task_nodes (list): four task nodes for current rank
          task_id_to_rank (dict): a fake dict, since there is no upstream or downstream, this dict won't be used
docstring_summary: Origin scheduler for fleet executor, supports non-pp mode
      :param program: The origin program.
      :param cur_rank: Current rank (can be got from fleet.worker_index()).
      :return:
          task_nodes (list): four task nodes for current rank
          task_id_to_rank (dict): a fake dict, since there is no upstream or downstream, this dict won't be used
docstring_tokens: Origin scheduler for fleet executor supports non - pp mode : param program : The origin program . : param cur_rank : Current rank ( can be got from fleet . worker_index () ) . : return : task_nodes ( list ) : four task nodes for current rank task_id_to_rank ( dict ) : a fake dict since there is no upstream or downstream this dict won t be used
function: def origin(program, cur_rank):
      """
      Origin scheduler for fleet executor, supports non-pp mode
      :param program: The origin program.
      :param cur_rank: Current rank (can be got from fleet.worker_index()).
      :return:
          task_nodes (list): four task nodes for current rank
          task_id_to_rank (dict): a fake dict, since there is no upstream or downstream, this dict won't be used
      """
      print("fleet executor will use python side origin scheduler.")
      task_node = TaskNode(
          program=program, cur_rank=cur_rank, max_run_times=1, max_slot_times=1)
      task_node.set_type("Compute")
      task_id = task_node.task_id()
      task_id_to_rank = {task_id: cur_rank}
      return [task_node.task_node()], task_id_to_rank
function_tokens: def origin ( program , cur_rank ) : print ( "fleet executor will use python side origin scheduler." ) task_node = TaskNode ( program = program , cur_rank = cur_rank , max_run_times = 1 , max_slot_times = 1 ) task_node . set_type ( "Compute" ) task_id = task_node . task_id ( ) task_id_to_rank = { task_id : cur_rank } return [ task_node . task_node ( ) ] , task_id_to_rank