code: def set_parallel_input(input_parallel: bool):
      env.parallel_input_1d = input_parallel
  
imports: import torch
  import torch.distributed
  from colossalai.legacy.core import global_context
  from colossalai.legacy.global_variables import tensor_parallel_env
  from ..utils import divide
label_desc: Use image node_id 1 for calling a global function with example usage: set_parallel_input(input_parallel) without return types