ColossalAI/tests/test_layers/test_sequence
..
checks_seq
test_sequence.py