Build uploaded using `kernels`.

build/torch210-cxx11-cu126-x86_64-linux/__init__.py

@@ -1,14 +1,12 @@
 __version__ = "2.2.4"
 
+from .ops import selective_state_update, mamba_chunk_scan_combined, mamba_split_conv1d_scan_combined
 from .ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
-from .modules.mamba_simple import Mamba
-from .modules.mamba2 import Mamba2
-from .models.mixer_seq_simple import MambaLMHeadModel
+from .modules import Mamba, Mamba2
+from .models import MambaLMHeadModel
 
+falcon_mamba_inner_fn = mamba_inner_fn
 __all__ = [
-    "selective_scan_fn",
-    "mamba_inner_fn",
-    "Mamba",
-    "Mamba2",
-    "MambaLMHeadModel",
+    "selective_scan_fn", "mamba_inner_fn", "falcon_mamba_inner_fn", "selective_state_update", "mamba_chunk_scan_combined", "mamba_split_conv1d_scan_combined",
+    "Mamba", "Mamba2", "MambaLMHeadModel",
 ]

build/torch210-cxx11-cu126-x86_64-linux/{_mamba_ssm_1a63054.abi3.so → _mamba_ssm_cuda_muomt3evauf7u.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ed029d363b3bb2149788b3577e7b0a7c5783d7ce40d5ad0de45679b5cdfe5484
-size 444257200
+oid sha256:8eb110259b4313446e301660d897825a591775c7f9ed9853970b2fd6358e1eb9
+size 444257232

build/torch210-cxx11-cu126-x86_64-linux/_ops.py

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

build/torch210-cxx11-cu126-x86_64-linux/metadata.json

@@ -1,4 +1,6 @@
 {
   "version": 1,
-  "python-depends": []
+  "python-depends": [
+    "einops"
+  ]
 }

build/torch210-cxx11-cu126-x86_64-linux/models/__init__.py

@@ -0,0 +1,5 @@
+from .mixer_seq_simple import MambaLMHeadModel
+
+__all__ = [
+    "MambaLMHeadModel",
+]

build/torch210-cxx11-cu126-x86_64-linux/modules/__init__.py

@@ -0,0 +1,7 @@
+from .mamba_simple import Mamba
+from .mamba2 import Mamba2
+
+__all__ = [
+    "Mamba",
+    "Mamba2",
+]

build/torch210-cxx11-cu126-x86_64-linux/ops/__init__.py

@@ -0,0 +1 @@
+from .triton import selective_state_update, mamba_chunk_scan_combined, mamba_split_conv1d_scan_combined

build/torch210-cxx11-cu126-x86_64-linux/ops/triton/__init__.py

@@ -0,0 +1,2 @@
+from .selective_state_update import selective_state_update
+from .ssd_combined import mamba_chunk_scan_combined, mamba_split_conv1d_scan_combined

build/torch210-cxx11-cu126-x86_64-linux/utils/generation.py

@@ -11,8 +11,7 @@ import torch.nn.functional as F
 from einops import rearrange, repeat
 from torch import Tensor
 from torch.profiler import ProfilerActivity, profile, record_function
-from transformers.generation import GreedySearchDecoderOnlyOutput, SampleDecoderOnlyOutput, TextStreamer
-
+from transformers.generation import GenerateDecoderOnlyOutput, TextStreamer
 
 @dataclass
 class InferenceParams:

build/torch210-cxx11-cu128-x86_64-linux/__init__.py

@@ -1,14 +1,12 @@
 __version__ = "2.2.4"
 
+from .ops import selective_state_update, mamba_chunk_scan_combined, mamba_split_conv1d_scan_combined
 from .ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
-from .modules.mamba_simple import Mamba
-from .modules.mamba2 import Mamba2
-from .models.mixer_seq_simple import MambaLMHeadModel
+from .modules import Mamba, Mamba2
+from .models import MambaLMHeadModel
 
+falcon_mamba_inner_fn = mamba_inner_fn
 __all__ = [
-    "selective_scan_fn",
-    "mamba_inner_fn",
-    "Mamba",
-    "Mamba2",
-    "MambaLMHeadModel",
+    "selective_scan_fn", "mamba_inner_fn", "falcon_mamba_inner_fn", "selective_state_update", "mamba_chunk_scan_combined", "mamba_split_conv1d_scan_combined",
+    "Mamba", "Mamba2", "MambaLMHeadModel",
 ]

build/torch210-cxx11-cu128-x86_64-linux/{_mamba_ssm_1a63054.abi3.so → _mamba_ssm_cuda_p3tnkmr4mmukm.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b0997257c9217211de7446898e89756ab9a6237989ea6817f3bb018abde9df3c
-size 610662240
+oid sha256:81bd32914c56686dc50c48ed195db1540a9a90658c9dbcb0355f995e90dfe180
+size 610662272

build/torch210-cxx11-cu128-x86_64-linux/_ops.py

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

build/torch210-cxx11-cu128-x86_64-linux/metadata.json

@@ -1,4 +1,6 @@
 {
   "version": 1,
-  "python-depends": []
+  "python-depends": [
+    "einops"
+  ]
 }

build/torch210-cxx11-cu128-x86_64-linux/models/__init__.py

@@ -0,0 +1,5 @@
+from .mixer_seq_simple import MambaLMHeadModel
+
+__all__ = [
+    "MambaLMHeadModel",
+]

build/torch210-cxx11-cu128-x86_64-linux/modules/__init__.py

@@ -0,0 +1,7 @@
+from .mamba_simple import Mamba
+from .mamba2 import Mamba2
+
+__all__ = [
+    "Mamba",
+    "Mamba2",
+]

build/torch210-cxx11-cu128-x86_64-linux/ops/__init__.py

@@ -0,0 +1 @@
+from .triton import selective_state_update, mamba_chunk_scan_combined, mamba_split_conv1d_scan_combined

build/torch210-cxx11-cu128-x86_64-linux/ops/triton/__init__.py

@@ -0,0 +1,2 @@
+from .selective_state_update import selective_state_update
+from .ssd_combined import mamba_chunk_scan_combined, mamba_split_conv1d_scan_combined

build/torch210-cxx11-cu128-x86_64-linux/utils/generation.py

@@ -11,8 +11,7 @@ import torch.nn.functional as F
 from einops import rearrange, repeat
 from torch import Tensor
 from torch.profiler import ProfilerActivity, profile, record_function
-from transformers.generation import GreedySearchDecoderOnlyOutput, SampleDecoderOnlyOutput, TextStreamer
-
+from transformers.generation import GenerateDecoderOnlyOutput, TextStreamer
 
 @dataclass
 class InferenceParams:

build/torch210-cxx11-cu130-x86_64-linux/__init__.py

@@ -1,14 +1,12 @@
 __version__ = "2.2.4"
 
+from .ops import selective_state_update, mamba_chunk_scan_combined, mamba_split_conv1d_scan_combined
 from .ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
-from .modules.mamba_simple import Mamba
-from .modules.mamba2 import Mamba2
-from .models.mixer_seq_simple import MambaLMHeadModel
+from .modules import Mamba, Mamba2
+from .models import MambaLMHeadModel
 
+falcon_mamba_inner_fn = mamba_inner_fn
 __all__ = [
-    "selective_scan_fn",
-    "mamba_inner_fn",
-    "Mamba",
-    "Mamba2",
-    "MambaLMHeadModel",
+    "selective_scan_fn", "mamba_inner_fn", "falcon_mamba_inner_fn", "selective_state_update", "mamba_chunk_scan_combined", "mamba_split_conv1d_scan_combined",
+    "Mamba", "Mamba2", "MambaLMHeadModel",
 ]

build/torch210-cxx11-cu130-x86_64-linux/{_mamba_ssm_1a63054.abi3.so → _mamba_ssm_cuda_yndcxrj64jqjy.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:722c9a77eedb9a593885db984a9fc3a2e8bcea0e1c769a1b79c6f1312a68dddb
-size 446637448
+oid sha256:f1416b3f9ce1d7c2a33874baa9fca4de03305f693aff3aa952f990943002d3d7
+size 446637480

build/torch210-cxx11-cu130-x86_64-linux/_ops.py

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

build/torch210-cxx11-cu130-x86_64-linux/metadata.json

@@ -1,4 +1,6 @@
 {
   "version": 1,
-  "python-depends": []
+  "python-depends": [
+    "einops"
+  ]
 }

build/torch210-cxx11-cu130-x86_64-linux/models/__init__.py

@@ -0,0 +1,5 @@
+from .mixer_seq_simple import MambaLMHeadModel
+
+__all__ = [
+    "MambaLMHeadModel",
+]

build/torch210-cxx11-cu130-x86_64-linux/modules/__init__.py

@@ -0,0 +1,7 @@
+from .mamba_simple import Mamba
+from .mamba2 import Mamba2
+
+__all__ = [
+    "Mamba",
+    "Mamba2",
+]

build/torch210-cxx11-cu130-x86_64-linux/ops/__init__.py

@@ -0,0 +1 @@
+from .triton import selective_state_update, mamba_chunk_scan_combined, mamba_split_conv1d_scan_combined

build/torch210-cxx11-cu130-x86_64-linux/ops/triton/__init__.py

@@ -0,0 +1,2 @@
+from .selective_state_update import selective_state_update
+from .ssd_combined import mamba_chunk_scan_combined, mamba_split_conv1d_scan_combined

build/torch210-cxx11-cu130-x86_64-linux/utils/generation.py

@@ -11,8 +11,7 @@ import torch.nn.functional as F
 from einops import rearrange, repeat
 from torch import Tensor
 from torch.profiler import ProfilerActivity, profile, record_function
-from transformers.generation import GreedySearchDecoderOnlyOutput, SampleDecoderOnlyOutput, TextStreamer
-
+from transformers.generation import GenerateDecoderOnlyOutput, TextStreamer
 
 @dataclass
 class InferenceParams:

build/torch29-cxx11-cu126-x86_64-linux/__init__.py

@@ -1,14 +1,12 @@
 __version__ = "2.2.4"
 
+from .ops import selective_state_update, mamba_chunk_scan_combined, mamba_split_conv1d_scan_combined
 from .ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
-from .modules.mamba_simple import Mamba
-from .modules.mamba2 import Mamba2
-from .models.mixer_seq_simple import MambaLMHeadModel
+from .modules import Mamba, Mamba2
+from .models import MambaLMHeadModel
 
+falcon_mamba_inner_fn = mamba_inner_fn
 __all__ = [
-    "selective_scan_fn",
-    "mamba_inner_fn",
-    "Mamba",
-    "Mamba2",
-    "MambaLMHeadModel",
+    "selective_scan_fn", "mamba_inner_fn", "falcon_mamba_inner_fn", "selective_state_update", "mamba_chunk_scan_combined", "mamba_split_conv1d_scan_combined",
+    "Mamba", "Mamba2", "MambaLMHeadModel",
 ]

build/torch29-cxx11-cu126-x86_64-linux/{_mamba_ssm_1a63054.abi3.so → _mamba_ssm_cuda_lknozciqovp5m.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5a243449941f1718d8fb655cdeb3a41dafc4984d7717ac774a79f6dbcc215d20
-size 444242136
+oid sha256:81da74fdca330af5de6fdbfed5ee56c7acb25cd8799f2db37fbd58690f80c810
+size 444242168

build/torch29-cxx11-cu126-x86_64-linux/_ops.py

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

build/torch29-cxx11-cu126-x86_64-linux/metadata.json

@@ -1,4 +1,6 @@
 {
   "version": 1,
-  "python-depends": []
+  "python-depends": [
+    "einops"
+  ]
 }

build/torch29-cxx11-cu126-x86_64-linux/models/__init__.py

@@ -0,0 +1,5 @@
+from .mixer_seq_simple import MambaLMHeadModel
+
+__all__ = [
+    "MambaLMHeadModel",
+]

build/torch29-cxx11-cu126-x86_64-linux/modules/__init__.py

@@ -0,0 +1,7 @@
+from .mamba_simple import Mamba
+from .mamba2 import Mamba2
+
+__all__ = [
+    "Mamba",
+    "Mamba2",
+]

build/torch29-cxx11-cu126-x86_64-linux/ops/__init__.py

@@ -0,0 +1 @@
+from .triton import selective_state_update, mamba_chunk_scan_combined, mamba_split_conv1d_scan_combined

build/torch29-cxx11-cu126-x86_64-linux/ops/triton/__init__.py

@@ -0,0 +1,2 @@
+from .selective_state_update import selective_state_update
+from .ssd_combined import mamba_chunk_scan_combined, mamba_split_conv1d_scan_combined

build/torch29-cxx11-cu126-x86_64-linux/utils/generation.py

@@ -11,8 +11,7 @@ import torch.nn.functional as F
 from einops import rearrange, repeat
 from torch import Tensor
 from torch.profiler import ProfilerActivity, profile, record_function
-from transformers.generation import GreedySearchDecoderOnlyOutput, SampleDecoderOnlyOutput, TextStreamer
-
+from transformers.generation import GenerateDecoderOnlyOutput, TextStreamer
 
 @dataclass
 class InferenceParams:

build/torch29-cxx11-cu128-x86_64-linux/__init__.py

@@ -1,14 +1,12 @@
 __version__ = "2.2.4"
 
+from .ops import selective_state_update, mamba_chunk_scan_combined, mamba_split_conv1d_scan_combined
 from .ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
-from .modules.mamba_simple import Mamba
-from .modules.mamba2 import Mamba2
-from .models.mixer_seq_simple import MambaLMHeadModel
+from .modules import Mamba, Mamba2
+from .models import MambaLMHeadModel
 
+falcon_mamba_inner_fn = mamba_inner_fn
 __all__ = [
-    "selective_scan_fn",
-    "mamba_inner_fn",
-    "Mamba",
-    "Mamba2",
-    "MambaLMHeadModel",
+    "selective_scan_fn", "mamba_inner_fn", "falcon_mamba_inner_fn", "selective_state_update", "mamba_chunk_scan_combined", "mamba_split_conv1d_scan_combined",
+    "Mamba", "Mamba2", "MambaLMHeadModel",
 ]

build/torch29-cxx11-cu128-x86_64-linux/_mamba_ssm_1a63054.abi3.so

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

build/torch29-cxx11-cu128-x86_64-linux/_mamba_ssm_cuda_eomlxkg5rvjoa.abi3.so

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

build/torch29-cxx11-cu128-x86_64-linux/_ops.py

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

build/torch29-cxx11-cu128-x86_64-linux/metadata.json

@@ -1,4 +1,6 @@
 {
   "version": 1,
-  "python-depends": []
+  "python-depends": [
+    "einops"
+  ]
 }

build/torch29-cxx11-cu128-x86_64-linux/models/__init__.py

@@ -0,0 +1,5 @@
+from .mixer_seq_simple import MambaLMHeadModel
+
+__all__ = [
+    "MambaLMHeadModel",
+]

build/torch29-cxx11-cu128-x86_64-linux/modules/__init__.py

@@ -0,0 +1,7 @@
+from .mamba_simple import Mamba
+from .mamba2 import Mamba2
+
+__all__ = [
+    "Mamba",
+    "Mamba2",
+]

build/torch29-cxx11-cu128-x86_64-linux/ops/__init__.py

@@ -0,0 +1 @@
+from .triton import selective_state_update, mamba_chunk_scan_combined, mamba_split_conv1d_scan_combined

build/torch29-cxx11-cu128-x86_64-linux/ops/triton/__init__.py

@@ -0,0 +1,2 @@
+from .selective_state_update import selective_state_update
+from .ssd_combined import mamba_chunk_scan_combined, mamba_split_conv1d_scan_combined

build/torch29-cxx11-cu128-x86_64-linux/utils/generation.py

@@ -11,8 +11,7 @@ import torch.nn.functional as F
 from einops import rearrange, repeat
 from torch import Tensor
 from torch.profiler import ProfilerActivity, profile, record_function
-from transformers.generation import GreedySearchDecoderOnlyOutput, SampleDecoderOnlyOutput, TextStreamer
-
+from transformers.generation import GenerateDecoderOnlyOutput, TextStreamer
 
 @dataclass
 class InferenceParams:

build/torch29-cxx11-cu130-x86_64-linux/__init__.py

@@ -1,14 +1,12 @@
 __version__ = "2.2.4"
 
+from .ops import selective_state_update, mamba_chunk_scan_combined, mamba_split_conv1d_scan_combined
 from .ops.selective_scan_interface import selective_scan_fn, mamba_inner_fn
-from .modules.mamba_simple import Mamba
-from .modules.mamba2 import Mamba2
-from .models.mixer_seq_simple import MambaLMHeadModel
+from .modules import Mamba, Mamba2
+from .models import MambaLMHeadModel
 
+falcon_mamba_inner_fn = mamba_inner_fn
 __all__ = [
-    "selective_scan_fn",
-    "mamba_inner_fn",
-    "Mamba",
-    "Mamba2",
-    "MambaLMHeadModel",
+    "selective_scan_fn", "mamba_inner_fn", "falcon_mamba_inner_fn", "selective_state_update", "mamba_chunk_scan_combined", "mamba_split_conv1d_scan_combined",
+    "Mamba", "Mamba2", "MambaLMHeadModel",
 ]

build/torch29-cxx11-cu130-x86_64-linux/_mamba_ssm_1a63054.abi3.so

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

build/torch29-cxx11-cu130-x86_64-linux/_mamba_ssm_cuda_szbxxtsgftczo.abi3.so

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

build/torch29-cxx11-cu130-x86_64-linux/_ops.py

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