Files
ColossalAI/colossalai/nn/layer/parallel_sequence/_utils.py