Hugging Face's logo

kernels-community
/
mlx_rmsnorm

Model card Files
xet
Community
medmekk HF Staff commited on
Commit
9c080e8
·
verified ·
1 Parent(s): 5d24fa7

Build uploaded using `kernels`.

Browse files
Files changed (15) hide show
  1. .gitattributes +2 -0
  2. build/torch28-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/__init__.cpython-313.pyc +0 -0
  3. build/torch28-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/_ops.cpython-313.pyc +0 -0
  4. build/torch28-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/functions.cpython-313.pyc +0 -0
  5. build/torch28-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/layers.cpython-313.pyc +0 -0
  6. build/torch28-metal-aarch64-darwin/mlx_rmsnorm/{_mlx_rmsnorm_06e17fa.abi3.so → _mlx_rmsnorm_06e17fa_dirty.abi3.so} +2 -2
  7. build/torch28-metal-aarch64-darwin/mlx_rmsnorm/_ops.py +3 -3
  8. build/torch28-metal-aarch64-darwin/mlx_rmsnorm/layers.py +2 -2
  9. build/torch29-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/__init__.cpython-313.pyc +0 -0
  10. build/torch29-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/_ops.cpython-313.pyc +0 -0
  11. build/torch29-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/functions.cpython-313.pyc +0 -0
  12. build/torch29-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/layers.cpython-313.pyc +0 -0
  13. build/torch29-metal-aarch64-darwin/mlx_rmsnorm/{_mlx_rmsnorm_06e17fa.abi3.so → _mlx_rmsnorm_06e17fa_dirty.abi3.so} +2 -2
  14. build/torch29-metal-aarch64-darwin/mlx_rmsnorm/_ops.py +3 -3
  15. build/torch29-metal-aarch64-darwin/mlx_rmsnorm/layers.py +2 -2
.gitattributes CHANGED
@@ -39,3 +39,5 @@ build/torch28-metal-aarch64-darwin/mlx_rmsnorm/_mlx_rmsnorm_57d865f.abi3.so filt
39
  build/torch29-metal-aarch64-darwin/mlx_rmsnorm/_mlx_rmsnorm_57d865f.abi3.so filter=lfs diff=lfs merge=lfs -text
40
  build/torch28-metal-aarch64-darwin/mlx_rmsnorm/_mlx_rmsnorm_06e17fa.abi3.so filter=lfs diff=lfs merge=lfs -text
41
  build/torch29-metal-aarch64-darwin/mlx_rmsnorm/_mlx_rmsnorm_06e17fa.abi3.so filter=lfs diff=lfs merge=lfs -text
 
 
 
39
  build/torch29-metal-aarch64-darwin/mlx_rmsnorm/_mlx_rmsnorm_57d865f.abi3.so filter=lfs diff=lfs merge=lfs -text
40
  build/torch28-metal-aarch64-darwin/mlx_rmsnorm/_mlx_rmsnorm_06e17fa.abi3.so filter=lfs diff=lfs merge=lfs -text
41
  build/torch29-metal-aarch64-darwin/mlx_rmsnorm/_mlx_rmsnorm_06e17fa.abi3.so filter=lfs diff=lfs merge=lfs -text
42
+ build/torch28-metal-aarch64-darwin/mlx_rmsnorm/_mlx_rmsnorm_06e17fa_dirty.abi3.so filter=lfs diff=lfs merge=lfs -text
43
+ build/torch29-metal-aarch64-darwin/mlx_rmsnorm/_mlx_rmsnorm_06e17fa_dirty.abi3.so filter=lfs diff=lfs merge=lfs -text
build/torch28-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/__init__.cpython-313.pyc CHANGED
Binary files a/build/torch28-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/__init__.cpython-313.pyc and b/build/torch28-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/__init__.cpython-313.pyc differ
 
build/torch28-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/_ops.cpython-313.pyc CHANGED
Binary files a/build/torch28-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/_ops.cpython-313.pyc and b/build/torch28-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/_ops.cpython-313.pyc differ
 
build/torch28-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/functions.cpython-313.pyc CHANGED
Binary files a/build/torch28-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/functions.cpython-313.pyc and b/build/torch28-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/functions.cpython-313.pyc differ
 
build/torch28-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/layers.cpython-313.pyc CHANGED
Binary files a/build/torch28-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/layers.cpython-313.pyc and b/build/torch28-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/layers.cpython-313.pyc differ
 
