[Bug]: dear teachers, when i try to change the softmax fuction on gpt2_spu_torch_version, there is a question

[Bug]: dear teachers, when i try to change the softmax fuction on gpt2_spu_torch_version, there is a question #3321