Build
Browse files- build/torch25-cxx11-cu118-x86_64-linux/rotary/__init__.py +19 -0
- build/torch25-cxx11-cu118-x86_64-linux/rotary/_ops.py +9 -0
- build/torch25-cxx11-cu118-x86_64-linux/rotary/_rotary_5yzc45v7kk3yu.abi3.so +3 -0
- build/torch25-cxx11-cu121-x86_64-linux/rotary/__init__.py +19 -0
- build/torch25-cxx11-cu121-x86_64-linux/rotary/_ops.py +9 -0
- build/torch25-cxx11-cu121-x86_64-linux/rotary/_rotary_tbiepw2a2ep3e.abi3.so +3 -0
- build/torch25-cxx11-cu124-x86_64-linux/rotary/__init__.py +19 -0
- build/torch25-cxx11-cu124-x86_64-linux/rotary/_ops.py +9 -0
- build/torch25-cxx11-cu124-x86_64-linux/rotary/_rotary_6w5syhrhmerj6.abi3.so +3 -0
- build/torch25-cxx98-cu118-x86_64-linux/rotary/__init__.py +19 -0
- build/torch25-cxx98-cu118-x86_64-linux/rotary/_ops.py +9 -0
- build/torch25-cxx98-cu118-x86_64-linux/rotary/_rotary_joujmbgvsytzg.abi3.so +3 -0
- build/torch25-cxx98-cu121-x86_64-linux/rotary/__init__.py +19 -0
- build/torch25-cxx98-cu121-x86_64-linux/rotary/_ops.py +9 -0
- build/torch25-cxx98-cu121-x86_64-linux/rotary/_rotary_mi2o7e7sishyw.abi3.so +3 -0
- build/torch25-cxx98-cu124-x86_64-linux/rotary/__init__.py +19 -0
- build/torch25-cxx98-cu124-x86_64-linux/rotary/_ops.py +9 -0
- build/torch25-cxx98-cu124-x86_64-linux/rotary/_rotary_rngiohfhfwuge.abi3.so +3 -0
- build/torch26-cxx11-cu118-x86_64-linux/rotary/__init__.py +19 -0
- build/torch26-cxx11-cu118-x86_64-linux/rotary/_ops.py +9 -0
- build/torch26-cxx11-cu118-x86_64-linux/rotary/_rotary_alv7mzltcxxpq.abi3.so +3 -0
- build/torch26-cxx11-cu124-x86_64-linux/rotary/__init__.py +19 -0
- build/torch26-cxx11-cu124-x86_64-linux/rotary/_ops.py +9 -0
- build/torch26-cxx11-cu124-x86_64-linux/rotary/_rotary_c4eyapeep6gty.abi3.so +3 -0
- build/torch26-cxx11-cu126-x86_64-linux/rotary/__init__.py +19 -0
- build/torch26-cxx11-cu126-x86_64-linux/rotary/_ops.py +9 -0
- build/torch26-cxx11-cu126-x86_64-linux/rotary/_rotary_lodp6xeztste6.abi3.so +3 -0
- build/torch26-cxx98-cu118-x86_64-linux/rotary/__init__.py +19 -0
- build/torch26-cxx98-cu118-x86_64-linux/rotary/_ops.py +9 -0
- build/torch26-cxx98-cu118-x86_64-linux/rotary/_rotary_z27mls7mz4e7m.abi3.so +3 -0
- build/torch26-cxx98-cu124-x86_64-linux/rotary/__init__.py +19 -0
- build/torch26-cxx98-cu124-x86_64-linux/rotary/_ops.py +9 -0
- build/torch26-cxx98-cu124-x86_64-linux/rotary/_rotary_3bktke4p3hz3a.abi3.so +3 -0
- build/torch26-cxx98-cu126-x86_64-linux/rotary/__init__.py +19 -0
- build/torch26-cxx98-cu126-x86_64-linux/rotary/_ops.py +9 -0
- build/torch26-cxx98-cu126-x86_64-linux/rotary/_rotary_fvednlzeqgg5s.abi3.so +3 -0
build/torch25-cxx11-cu118-x86_64-linux/rotary/__init__.py
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import Tuple
|
2 |
+
import torch
|
3 |
+
|
4 |
+
from ._ops import ops
|
5 |
+
|
6 |
+
|
7 |
+
def apply_rotary(
|
8 |
+
x1: torch.Tensor,
|
9 |
+
x2: torch.Tensor,
|
10 |
+
cos: torch.Tensor,
|
11 |
+
sin: torch.Tensor,
|
12 |
+
out1: torch.Tensor,
|
13 |
+
out2: torch.Tensor,
|
14 |
+
conj: bool,
|
15 |
+
):
|
16 |
+
ops.apply_rotary(x1, x2, cos, sin, out1, out2, conj)
|
17 |
+
|
18 |
+
|
19 |
+
__all__ = ["apply_rotary"]
|
build/torch25-cxx11-cu118-x86_64-linux/rotary/_ops.py
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from . import _rotary_5yzc45v7kk3yu
|
3 |
+
ops = torch.ops._rotary_5yzc45v7kk3yu
|
4 |
+
|
5 |
+
def add_op_namespace_prefix(op_name: str):
|
6 |
+
"""
|
7 |
+
Prefix op by namespace.
|
8 |
+
"""
|
9 |
+
return f"_rotary_5yzc45v7kk3yu::{op_name}"
|
build/torch25-cxx11-cu118-x86_64-linux/rotary/_rotary_5yzc45v7kk3yu.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:10612c2e12298dc582d6b5bb4f5c03f0c22de9ce9dc1c2722e0ec8a46380234c
|
3 |
+
size 4006264
|
build/torch25-cxx11-cu121-x86_64-linux/rotary/__init__.py
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import Tuple
|
2 |
+
import torch
|
3 |
+
|
4 |
+
from ._ops import ops
|
5 |
+
|
6 |
+
|
7 |
+
def apply_rotary(
|
8 |
+
x1: torch.Tensor,
|
9 |
+
x2: torch.Tensor,
|
10 |
+
cos: torch.Tensor,
|
11 |
+
sin: torch.Tensor,
|
12 |
+
out1: torch.Tensor,
|
13 |
+
out2: torch.Tensor,
|
14 |
+
conj: bool,
|
15 |
+
):
|
16 |
+
ops.apply_rotary(x1, x2, cos, sin, out1, out2, conj)
|
17 |
+
|
18 |
+
|
19 |
+
__all__ = ["apply_rotary"]
|
build/torch25-cxx11-cu121-x86_64-linux/rotary/_ops.py
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from . import _rotary_tbiepw2a2ep3e
|
3 |
+
ops = torch.ops._rotary_tbiepw2a2ep3e
|
4 |
+
|
5 |
+
def add_op_namespace_prefix(op_name: str):
|
6 |
+
"""
|
7 |
+
Prefix op by namespace.
|
8 |
+
"""
|
9 |
+
return f"_rotary_tbiepw2a2ep3e::{op_name}"
|
build/torch25-cxx11-cu121-x86_64-linux/rotary/_rotary_tbiepw2a2ep3e.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:011fa4e0d1ab3b26f540a6265c1d875ed32e6bfa676ee5a511cb498c03e88f46
|
3 |
+
size 3971536
|
build/torch25-cxx11-cu124-x86_64-linux/rotary/__init__.py
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import Tuple
|
2 |
+
import torch
|
3 |
+
|
4 |
+
from ._ops import ops
|
5 |
+
|
6 |
+
|
7 |
+
def apply_rotary(
|
8 |
+
x1: torch.Tensor,
|
9 |
+
x2: torch.Tensor,
|
10 |
+
cos: torch.Tensor,
|
11 |
+
sin: torch.Tensor,
|
12 |
+
out1: torch.Tensor,
|
13 |
+
out2: torch.Tensor,
|
14 |
+
conj: bool,
|
15 |
+
):
|
16 |
+
ops.apply_rotary(x1, x2, cos, sin, out1, out2, conj)
|
17 |
+
|
18 |
+
|
19 |
+
__all__ = ["apply_rotary"]
|
build/torch25-cxx11-cu124-x86_64-linux/rotary/_ops.py
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from . import _rotary_6w5syhrhmerj6
|
3 |
+
ops = torch.ops._rotary_6w5syhrhmerj6
|
4 |
+
|
5 |
+
def add_op_namespace_prefix(op_name: str):
|
6 |
+
"""
|
7 |
+
Prefix op by namespace.
|
8 |
+
"""
|
9 |
+
return f"_rotary_6w5syhrhmerj6::{op_name}"
|
build/torch25-cxx11-cu124-x86_64-linux/rotary/_rotary_6w5syhrhmerj6.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ac12c6e9353d38c4284d7f336e33a7fb10ce8baa41fd664d19d83c44bea6ef45
|
3 |
+
size 4037768
|
build/torch25-cxx98-cu118-x86_64-linux/rotary/__init__.py
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import Tuple
|
2 |
+
import torch
|
3 |
+
|
4 |
+
from ._ops import ops
|
5 |
+
|
6 |
+
|
7 |
+
def apply_rotary(
|
8 |
+
x1: torch.Tensor,
|
9 |
+
x2: torch.Tensor,
|
10 |
+
cos: torch.Tensor,
|
11 |
+
sin: torch.Tensor,
|
12 |
+
out1: torch.Tensor,
|
13 |
+
out2: torch.Tensor,
|
14 |
+
conj: bool,
|
15 |
+
):
|
16 |
+
ops.apply_rotary(x1, x2, cos, sin, out1, out2, conj)
|
17 |
+
|
18 |
+
|
19 |
+
__all__ = ["apply_rotary"]
|
build/torch25-cxx98-cu118-x86_64-linux/rotary/_ops.py
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from . import _rotary_joujmbgvsytzg
|
3 |
+
ops = torch.ops._rotary_joujmbgvsytzg
|
4 |
+
|
5 |
+
def add_op_namespace_prefix(op_name: str):
|
6 |
+
"""
|
7 |
+
Prefix op by namespace.
|
8 |
+
"""
|
9 |
+
return f"_rotary_joujmbgvsytzg::{op_name}"
|
build/torch25-cxx98-cu118-x86_64-linux/rotary/_rotary_joujmbgvsytzg.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3678ca0157d21b376dcba6e797c2668bfb6de23aae45f5596cbe37b1e4758977
|
3 |
+
size 3994592
|
build/torch25-cxx98-cu121-x86_64-linux/rotary/__init__.py
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import Tuple
|
2 |
+
import torch
|
3 |
+
|
4 |
+
from ._ops import ops
|
5 |
+
|
6 |
+
|
7 |
+
def apply_rotary(
|
8 |
+
x1: torch.Tensor,
|
9 |
+
x2: torch.Tensor,
|
10 |
+
cos: torch.Tensor,
|
11 |
+
sin: torch.Tensor,
|
12 |
+
out1: torch.Tensor,
|
13 |
+
out2: torch.Tensor,
|
14 |
+
conj: bool,
|
15 |
+
):
|
16 |
+
ops.apply_rotary(x1, x2, cos, sin, out1, out2, conj)
|
17 |
+
|
18 |
+
|
19 |
+
__all__ = ["apply_rotary"]
|
build/torch25-cxx98-cu121-x86_64-linux/rotary/_ops.py
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from . import _rotary_mi2o7e7sishyw
|
3 |
+
ops = torch.ops._rotary_mi2o7e7sishyw
|
4 |
+
|
5 |
+
def add_op_namespace_prefix(op_name: str):
|
6 |
+
"""
|
7 |
+
Prefix op by namespace.
|
8 |
+
"""
|
9 |
+
return f"_rotary_mi2o7e7sishyw::{op_name}"
|
build/torch25-cxx98-cu121-x86_64-linux/rotary/_rotary_mi2o7e7sishyw.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:58ad7b1d6576bf63104b4bc0d621e4db61f7f0cca90f53300a76210f85e75647
|
3 |
+
size 3964000
|
build/torch25-cxx98-cu124-x86_64-linux/rotary/__init__.py
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import Tuple
|
2 |
+
import torch
|
3 |
+
|
4 |
+
from ._ops import ops
|
5 |
+
|
6 |
+
|
7 |
+
def apply_rotary(
|
8 |
+
x1: torch.Tensor,
|
9 |
+
x2: torch.Tensor,
|
10 |
+
cos: torch.Tensor,
|
11 |
+
sin: torch.Tensor,
|
12 |
+
out1: torch.Tensor,
|
13 |
+
out2: torch.Tensor,
|
14 |
+
conj: bool,
|
15 |
+
):
|
16 |
+
ops.apply_rotary(x1, x2, cos, sin, out1, out2, conj)
|
17 |
+
|
18 |
+
|
19 |
+
__all__ = ["apply_rotary"]
|
build/torch25-cxx98-cu124-x86_64-linux/rotary/_ops.py
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from . import _rotary_rngiohfhfwuge
|
3 |
+
ops = torch.ops._rotary_rngiohfhfwuge
|
4 |
+
|
5 |
+
def add_op_namespace_prefix(op_name: str):
|
6 |
+
"""
|
7 |
+
Prefix op by namespace.
|
8 |
+
"""
|
9 |
+
return f"_rotary_rngiohfhfwuge::{op_name}"
|
build/torch25-cxx98-cu124-x86_64-linux/rotary/_rotary_rngiohfhfwuge.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:0e356b8577926664f666a7f2116301350f187796f8eff0ee2455ab557f8fef82
|
3 |
+
size 4030264
|
build/torch26-cxx11-cu118-x86_64-linux/rotary/__init__.py
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import Tuple
|
2 |
+
import torch
|
3 |
+
|
4 |
+
from ._ops import ops
|
5 |
+
|
6 |
+
|
7 |
+
def apply_rotary(
|
8 |
+
x1: torch.Tensor,
|
9 |
+
x2: torch.Tensor,
|
10 |
+
cos: torch.Tensor,
|
11 |
+
sin: torch.Tensor,
|
12 |
+
out1: torch.Tensor,
|
13 |
+
out2: torch.Tensor,
|
14 |
+
conj: bool,
|
15 |
+
):
|
16 |
+
ops.apply_rotary(x1, x2, cos, sin, out1, out2, conj)
|
17 |
+
|
18 |
+
|
19 |
+
__all__ = ["apply_rotary"]
|
build/torch26-cxx11-cu118-x86_64-linux/rotary/_ops.py
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from . import _rotary_alv7mzltcxxpq
|
3 |
+
ops = torch.ops._rotary_alv7mzltcxxpq
|
4 |
+
|
5 |
+
def add_op_namespace_prefix(op_name: str):
|
6 |
+
"""
|
7 |
+
Prefix op by namespace.
|
8 |
+
"""
|
9 |
+
return f"_rotary_alv7mzltcxxpq::{op_name}"
|
build/torch26-cxx11-cu118-x86_64-linux/rotary/_rotary_alv7mzltcxxpq.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ed401bda2448c0b5f7ba58134502044ff84174c821cb043bc0347abb88b5c032
|
3 |
+
size 4002064
|
build/torch26-cxx11-cu124-x86_64-linux/rotary/__init__.py
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import Tuple
|
2 |
+
import torch
|
3 |
+
|
4 |
+
from ._ops import ops
|
5 |
+
|
6 |
+
|
7 |
+
def apply_rotary(
|
8 |
+
x1: torch.Tensor,
|
9 |
+
x2: torch.Tensor,
|
10 |
+
cos: torch.Tensor,
|
11 |
+
sin: torch.Tensor,
|
12 |
+
out1: torch.Tensor,
|
13 |
+
out2: torch.Tensor,
|
14 |
+
conj: bool,
|
15 |
+
):
|
16 |
+
ops.apply_rotary(x1, x2, cos, sin, out1, out2, conj)
|
17 |
+
|
18 |
+
|
19 |
+
__all__ = ["apply_rotary"]
|
build/torch26-cxx11-cu124-x86_64-linux/rotary/_ops.py
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from . import _rotary_c4eyapeep6gty
|
3 |
+
ops = torch.ops._rotary_c4eyapeep6gty
|
4 |
+
|
5 |
+
def add_op_namespace_prefix(op_name: str):
|
6 |
+
"""
|
7 |
+
Prefix op by namespace.
|
8 |
+
"""
|
9 |
+
return f"_rotary_c4eyapeep6gty::{op_name}"
|
build/torch26-cxx11-cu124-x86_64-linux/rotary/_rotary_c4eyapeep6gty.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e1b95eacddea0ae6a837ded34f0e6aa5493e1b89e21440a591e4947d84e5a49b
|
3 |
+
size 4029480
|
build/torch26-cxx11-cu126-x86_64-linux/rotary/__init__.py
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import Tuple
|
2 |
+
import torch
|
3 |
+
|
4 |
+
from ._ops import ops
|
5 |
+
|
6 |
+
|
7 |
+
def apply_rotary(
|
8 |
+
x1: torch.Tensor,
|
9 |
+
x2: torch.Tensor,
|
10 |
+
cos: torch.Tensor,
|
11 |
+
sin: torch.Tensor,
|
12 |
+
out1: torch.Tensor,
|
13 |
+
out2: torch.Tensor,
|
14 |
+
conj: bool,
|
15 |
+
):
|
16 |
+
ops.apply_rotary(x1, x2, cos, sin, out1, out2, conj)
|
17 |
+
|
18 |
+
|
19 |
+
__all__ = ["apply_rotary"]
|
build/torch26-cxx11-cu126-x86_64-linux/rotary/_ops.py
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from . import _rotary_lodp6xeztste6
|
3 |
+
ops = torch.ops._rotary_lodp6xeztste6
|
4 |
+
|
5 |
+
def add_op_namespace_prefix(op_name: str):
|
6 |
+
"""
|
7 |
+
Prefix op by namespace.
|
8 |
+
"""
|
9 |
+
return f"_rotary_lodp6xeztste6::{op_name}"
|
build/torch26-cxx11-cu126-x86_64-linux/rotary/_rotary_lodp6xeztste6.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9d63ff54a10967f2ddde757581332486ca3be1de0b5f36005427ac9845e99d6b
|
3 |
+
size 4039392
|
build/torch26-cxx98-cu118-x86_64-linux/rotary/__init__.py
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import Tuple
|
2 |
+
import torch
|
3 |
+
|
4 |
+
from ._ops import ops
|
5 |
+
|
6 |
+
|
7 |
+
def apply_rotary(
|
8 |
+
x1: torch.Tensor,
|
9 |
+
x2: torch.Tensor,
|
10 |
+
cos: torch.Tensor,
|
11 |
+
sin: torch.Tensor,
|
12 |
+
out1: torch.Tensor,
|
13 |
+
out2: torch.Tensor,
|
14 |
+
conj: bool,
|
15 |
+
):
|
16 |
+
ops.apply_rotary(x1, x2, cos, sin, out1, out2, conj)
|
17 |
+
|
18 |
+
|
19 |
+
__all__ = ["apply_rotary"]
|
build/torch26-cxx98-cu118-x86_64-linux/rotary/_ops.py
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from . import _rotary_z27mls7mz4e7m
|
3 |
+
ops = torch.ops._rotary_z27mls7mz4e7m
|
4 |
+
|
5 |
+
def add_op_namespace_prefix(op_name: str):
|
6 |
+
"""
|
7 |
+
Prefix op by namespace.
|
8 |
+
"""
|
9 |
+
return f"_rotary_z27mls7mz4e7m::{op_name}"
|
build/torch26-cxx98-cu118-x86_64-linux/rotary/_rotary_z27mls7mz4e7m.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:78e48f0e972b845d7667b1e1e6c807d8ccbe0e12442d227240a066ab430951a5
|
3 |
+
size 3990448
|
build/torch26-cxx98-cu124-x86_64-linux/rotary/__init__.py
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import Tuple
|
2 |
+
import torch
|
3 |
+
|
4 |
+
from ._ops import ops
|
5 |
+
|
6 |
+
|
7 |
+
def apply_rotary(
|
8 |
+
x1: torch.Tensor,
|
9 |
+
x2: torch.Tensor,
|
10 |
+
cos: torch.Tensor,
|
11 |
+
sin: torch.Tensor,
|
12 |
+
out1: torch.Tensor,
|
13 |
+
out2: torch.Tensor,
|
14 |
+
conj: bool,
|
15 |
+
):
|
16 |
+
ops.apply_rotary(x1, x2, cos, sin, out1, out2, conj)
|
17 |
+
|
18 |
+
|
19 |
+
__all__ = ["apply_rotary"]
|
build/torch26-cxx98-cu124-x86_64-linux/rotary/_ops.py
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from . import _rotary_3bktke4p3hz3a
|
3 |
+
ops = torch.ops._rotary_3bktke4p3hz3a
|
4 |
+
|
5 |
+
def add_op_namespace_prefix(op_name: str):
|
6 |
+
"""
|
7 |
+
Prefix op by namespace.
|
8 |
+
"""
|
9 |
+
return f"_rotary_3bktke4p3hz3a::{op_name}"
|
build/torch26-cxx98-cu124-x86_64-linux/rotary/_rotary_3bktke4p3hz3a.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:fb5710d06ef1c68ecde248253c92eea5073ca98411b270accdd13f73470bdc66
|
3 |
+
size 4022072
|
build/torch26-cxx98-cu126-x86_64-linux/rotary/__init__.py
ADDED
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from typing import Tuple
|
2 |
+
import torch
|
3 |
+
|
4 |
+
from ._ops import ops
|
5 |
+
|
6 |
+
|
7 |
+
def apply_rotary(
|
8 |
+
x1: torch.Tensor,
|
9 |
+
x2: torch.Tensor,
|
10 |
+
cos: torch.Tensor,
|
11 |
+
sin: torch.Tensor,
|
12 |
+
out1: torch.Tensor,
|
13 |
+
out2: torch.Tensor,
|
14 |
+
conj: bool,
|
15 |
+
):
|
16 |
+
ops.apply_rotary(x1, x2, cos, sin, out1, out2, conj)
|
17 |
+
|
18 |
+
|
19 |
+
__all__ = ["apply_rotary"]
|
build/torch26-cxx98-cu126-x86_64-linux/rotary/_ops.py
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from . import _rotary_fvednlzeqgg5s
|
3 |
+
ops = torch.ops._rotary_fvednlzeqgg5s
|
4 |
+
|
5 |
+
def add_op_namespace_prefix(op_name: str):
|
6 |
+
"""
|
7 |
+
Prefix op by namespace.
|
8 |
+
"""
|
9 |
+
return f"_rotary_fvednlzeqgg5s::{op_name}"
|
build/torch26-cxx98-cu126-x86_64-linux/rotary/_rotary_fvednlzeqgg5s.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:923d3a0436f6279d7709a48cf30b98598243706cbd14fe1010361c2d2a1791de
|
3 |
+
size 4031984
|