Build

build/torch25-cxx11-cu118-x86_64-linux/flash_mla/{_flash_mla_ogm5yqv7ipuoy.abi3.so → _flash_mla_d4f4195.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6fd7ac5a2b5a201bfcef971204eeb50e892454281765a601f89c711af59452fb
-size 2564168
+oid sha256:a9a2b99b276b5aa714b27d1f54cc5da2d451e65a9ed385c583daf528f2c030a9
+size 2564144

build/torch25-cxx11-cu118-x86_64-linux/flash_mla/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _flash_mla_ogm5yqv7ipuoy
-ops = torch.ops._flash_mla_ogm5yqv7ipuoy
+from . import _flash_mla_d4f4195
+ops = torch.ops._flash_mla_d4f4195
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_flash_mla_ogm5yqv7ipuoy::{op_name}"
+    return f"_flash_mla_d4f4195::{op_name}"

build/torch25-cxx11-cu121-x86_64-linux/flash_mla/{_flash_mla_tq7soo7mu65hc.abi3.so → _flash_mla_d4f4195.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:536255477abf6714906de93e54c51a4390f0f4981616d59d1450e64a96bef3ad
-size 2595192
+oid sha256:77e241f633fa5b103f379ba6ac58d2cc068e0c3fc4d4f20ac1e1c679fc19614f
+size 2595176

build/torch25-cxx11-cu121-x86_64-linux/flash_mla/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _flash_mla_tq7soo7mu65hc
-ops = torch.ops._flash_mla_tq7soo7mu65hc
+from . import _flash_mla_d4f4195
+ops = torch.ops._flash_mla_d4f4195
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_flash_mla_tq7soo7mu65hc::{op_name}"
+    return f"_flash_mla_d4f4195::{op_name}"

build/torch25-cxx11-cu124-x86_64-linux/flash_mla/{_flash_mla_jmusvvowiwydi.abi3.so → _flash_mla_d4f4195.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:cb335fa92dfca6e14f6cbb892f42384623484a4a679df83a6e633079fa29790e
-size 2580264
+oid sha256:31aa895a57efbd29aeff693b65b02842926bf1788d6f98022c32470a60265f9e
+size 2580248

build/torch25-cxx11-cu124-x86_64-linux/flash_mla/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _flash_mla_jmusvvowiwydi
-ops = torch.ops._flash_mla_jmusvvowiwydi
+from . import _flash_mla_d4f4195
+ops = torch.ops._flash_mla_d4f4195
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_flash_mla_jmusvvowiwydi::{op_name}"
+    return f"_flash_mla_d4f4195::{op_name}"

build/torch25-cxx98-cu118-x86_64-linux/flash_mla/{_flash_mla_cqxjqwwlf7suu.abi3.so → _flash_mla_d4f4195.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c7dfe9f1ffc2669f9fc61e4d1c633bd3008e939d5b4da5fc9ae5f354ad99acfd
-size 2556816
+oid sha256:7454c10a3b29128e035bdb3fa18d5fc3706f7970542a0bcb55d9714f0999d42f
+size 2556792

build/torch25-cxx98-cu118-x86_64-linux/flash_mla/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _flash_mla_cqxjqwwlf7suu
-ops = torch.ops._flash_mla_cqxjqwwlf7suu
+from . import _flash_mla_d4f4195
+ops = torch.ops._flash_mla_d4f4195
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_flash_mla_cqxjqwwlf7suu::{op_name}"
+    return f"_flash_mla_d4f4195::{op_name}"

build/torch25-cxx98-cu121-x86_64-linux/flash_mla/_flash_mla_d4f4195.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c8cb9402f3091420227cbccf1ec4938a444765e26f5d34c356c76bf7c85630d0
+size 2587896

build/torch25-cxx98-cu121-x86_64-linux/flash_mla/_flash_mla_uqlcnehkqrgzo.abi3.so

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:7d6e13fb4833b751d089ff064509035402190c46e2234dfc7fc9a9f0524df6a1
-size 2587928

build/torch25-cxx98-cu121-x86_64-linux/flash_mla/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _flash_mla_uqlcnehkqrgzo
-ops = torch.ops._flash_mla_uqlcnehkqrgzo
+from . import _flash_mla_d4f4195
+ops = torch.ops._flash_mla_d4f4195
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_flash_mla_uqlcnehkqrgzo::{op_name}"
+    return f"_flash_mla_d4f4195::{op_name}"

build/torch25-cxx98-cu124-x86_64-linux/flash_mla/_flash_mla_d4f4195.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bb4be09cbde1979c1aa17e3bc93c1538f129b438d305bee0fe96f3c08efeee04
+size 2572968

build/torch25-cxx98-cu124-x86_64-linux/flash_mla/_flash_mla_iawwb234swad4.abi3.so

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:5c5feacdd7564b407f8ad544d66e9c1bf751a5d72f4142018fe24089663a977c
-size 2573000

build/torch25-cxx98-cu124-x86_64-linux/flash_mla/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _flash_mla_iawwb234swad4
-ops = torch.ops._flash_mla_iawwb234swad4
+from . import _flash_mla_d4f4195
+ops = torch.ops._flash_mla_d4f4195
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_flash_mla_iawwb234swad4::{op_name}"
+    return f"_flash_mla_d4f4195::{op_name}"

build/torch26-cxx11-cu118-x86_64-linux/flash_mla/_flash_mla_cb2u2ikyeixke.abi3.so

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:613fbc6946cf9feca9db47dcfdc1a436cf599024d64bf4308da65f3166ee31aa
-size 2564512

build/torch26-cxx11-cu118-x86_64-linux/flash_mla/_flash_mla_d4f4195.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:35c67c788220d8988e47cd4ad976495450b71cd682bd8ab08af3db066d625126
+size 2564496

build/torch26-cxx11-cu118-x86_64-linux/flash_mla/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _flash_mla_cb2u2ikyeixke
-ops = torch.ops._flash_mla_cb2u2ikyeixke
+from . import _flash_mla_d4f4195
+ops = torch.ops._flash_mla_d4f4195
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_flash_mla_cb2u2ikyeixke::{op_name}"
+    return f"_flash_mla_d4f4195::{op_name}"

build/torch26-cxx11-cu124-x86_64-linux/flash_mla/_flash_mla_d4f4195.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:069fb3e3a051c91e73390245c7463218829b8decf0f60bd6fc9a0ba8127b5bd2
+size 2580592

build/torch26-cxx11-cu124-x86_64-linux/flash_mla/_flash_mla_zdwdze3ckbyyo.abi3.so

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:f44dd1f22cf30be7062401e4c6b353d3da9d07bb2dbb41b0d81b0733490835ea
-size 2580608

build/torch26-cxx11-cu124-x86_64-linux/flash_mla/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _flash_mla_zdwdze3ckbyyo
-ops = torch.ops._flash_mla_zdwdze3ckbyyo
+from . import _flash_mla_d4f4195
+ops = torch.ops._flash_mla_d4f4195
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_flash_mla_zdwdze3ckbyyo::{op_name}"
+    return f"_flash_mla_d4f4195::{op_name}"

build/torch26-cxx11-cu126-x86_64-linux/flash_mla/_flash_mla_d4f4195.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:50fad86fa7bc15096c2a1feadf8091b20e188e32b8c0633423ec26e4e8e8e7ce
+size 2560552

build/torch26-cxx11-cu126-x86_64-linux/flash_mla/_flash_mla_rv7dqsm5x7jvw.abi3.so

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:e3710dfc1a63daecc4e7965555e075e2dfd498c4695b54f10c66ee451931d93a
-size 2560576

build/torch26-cxx11-cu126-x86_64-linux/flash_mla/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _flash_mla_rv7dqsm5x7jvw
-ops = torch.ops._flash_mla_rv7dqsm5x7jvw
+from . import _flash_mla_d4f4195
+ops = torch.ops._flash_mla_d4f4195
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_flash_mla_rv7dqsm5x7jvw::{op_name}"
+    return f"_flash_mla_d4f4195::{op_name}"

build/torch26-cxx98-cu118-x86_64-linux/flash_mla/_flash_mla_d4f4195.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ae937ddfbc3e6097b2fdd9197f2ddb5b9f66c65146a4de30ccab59dab6e18dd4
+size 2557136

build/torch26-cxx98-cu118-x86_64-linux/flash_mla/_flash_mla_qgsddurmm4eh6.abi3.so

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:d61051eaa3e3359c8a5fc3b781df27b7aa90c4733f31c90836414fa629a1d1a8
-size 2557160

build/torch26-cxx98-cu118-x86_64-linux/flash_mla/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _flash_mla_qgsddurmm4eh6
-ops = torch.ops._flash_mla_qgsddurmm4eh6
+from . import _flash_mla_d4f4195
+ops = torch.ops._flash_mla_d4f4195
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_flash_mla_qgsddurmm4eh6::{op_name}"
+    return f"_flash_mla_d4f4195::{op_name}"

build/torch26-cxx98-cu124-x86_64-linux/flash_mla/_flash_mla_d4f4195.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:503910324475f8bd9dab47687339005f58e5b623bf0c9e4234fabf099c08da33
+size 2573312

build/torch26-cxx98-cu124-x86_64-linux/flash_mla/_flash_mla_yu2nxxotkj3gw.abi3.so

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:c746b130a2057801ed54afd41147a45a330f5867755c7efa0ecf9cf6b0d3a2d0
-size 2573344

build/torch26-cxx98-cu124-x86_64-linux/flash_mla/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _flash_mla_yu2nxxotkj3gw
-ops = torch.ops._flash_mla_yu2nxxotkj3gw
+from . import _flash_mla_d4f4195
+ops = torch.ops._flash_mla_d4f4195
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_flash_mla_yu2nxxotkj3gw::{op_name}"
+    return f"_flash_mla_d4f4195::{op_name}"

build/torch26-cxx98-cu126-x86_64-linux/flash_mla/_flash_mla_7mo4qaqpi6iae.abi3.so

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:7372523fd91f328f78058f257d8ceff5f2dfc1b783b1bf4d6a4836c237fe7147
-size 2553304

build/torch26-cxx98-cu126-x86_64-linux/flash_mla/_flash_mla_d4f4195.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2c41fa4058ee2bb5d3d90458a7f92f0ef1c10e8bc854329cf7c208025bb244b2
+size 2553280

build/torch26-cxx98-cu126-x86_64-linux/flash_mla/_ops.py

@@ -1,9 +1,9 @@
 import torch
-from . import _flash_mla_7mo4qaqpi6iae
-ops = torch.ops._flash_mla_7mo4qaqpi6iae
+from . import _flash_mla_d4f4195
+ops = torch.ops._flash_mla_d4f4195
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_flash_mla_7mo4qaqpi6iae::{op_name}"
+    return f"_flash_mla_d4f4195::{op_name}"

build/torch27-cxx11-cu118-x86_64-linux/flash_mla/__init__.py

@@ -0,0 +1,33 @@
+import torch
+
+from ._ops import ops
+
+
+def get_mla_metadata(seqlens_k: torch.Tensor, s_q: int, h_kv: int):
+    return ops.get_mla_metadata(seqlens_k, s_q, h_kv)
+
+
+def mha_fwd_kvcache_mla(
+    q: torch.Tensor,
+    kcache: torch.Tensor,
+    vcache_: torch.Tensor,
+    head_size_v: int,
+    seqlens_k: torch.Tensor,
+    block_table: torch.Tensor,
+    softmax_scale: float,
+    is_causal_: bool,
+    tile_scheduler_metadata: torch.Tensor,
+    num_splits: torch.Tensor,
+) -> torch.Tensor:
+    return ops.mha_fwd_kvcache_mla(
+        q,
+        kcache,
+        vcache_,
+        head_size_v,
+        seqlens_k,
+        block_table,
+        softmax_scale,
+        is_causal_,
+        tile_scheduler_metadata,
+        num_splits
+    )

build/torch27-cxx11-cu118-x86_64-linux/flash_mla/_flash_mla_d4f4195.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:59c4034880f4482b06e447a2c4810aaf8009b7d4c86a4fd71356f169df986535
+size 2564632

build/torch27-cxx11-cu118-x86_64-linux/flash_mla/_ops.py

@@ -0,0 +1,9 @@
+import torch
+from . import _flash_mla_d4f4195
+ops = torch.ops._flash_mla_d4f4195
+
+def add_op_namespace_prefix(op_name: str):
+    """
+    Prefix op by namespace.
+    """
+    return f"_flash_mla_d4f4195::{op_name}"

build/torch27-cxx11-cu126-x86_64-linux/flash_mla/__init__.py

@@ -0,0 +1,33 @@
+import torch
+
+from ._ops import ops
+
+
+def get_mla_metadata(seqlens_k: torch.Tensor, s_q: int, h_kv: int):
+    return ops.get_mla_metadata(seqlens_k, s_q, h_kv)
+
+
+def mha_fwd_kvcache_mla(
+    q: torch.Tensor,
+    kcache: torch.Tensor,
+    vcache_: torch.Tensor,
+    head_size_v: int,
+    seqlens_k: torch.Tensor,
+    block_table: torch.Tensor,
+    softmax_scale: float,
+    is_causal_: bool,
+    tile_scheduler_metadata: torch.Tensor,
+    num_splits: torch.Tensor,
+) -> torch.Tensor:
+    return ops.mha_fwd_kvcache_mla(
+        q,
+        kcache,
+        vcache_,
+        head_size_v,
+        seqlens_k,
+        block_table,
+        softmax_scale,
+        is_causal_,
+        tile_scheduler_metadata,
+        num_splits
+    )

build/torch27-cxx11-cu126-x86_64-linux/flash_mla/_flash_mla_d4f4195.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:5db69ef4975e2eee001e6a9b7466c1fe40bc2228ed64eb8c24caf3e0fb6ed0b2
+size 2560584

build/torch27-cxx11-cu126-x86_64-linux/flash_mla/_ops.py

@@ -0,0 +1,9 @@
+import torch
+from . import _flash_mla_d4f4195
+ops = torch.ops._flash_mla_d4f4195
+
+def add_op_namespace_prefix(op_name: str):
+    """
+    Prefix op by namespace.
+    """
+    return f"_flash_mla_d4f4195::{op_name}"

build/torch27-cxx11-cu128-x86_64-linux/flash_mla/__init__.py

@@ -0,0 +1,33 @@
+import torch
+
+from ._ops import ops
+
+
+def get_mla_metadata(seqlens_k: torch.Tensor, s_q: int, h_kv: int):
+    return ops.get_mla_metadata(seqlens_k, s_q, h_kv)
+
+
+def mha_fwd_kvcache_mla(
+    q: torch.Tensor,
+    kcache: torch.Tensor,
+    vcache_: torch.Tensor,
+    head_size_v: int,
+    seqlens_k: torch.Tensor,
+    block_table: torch.Tensor,
+    softmax_scale: float,
+    is_causal_: bool,
+    tile_scheduler_metadata: torch.Tensor,
+    num_splits: torch.Tensor,
+) -> torch.Tensor:
+    return ops.mha_fwd_kvcache_mla(
+        q,
+        kcache,
+        vcache_,
+        head_size_v,
+        seqlens_k,
+        block_table,
+        softmax_scale,
+        is_causal_,
+        tile_scheduler_metadata,
+        num_splits
+    )

build/torch27-cxx11-cu128-x86_64-linux/flash_mla/_flash_mla_d4f4195.abi3.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:3fc7eb9341c975d0e313d837977ca3ed13556e6fe63926e0bf117f62499ea052
+size 2615448

build/torch27-cxx11-cu128-x86_64-linux/flash_mla/_ops.py

@@ -0,0 +1,9 @@
+import torch
+from . import _flash_mla_d4f4195
+ops = torch.ops._flash_mla_d4f4195
+
+def add_op_namespace_prefix(op_name: str):
+    """
+    Prefix op by namespace.
+    """
+    return f"_flash_mla_d4f4195::{op_name}"