File tree Expand file tree Collapse file tree 2 files changed +503
-0
lines changed
src/liger_kernel/ops/backends/_ascend/ops Expand file tree Collapse file tree 2 files changed +503
-0
lines changed Original file line number Diff line number Diff line change 1717from liger_kernel .ops .backends ._ascend .ops .embedding import LigerEmbeddingFunction
1818from liger_kernel .ops .backends ._ascend .ops .embedding import embedding_backward
1919from liger_kernel .ops .backends ._ascend .ops .embedding import embedding_forward
20+ from liger_kernel .ops .backends ._ascend .ops .fused_add_rms_norm import LigerFusedAddRMSNormFunction
21+ from liger_kernel .ops .backends ._ascend .ops .fused_add_rms_norm import fused_add_rms_norm_backward
22+ from liger_kernel .ops .backends ._ascend .ops .fused_add_rms_norm import fused_add_rms_norm_forward
2023from liger_kernel .ops .backends ._ascend .ops .geglu import LigerGELUMulFunction
2124from liger_kernel .ops .backends ._ascend .ops .geglu import geglu_backward
2225from liger_kernel .ops .backends ._ascend .ops .geglu import geglu_forward
4043 "LigerEmbeddingFunction" ,
4144 "embedding_forward" ,
4245 "embedding_backward" ,
46+ "LigerFusedAddRMSNormFunction" ,
47+ "fused_add_rms_norm_forward" ,
48+ "fused_add_rms_norm_backward" ,
4349 "LigerGELUMulFunction" ,
4450 "geglu_forward" ,
4551 "geglu_backward" ,
You can’t perform that action at this time.
0 commit comments