Files
ColossalAI/colossalai/fx/passes/adding_split_node_pass.py