What's the point of splitting src_ids in list holding only one element and group them together back in the caller of task_arg_gen?