LMLA attention_linear.py include core code of LMLA and RPE-style LA module we tested. test_cosformer_official.py include demo for us to test gpu inference time. Full program will be uploaded later.