new builds

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

@@ -1,21 +1,13 @@
 import torch
-import torch.nn as nn
 
 from ._ops import ops
 
+from . import layers
 
-class LlamaRMSNorm(nn.Module):
-    weight: torch.Tensor
-    variance_epsilon: float
+def rmsnorm_forward(x: torch.Tensor, weight: torch.Tensor) -> torch.Tensor:
+    return ops.rmsnorm_forward(x, weight)
 
-    def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
-        return ops.rmsnorm_forward(
-            hidden_states,
-            self.weight,
-            bias=None,
-            residual=None,
-            eps=self.variance_epsilon,
-            dropout_p=0.0,
-            prenorm=False,
-            residual_in_fp32=False,
-        )
+__all__ = [
+    "layers",
+    "rmsnorm_forward",
+]

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

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

build/torch27-cxx11-cu118-x86_64-linux/rmsnorm_kernel/{_rmsnorm_kernel_538355f_dirty.abi3.so → _rmsnorm_kernel_20250918082933.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9f5117ccb9a81d1c1b5330385a72dd2d17eadd56c2f87d584d5b7259d02715cb
+oid sha256:dab5f4cfde265a04ba3805ecc82b4f4572e50d3a50e4a4d2aef1b8ccb9d57ba4
 size 2111512

build/torch27-cxx11-cu118-x86_64-linux/rmsnorm_kernel/layers.py

@@ -0,0 +1,15 @@
+import torch
+import torch.nn as nn
+
+from ._ops import ops
+
+
+class LlamaRMSNorm(nn.Module):
+    weight: torch.Tensor
+    variance_epsilon: float
+
+    def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
+        return ops.rmsnorm_forward(
+            hidden_states,
+            self.weight,
+        )

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

@@ -1,21 +1,13 @@
 import torch
-import torch.nn as nn
 
 from ._ops import ops
 
+from . import layers
 
-class LlamaRMSNorm(nn.Module):
-    weight: torch.Tensor
-    variance_epsilon: float
+def rmsnorm_forward(x: torch.Tensor, weight: torch.Tensor) -> torch.Tensor:
+    return ops.rmsnorm_forward(x, weight)
 
-    def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
-        return ops.rmsnorm_forward(
-            hidden_states,
-            self.weight,
-            bias=None,
-            residual=None,
-            eps=self.variance_epsilon,
-            dropout_p=0.0,
-            prenorm=False,
-            residual_in_fp32=False,
-        )
+__all__ = [
+    "layers",
+    "rmsnorm_forward",
+]

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

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

build/torch27-cxx11-cu126-x86_64-linux/rmsnorm_kernel/{_rmsnorm_kernel_538355f_dirty.abi3.so → _rmsnorm_kernel_20250918082933.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bced7bf4cd62415785bb30e3808e907246341d4e63572e70f22397258a022a41
-size 2250040
+oid sha256:1bcf1972c75c6fbb72321d2498cd67c18fc640be087f368fbdc1d0829e5f6382
+size 2250048

build/torch27-cxx11-cu126-x86_64-linux/rmsnorm_kernel/layers.py

@@ -0,0 +1,15 @@
+import torch
+import torch.nn as nn
+
+from ._ops import ops
+
+
+class LlamaRMSNorm(nn.Module):
+    weight: torch.Tensor
+    variance_epsilon: float
+
+    def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
+        return ops.rmsnorm_forward(
+            hidden_states,
+            self.weight,
+        )

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

@@ -1,21 +1,13 @@
 import torch
-import torch.nn as nn
 
 from ._ops import ops
 
+from . import layers
 
-class LlamaRMSNorm(nn.Module):
-    weight: torch.Tensor
-    variance_epsilon: float
+def rmsnorm_forward(x: torch.Tensor, weight: torch.Tensor) -> torch.Tensor:
+    return ops.rmsnorm_forward(x, weight)
 
-    def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
-        return ops.rmsnorm_forward(
-            hidden_states,
-            self.weight,
-            bias=None,
-            residual=None,
-            eps=self.variance_epsilon,
-            dropout_p=0.0,
-            prenorm=False,
-            residual_in_fp32=False,
-        )
+__all__ = [
+    "layers",
+    "rmsnorm_forward",
+]

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

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

build/torch27-cxx11-cu128-x86_64-linux/rmsnorm_kernel/{_rmsnorm_kernel_538355f_dirty.abi3.so → _rmsnorm_kernel_20250918082933.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2b1eae77e817b754c22ccf4828f0f022866b97c4815059edc0f2a7ca13f5a82f
+oid sha256:1baa27ab1648e5ca308718ce3c845bbfe6106ced438dce9c4bb6881264ed6354
 size 2506984

build/torch27-cxx11-cu128-x86_64-linux/rmsnorm_kernel/layers.py

@@ -0,0 +1,15 @@
+import torch
+import torch.nn as nn
+
+from ._ops import ops
+
+
+class LlamaRMSNorm(nn.Module):
+    weight: torch.Tensor
+    variance_epsilon: float
+
+    def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
+        return ops.rmsnorm_forward(
+            hidden_states,
+            self.weight,
+        )

build/torch28-cxx11-cu126-x86_64-linux/rmsnorm_kernel/__init__.py

@@ -1,21 +1,13 @@
 import torch
-import torch.nn as nn
 
 from ._ops import ops
 
+from . import layers
 
-class LlamaRMSNorm(nn.Module):
-    weight: torch.Tensor
-    variance_epsilon: float
+def rmsnorm_forward(x: torch.Tensor, weight: torch.Tensor) -> torch.Tensor:
+    return ops.rmsnorm_forward(x, weight)
 
-    def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
-        return ops.rmsnorm_forward(
-            hidden_states,
-            self.weight,
-            bias=None,
-            residual=None,
-            eps=self.variance_epsilon,
-            dropout_p=0.0,
-            prenorm=False,
-            residual_in_fp32=False,
-        )
+__all__ = [
+    "layers",
+    "rmsnorm_forward",
+]

build/torch28-cxx11-cu126-x86_64-linux/rmsnorm_kernel/_ops.py

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

build/torch28-cxx11-cu126-x86_64-linux/rmsnorm_kernel/{_rmsnorm_kernel_538355f_dirty.abi3.so → _rmsnorm_kernel_20250918082933.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d17e952ae2d5df0c7f4357b8bfbdc485c06750274dbadeda9e143272f3656332
-size 2198136
+oid sha256:965341541164ead5c9220b980a8bfcdadea6b755f64d3a3510ef3a112ecb3961
+size 2198144

build/torch28-cxx11-cu126-x86_64-linux/rmsnorm_kernel/layers.py

@@ -0,0 +1,15 @@
+import torch
+import torch.nn as nn
+
+from ._ops import ops
+
+
+class LlamaRMSNorm(nn.Module):
+    weight: torch.Tensor
+    variance_epsilon: float
+
+    def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
+        return ops.rmsnorm_forward(
+            hidden_states,
+            self.weight,
+        )

build/torch28-cxx11-cu128-x86_64-linux/rmsnorm_kernel/__init__.py

@@ -1,21 +1,13 @@
 import torch
-import torch.nn as nn
 
 from ._ops import ops
 
+from . import layers
 
-class LlamaRMSNorm(nn.Module):
-    weight: torch.Tensor
-    variance_epsilon: float
+def rmsnorm_forward(x: torch.Tensor, weight: torch.Tensor) -> torch.Tensor:
+    return ops.rmsnorm_forward(x, weight)
 
-    def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
-        return ops.rmsnorm_forward(
-            hidden_states,
-            self.weight,
-            bias=None,
-            residual=None,
-            eps=self.variance_epsilon,
-            dropout_p=0.0,
-            prenorm=False,
-            residual_in_fp32=False,
-        )
+__all__ = [
+    "layers",
+    "rmsnorm_forward",
+]

build/torch28-cxx11-cu128-x86_64-linux/rmsnorm_kernel/_ops.py

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

build/torch28-cxx11-cu128-x86_64-linux/rmsnorm_kernel/{_rmsnorm_kernel_538355f_dirty.abi3.so → _rmsnorm_kernel_20250918082933.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:ff6d1ec9ceead35e32eca260b38a3828167e32f5d911f50cc9752533f81b5d40
+oid sha256:bf1fe9ac30684982254023f11727a267fc62778660dca7f28b7cd4889815f8b1
 size 2446200

build/torch28-cxx11-cu128-x86_64-linux/rmsnorm_kernel/layers.py

@@ -0,0 +1,15 @@
+import torch
+import torch.nn as nn
+
+from ._ops import ops
+
+
+class LlamaRMSNorm(nn.Module):
+    weight: torch.Tensor
+    variance_epsilon: float
+
+    def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
+        return ops.rmsnorm_forward(
+            hidden_states,
+            self.weight,
+        )

build/torch28-cxx11-cu129-x86_64-linux/rmsnorm_kernel/__init__.py

@@ -1,21 +1,13 @@
 import torch
-import torch.nn as nn
 
 from ._ops import ops
 
+from . import layers
 
-class LlamaRMSNorm(nn.Module):
-    weight: torch.Tensor
-    variance_epsilon: float
+def rmsnorm_forward(x: torch.Tensor, weight: torch.Tensor) -> torch.Tensor:
+    return ops.rmsnorm_forward(x, weight)
 
-    def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
-        return ops.rmsnorm_forward(
-            hidden_states,
-            self.weight,
-            bias=None,
-            residual=None,
-            eps=self.variance_epsilon,
-            dropout_p=0.0,
-            prenorm=False,
-            residual_in_fp32=False,
-        )
+__all__ = [
+    "layers",
+    "rmsnorm_forward",
+]

build/torch28-cxx11-cu129-x86_64-linux/rmsnorm_kernel/_ops.py

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

build/torch28-cxx11-cu129-x86_64-linux/rmsnorm_kernel/_rmsnorm_kernel_20250918082933.abi3.so

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

build/torch28-cxx11-cu129-x86_64-linux/rmsnorm_kernel/_rmsnorm_kernel_538355f_dirty.abi3.so

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

build/torch28-cxx11-cu129-x86_64-linux/rmsnorm_kernel/layers.py

@@ -0,0 +1,15 @@
+import torch
+import torch.nn as nn
+
+from ._ops import ops
+
+
+class LlamaRMSNorm(nn.Module):
+    weight: torch.Tensor
+    variance_epsilon: float
+
+    def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
+        return ops.rmsnorm_forward(
+            hidden_states,
+            self.weight,
+        )