Build for Torch 2.8 (x86_64-linux)
Browse files- build/torch28-cxx11-cu126-x86_64-linux/rotary/__init__.py +19 -0
- build/torch28-cxx11-cu126-x86_64-linux/rotary/_ops.py +9 -0
- build/torch28-cxx11-cu126-x86_64-linux/rotary/_rotary_d5e8892.abi3.so +3 -0
- build/torch28-cxx11-cu128-x86_64-linux/rotary/__init__.py +19 -0
- build/torch28-cxx11-cu128-x86_64-linux/rotary/_ops.py +9 -0
- build/torch28-cxx11-cu128-x86_64-linux/rotary/_rotary_d5e8892.abi3.so +3 -0
- build/torch28-cxx11-cu129-x86_64-linux/rotary/__init__.py +19 -0
- build/torch28-cxx11-cu129-x86_64-linux/rotary/_ops.py +9 -0
- build/torch28-cxx11-cu129-x86_64-linux/rotary/_rotary_d5e8892.abi3.so +3 -0
build/torch28-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/torch28-cxx11-cu126-x86_64-linux/rotary/_ops.py
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from . import _rotary_d5e8892
|
3 |
+
ops = torch.ops._rotary_d5e8892
|
4 |
+
|
5 |
+
def add_op_namespace_prefix(op_name: str):
|
6 |
+
"""
|
7 |
+
Prefix op by namespace.
|
8 |
+
"""
|
9 |
+
return f"_rotary_d5e8892::{op_name}"
|
build/torch28-cxx11-cu126-x86_64-linux/rotary/_rotary_d5e8892.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:663ac605052413c175dcff4dec8545aa37ad67e8629002621741b89b182d98cd
|
3 |
+
size 6293168
|
build/torch28-cxx11-cu128-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/torch28-cxx11-cu128-x86_64-linux/rotary/_ops.py
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from . import _rotary_d5e8892
|
3 |
+
ops = torch.ops._rotary_d5e8892
|
4 |
+
|
5 |
+
def add_op_namespace_prefix(op_name: str):
|
6 |
+
"""
|
7 |
+
Prefix op by namespace.
|
8 |
+
"""
|
9 |
+
return f"_rotary_d5e8892::{op_name}"
|
build/torch28-cxx11-cu128-x86_64-linux/rotary/_rotary_d5e8892.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:264410fa13bca33e706d1c3eb12a2d966e8fa07e2b786cbd8332d462f4883d1a
|
3 |
+
size 10149824
|
build/torch28-cxx11-cu129-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/torch28-cxx11-cu129-x86_64-linux/rotary/_ops.py
ADDED
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from . import _rotary_d5e8892
|
3 |
+
ops = torch.ops._rotary_d5e8892
|
4 |
+
|
5 |
+
def add_op_namespace_prefix(op_name: str):
|
6 |
+
"""
|
7 |
+
Prefix op by namespace.
|
8 |
+
"""
|
9 |
+
return f"_rotary_d5e8892::{op_name}"
|
build/torch28-cxx11-cu129-x86_64-linux/rotary/_rotary_d5e8892.abi3.so
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:dd62535c2713d05e74f4b53c84564caeeba51aaf06f5fe59a3182b04a5ae3c5a
|
3 |
+
size 10169280
|