build/torch28-metal-aarch64-darwin/mlx_rmsnorm/{_mlx_rmsnorm_06e17fa.abi3.so → _mlx_rmsnorm_06e17fa_dirty.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6064021fd19860d705283dfb31cd15e6e4e1776dc4c1af0e29c8ee421deafa64
3
- size 219168
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a1ea7b58c37a1171e13ae823dae692fa6af104f6d7e7b77df0664067783d6537
3
+ size 219216
build/torch28-metal-aarch64-darwin/mlx_rmsnorm/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _mlx_rmsnorm_06e17fa
3
- ops = torch.ops._mlx_rmsnorm_06e17fa
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_mlx_rmsnorm_06e17fa::{op_name}"
 
1
  import torch
2
+ from . import _mlx_rmsnorm_06e17fa_dirty
3
+ ops = torch.ops._mlx_rmsnorm_06e17fa_dirty
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_mlx_rmsnorm_06e17fa_dirty::{op_name}"
build/torch28-metal-aarch64-darwin/mlx_rmsnorm/layers.py CHANGED
@@ -3,9 +3,9 @@ import torch
3
 
4
  class RMSNorm(torch.nn.Module):
5
  weight: torch.Tensor
6
- eps: float
7
 
8
  def forward(self, x: torch.Tensor) -> torch.Tensor:
9
- return rmsnorm_forward(x, self.weight, self.eps)
10
 
11
 
 
3
 
4
  class RMSNorm(torch.nn.Module):
5
  weight: torch.Tensor
6
+ variance_epsilon: float
7
 
8
  def forward(self, x: torch.Tensor) -> torch.Tensor:
9
+ return rmsnorm_forward(x, self.weight, self.variance_epsilon)
10
 
11
 
build/torch29-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/__init__.cpython-313.pyc CHANGED
Binary files a/build/torch29-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/__init__.cpython-313.pyc and b/build/torch29-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/__init__.cpython-313.pyc differ
 
build/torch29-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/_ops.cpython-313.pyc CHANGED
Binary files a/build/torch29-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/_ops.cpython-313.pyc and b/build/torch29-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/_ops.cpython-313.pyc differ
 
build/torch29-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/functions.cpython-313.pyc CHANGED
Binary files a/build/torch29-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/functions.cpython-313.pyc and b/build/torch29-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/functions.cpython-313.pyc differ
 
build/torch29-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/layers.cpython-313.pyc CHANGED
Binary files a/build/torch29-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/layers.cpython-313.pyc and b/build/torch29-metal-aarch64-darwin/mlx_rmsnorm/__pycache__/layers.cpython-313.pyc differ
 
build/torch29-metal-aarch64-darwin/mlx_rmsnorm/{_mlx_rmsnorm_06e17fa.abi3.so → _mlx_rmsnorm_06e17fa_dirty.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3eb0d24dc5c019fa77403103e936cff377e6003f2537fdfd0892f0a0f73ae922
3
- size 220128
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:78d53eec6a72a804940433209b70e5eae606272838124a352653cb25e071094d
3
+ size 220160
build/torch29-metal-aarch64-darwin/mlx_rmsnorm/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _mlx_rmsnorm_06e17fa
3
- ops = torch.ops._mlx_rmsnorm_06e17fa
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_mlx_rmsnorm_06e17fa::{op_name}"
 
1
  import torch
2
+ from . import _mlx_rmsnorm_06e17fa_dirty
3
+ ops = torch.ops._mlx_rmsnorm_06e17fa_dirty
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_mlx_rmsnorm_06e17fa_dirty::{op_name}"
build/torch29-metal-aarch64-darwin/mlx_rmsnorm/layers.py CHANGED
@@ -3,9 +3,9 @@ import torch
3
 
4
  class RMSNorm(torch.nn.Module):
5
  weight: torch.Tensor
6
- eps: float
7
 
8
  def forward(self, x: torch.Tensor) -> torch.Tensor:
9
- return rmsnorm_forward(x, self.weight, self.eps)
10
 
11
 
 
3
 
4
  class RMSNorm(torch.nn.Module):
5
  weight: torch.Tensor
6
+ variance_epsilon: float
7
 
8
  def forward(self, x: torch.Tensor) -> torch.Tensor:
9
+ return rmsnorm_forward(x, self.weight, self.variance_epsilon)
10
 
11