Upload custom kernels

build/torch-universal/liger_kernels/__init__.py

@@ -1,16 +1,16 @@
-from cross_entropy import LigerCrossEntropyFunction
-from fused_linear_cross_entropy import LigerFusedLinearCrossEntropyFunction
-from dyt import LigerDyTFunction
-from geglu import LigerGELUMulFunction
-from group_norm import LigerGroupNormFunction
-from kl_div import LigerKLDivLossFunction
-from layer_norm import LigerLayerNormFunction
-from qwen2vl_mrope import LigerQwen2VLMRopeFunction
-from rms_norm import LigerRMSNormFunction
-from jsd import LigerJSDFunction
-from rope import LigerRopeFunction
-from swiglu import LigerSiLUMulFunction
-from tvd import LigerTVDLossFunction
+from .cross_entropy import LigerCrossEntropyFunction
+from .fused_linear_cross_entropy import LigerFusedLinearCrossEntropyFunction
+from .dyt import LigerDyTFunction
+from .geglu import LigerGELUMulFunction
+from .group_norm import LigerGroupNormFunction
+from .kl_div import LigerKLDivLossFunction
+from .layer_norm import LigerLayerNormFunction
+from .qwen2vl_mrope import LigerQwen2VLMRopeFunction
+from .rms_norm import LigerRMSNormFunction
+from .jsd import LigerJSDFunction
+from .rope import LigerRopeFunction
+from .swiglu import LigerSiLUMulFunction
+from .tvd import LigerTVDLossFunction
 
 __all__ = [
     "LigerCrossEntropyFunction",

build/torch-universal/liger_kernels/_ops.py

@@ -1,8 +1,8 @@
 import torch
-ops = torch.ops._liger_kernels_20250505094412
+ops = torch.ops._liger_kernels_20250505100655
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_liger_kernels_20250505094412::{op_name}"
+    return f"_liger_kernels_20250505100655::{op_name}"

torch-ext/liger_kernels/__init__.py

@@ -1,16 +1,16 @@
-from cross_entropy import LigerCrossEntropyFunction
-from fused_linear_cross_entropy import LigerFusedLinearCrossEntropyFunction
-from dyt import LigerDyTFunction
-from geglu import LigerGELUMulFunction
-from group_norm import LigerGroupNormFunction
-from kl_div import LigerKLDivLossFunction
-from layer_norm import LigerLayerNormFunction
-from qwen2vl_mrope import LigerQwen2VLMRopeFunction
-from rms_norm import LigerRMSNormFunction
-from jsd import LigerJSDFunction
-from rope import LigerRopeFunction
-from swiglu import LigerSiLUMulFunction
-from tvd import LigerTVDLossFunction
+from .cross_entropy import LigerCrossEntropyFunction
+from .fused_linear_cross_entropy import LigerFusedLinearCrossEntropyFunction
+from .dyt import LigerDyTFunction
+from .geglu import LigerGELUMulFunction
+from .group_norm import LigerGroupNormFunction
+from .kl_div import LigerKLDivLossFunction
+from .layer_norm import LigerLayerNormFunction
+from .qwen2vl_mrope import LigerQwen2VLMRopeFunction
+from .rms_norm import LigerRMSNormFunction
+from .jsd import LigerJSDFunction
+from .rope import LigerRopeFunction
+from .swiglu import LigerSiLUMulFunction
+from .tvd import LigerTVDLossFunction
 
 __all__ = [
     "LigerCrossEntropyFunction",