mohitsha HF Staff commited on
Commit
2d8912e
·
verified ·
1 Parent(s): bf8c6f0

Upload 3 files

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ build/torch27-cxx11-rocm63-x86_64-linux/aiter_activation/_aiter_activation_a120ba4f5566b.so filter=lfs diff=lfs merge=lfs -text
build/torch27-cxx11-rocm63-x86_64-linux/aiter_activation/__init__.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from typing import Optional
2
+ import torch
3
+
4
+ from ._ops import ops
5
+
6
+ def silu_and_mul(out: torch.Tensor, input: torch.Tensor):
7
+ return ops.silu_and_mul(out, input)
8
+
9
+ def gelu_and_mul(out: torch.Tensor, input: torch.Tensor):
10
+ return ops.gelu_and_mul(out, input)
11
+
12
+ def gelu_tanh_and_mul(out: torch.Tensor, input: torch.Tensor):
13
+ return ops.gelu_tanh_and_mul(out, input)
14
+
15
+ __all__ = [
16
+ "silu_and_mul",
17
+ "gelu_and_mul",
18
+ "gelu_tanh_and_mul",
19
+ ]
build/torch27-cxx11-rocm63-x86_64-linux/aiter_activation/_aiter_activation_a120ba4f5566b.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3c742f896866c5c77be32a71ae954c34f0ac1610cbc6cce91264329f6591d7ef
3
+ size 2027336
build/torch27-cxx11-rocm63-x86_64-linux/aiter_activation/_ops.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from . import _aiter_activation_a120ba4f5566b
3
+ ops = torch.ops._aiter_activation_a120ba4f5566b
4
+
5
+ def add_op_namespace_prefix(op_name: str):
6
+ """
7
+ Prefix op by namespace.
8
+ """
9
+ return f"_aiter_activation_a120ba4f5566b::{op_name}"