| fix(pytorch): modify longcat mlp to enable TE
Co-authored-by: MissingPompeii<guohao120@huawei.com>
# message auto-generated for no-merge-commit merge:
!4318 merge master_te into master
fix(pytorch): modify longcat mlp to enable TE
Created-by: MissingPompeii
Commit-by: MissingPompeii
Merged-by: ascend-robot
Description: longcat原始实现默认mlps_0、mlps_0走非te分支(默认use_te=False),linear_fc* 调用ColumnParallelLinear和RowParallelLinear;
修改后mlps_0、mlps_0在使能use_te情况下,mlps_0、mlps_0可以走te分支,linear_fc* 调用TEColumnParallelLinear和TERowParallelLinear。
See merge request: Ascend/MindSpeed-LLM!4318 | 2 个月前 |