Merge remote-tracking branch 'upstream/main' into moe-kernel

Author: Mecoli1219

src/liger_kernel/ops/backends/_ascend/ops/__init__.py

@@ -26,6 +26,9 @@
 from liger_kernel.ops.backends._ascend.ops.fused_add_rms_norm import LigerFusedAddRMSNormFunction
 from liger_kernel.ops.backends._ascend.ops.fused_add_rms_norm import fused_add_rms_norm_backward
 from liger_kernel.ops.backends._ascend.ops.fused_add_rms_norm import fused_add_rms_norm_forward
+from liger_kernel.ops.backends._ascend.ops.fused_linear_cross_entropy import LigerFusedLinearCrossEntropyFunction
+from liger_kernel.ops.backends._ascend.ops.fused_linear_cross_entropy import fused_linear_cross_entropy_backward
+from liger_kernel.ops.backends._ascend.ops.fused_linear_cross_entropy import fused_linear_cross_entropy_forward
 from liger_kernel.ops.backends._ascend.ops.fused_linear_jsd import LigerFusedLinearJSDFunction
 from liger_kernel.ops.backends._ascend.ops.fused_linear_jsd import fused_linear_jsd_backward
 from liger_kernel.ops.backends._ascend.ops.fused_linear_jsd import fused_linear_jsd_forward
@@ -140,4 +143,7 @@
     "sparsemax_backward",
     "LigerFusedNeighborhoodAttentionFunction",
     "fused_neighborhood_attention_forward",
+    "LigerFusedLinearCrossEntropyFunction",
+    "fused_linear_cross_entropy_forward",
+    "fused_linear_cross_entropy_backward",
 ]