[kernel] add turbodiffusion kernels (#972)

2025-12-30 04:23:10 -06:00
parent 704bc9aaf9
commit 40ce2d72f5
28 changed files with 2893 additions and 24 deletions
@@ -11,11 +11,24 @@ from fastvideo_kernel.vmoba import (
    process_moba_output,
 )

+from fastvideo_kernel.turbodiffusion_ops import (
+    Int8Linear,
+    FastRMSNorm,
+    FastLayerNorm,
+    int8_linear,
+    int8_quant,
+)
+
 __all__ = [
    "sliding_tile_attention",
    "video_sparse_attn",
    "moba_attn_varlen",
    "process_moba_input",
    "process_moba_output",
+    "Int8Linear",
+    "FastRMSNorm",
+    "FastLayerNorm",
+    "int8_linear",
+    "int8_quant",
    "__version__",
 ]