Kernels:
Trusted publisher
Uploaded using `kernel-builder`.
Browse files- build/torch210-cxx11-cu126-aarch64-linux/__init__.py +4 -0
- build/torch210-cxx11-cu126-aarch64-linux/_ops.py +3 -3
- build/torch210-cxx11-cu126-aarch64-linux/{_rotary_cuda_725ebfc.abi3.so → _rotary_cuda_52256cf.abi3.so} +1 -1
- build/torch210-cxx11-cu126-aarch64-linux/metadata.json +1 -1
- build/torch210-cxx11-cu128-aarch64-linux/__init__.py +4 -0
- build/torch210-cxx11-cu128-aarch64-linux/_ops.py +3 -3
- build/torch210-cxx11-cu128-aarch64-linux/{_rotary_cuda_725ebfc.abi3.so → _rotary_cuda_52256cf.abi3.so} +1 -1
- build/torch210-cxx11-cu128-aarch64-linux/metadata.json +1 -1
- build/torch210-cxx11-cu130-aarch64-linux/__init__.py +4 -0
- build/torch210-cxx11-cu130-aarch64-linux/_ops.py +3 -3
- build/torch210-cxx11-cu130-aarch64-linux/{_rotary_cuda_725ebfc.abi3.so → _rotary_cuda_52256cf.abi3.so} +1 -1
- build/torch210-cxx11-cu130-aarch64-linux/metadata.json +1 -1
- build/torch211-cxx11-cu126-aarch64-linux/__init__.py +4 -0
- build/torch211-cxx11-cu126-aarch64-linux/_ops.py +3 -3
- build/torch211-cxx11-cu126-aarch64-linux/{_rotary_cuda_725ebfc.abi3.so → _rotary_cuda_52256cf.abi3.so} +1 -1
- build/torch211-cxx11-cu126-aarch64-linux/metadata.json +1 -1
- build/torch211-cxx11-cu128-aarch64-linux/__init__.py +4 -0
- build/torch211-cxx11-cu128-aarch64-linux/_ops.py +3 -3
- build/torch211-cxx11-cu128-aarch64-linux/{_rotary_cuda_725ebfc.abi3.so → _rotary_cuda_52256cf.abi3.so} +1 -1
- build/torch211-cxx11-cu128-aarch64-linux/metadata.json +1 -1
- build/torch211-cxx11-cu130-aarch64-linux/__init__.py +4 -0
- build/torch211-cxx11-cu130-aarch64-linux/_ops.py +3 -3
- build/torch211-cxx11-cu130-aarch64-linux/{_rotary_cuda_725ebfc.abi3.so → _rotary_cuda_52256cf.abi3.so} +1 -1
- build/torch211-cxx11-cu130-aarch64-linux/metadata.json +1 -1
- build/torch212-cxx11-cu126-aarch64-linux/__init__.py +4 -0
- build/torch212-cxx11-cu126-aarch64-linux/_ops.py +3 -3
- build/torch212-cxx11-cu126-aarch64-linux/{_rotary_cuda_725ebfc.abi3.so → _rotary_cuda_52256cf.abi3.so} +1 -1
- build/torch212-cxx11-cu126-aarch64-linux/metadata.json +1 -1
- build/torch212-cxx11-cu130-aarch64-linux/__init__.py +4 -0
- build/torch212-cxx11-cu130-aarch64-linux/_ops.py +3 -3
- build/torch212-cxx11-cu130-aarch64-linux/{_rotary_cuda_725ebfc.abi3.so → _rotary_cuda_52256cf.abi3.so} +1 -1
- build/torch212-cxx11-cu130-aarch64-linux/metadata.json +1 -1
- build/torch212-cxx11-cu132-aarch64-linux/__init__.py +4 -0
- build/torch212-cxx11-cu132-aarch64-linux/_ops.py +3 -3
- build/torch212-cxx11-cu132-aarch64-linux/{_rotary_cuda_725ebfc.abi3.so → _rotary_cuda_52256cf.abi3.so} +1 -1
- build/torch212-cxx11-cu132-aarch64-linux/metadata.json +1 -1
build/torch210-cxx11-cu126-aarch64-linux/__init__.py
CHANGED
|
@@ -49,4 +49,8 @@ def apply_rotary_transformers(
|
|
| 49 |
return q_rotated, k_rotated
|
| 50 |
|
| 51 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 52 |
__all__ = ["apply_rotary", "apply_rotary_transformers"]
|
|
|
|
| 49 |
return q_rotated, k_rotated
|
| 50 |
|
| 51 |
|
| 52 |
+
# Add torch compile support for functions
|
| 53 |
+
apply_rotary_transformers.can_torch_compile = True
|
| 54 |
+
|
| 55 |
+
|
| 56 |
__all__ = ["apply_rotary", "apply_rotary_transformers"]
|
build/torch210-cxx11-cu126-aarch64-linux/_ops.py
CHANGED
|
@@ -1,9 +1,9 @@
|
|
| 1 |
import torch
|
| 2 |
-
from . import
|
| 3 |
-
ops = torch.ops.
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
-
return f"
|
|
|
|
| 1 |
import torch
|
| 2 |
+
from . import _rotary_cuda_52256cf
|
| 3 |
+
ops = torch.ops._rotary_cuda_52256cf
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
+
return f"_rotary_cuda_52256cf::{op_name}"
|
build/torch210-cxx11-cu126-aarch64-linux/{_rotary_cuda_725ebfc.abi3.so → _rotary_cuda_52256cf.abi3.so}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 8282888
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ae934eaa40461a94ae95f120fb9765b44f14626d84d8af79d787c4dd7af9d9f3
|
| 3 |
size 8282888
|
build/torch210-cxx11-cu126-aarch64-linux/metadata.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"name": "rotary",
|
| 3 |
-
"id": "
|
| 4 |
"version": 1,
|
| 5 |
"license": "BSD-3-Clause",
|
| 6 |
"python-depends": [],
|
|
|
|
| 1 |
{
|
| 2 |
"name": "rotary",
|
| 3 |
+
"id": "_rotary_cuda_52256cf",
|
| 4 |
"version": 1,
|
| 5 |
"license": "BSD-3-Clause",
|
| 6 |
"python-depends": [],
|
build/torch210-cxx11-cu128-aarch64-linux/__init__.py
CHANGED
|
@@ -49,4 +49,8 @@ def apply_rotary_transformers(
|
|
| 49 |
return q_rotated, k_rotated
|
| 50 |
|
| 51 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 52 |
__all__ = ["apply_rotary", "apply_rotary_transformers"]
|
|
|
|
| 49 |
return q_rotated, k_rotated
|
| 50 |
|
| 51 |
|
| 52 |
+
# Add torch compile support for functions
|
| 53 |
+
apply_rotary_transformers.can_torch_compile = True
|
| 54 |
+
|
| 55 |
+
|
| 56 |
__all__ = ["apply_rotary", "apply_rotary_transformers"]
|
build/torch210-cxx11-cu128-aarch64-linux/_ops.py
CHANGED
|
@@ -1,9 +1,9 @@
|
|
| 1 |
import torch
|
| 2 |
-
from . import
|
| 3 |
-
ops = torch.ops.
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
-
return f"
|
|
|
|
| 1 |
import torch
|
| 2 |
+
from . import _rotary_cuda_52256cf
|
| 3 |
+
ops = torch.ops._rotary_cuda_52256cf
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
+
return f"_rotary_cuda_52256cf::{op_name}"
|
build/torch210-cxx11-cu128-aarch64-linux/{_rotary_cuda_725ebfc.abi3.so → _rotary_cuda_52256cf.abi3.so}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 12019200
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:29a713c01a19d2103122816b667e645d373447ec6d6350a2f59a00aca1aaca1f
|
| 3 |
size 12019200
|
build/torch210-cxx11-cu128-aarch64-linux/metadata.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"name": "rotary",
|
| 3 |
-
"id": "
|
| 4 |
"version": 1,
|
| 5 |
"license": "BSD-3-Clause",
|
| 6 |
"python-depends": [],
|
|
|
|
| 1 |
{
|
| 2 |
"name": "rotary",
|
| 3 |
+
"id": "_rotary_cuda_52256cf",
|
| 4 |
"version": 1,
|
| 5 |
"license": "BSD-3-Clause",
|
| 6 |
"python-depends": [],
|
build/torch210-cxx11-cu130-aarch64-linux/__init__.py
CHANGED
|
@@ -49,4 +49,8 @@ def apply_rotary_transformers(
|
|
| 49 |
return q_rotated, k_rotated
|
| 50 |
|
| 51 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 52 |
__all__ = ["apply_rotary", "apply_rotary_transformers"]
|
|
|
|
| 49 |
return q_rotated, k_rotated
|
| 50 |
|
| 51 |
|
| 52 |
+
# Add torch compile support for functions
|
| 53 |
+
apply_rotary_transformers.can_torch_compile = True
|
| 54 |
+
|
| 55 |
+
|
| 56 |
__all__ = ["apply_rotary", "apply_rotary_transformers"]
|
build/torch210-cxx11-cu130-aarch64-linux/_ops.py
CHANGED
|
@@ -1,9 +1,9 @@
|
|
| 1 |
import torch
|
| 2 |
-
from . import
|
| 3 |
-
ops = torch.ops.
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
-
return f"
|
|
|
|
| 1 |
import torch
|
| 2 |
+
from . import _rotary_cuda_52256cf
|
| 3 |
+
ops = torch.ops._rotary_cuda_52256cf
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
+
return f"_rotary_cuda_52256cf::{op_name}"
|
build/torch210-cxx11-cu130-aarch64-linux/{_rotary_cuda_725ebfc.abi3.so → _rotary_cuda_52256cf.abi3.so}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 10412048
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1747d2d5a7619865279cca31f4f662d23319395827458e4f87022a1b0dc3f250
|
| 3 |
size 10412048
|
build/torch210-cxx11-cu130-aarch64-linux/metadata.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"name": "rotary",
|
| 3 |
-
"id": "
|
| 4 |
"version": 1,
|
| 5 |
"license": "BSD-3-Clause",
|
| 6 |
"python-depends": [],
|
|
|
|
| 1 |
{
|
| 2 |
"name": "rotary",
|
| 3 |
+
"id": "_rotary_cuda_52256cf",
|
| 4 |
"version": 1,
|
| 5 |
"license": "BSD-3-Clause",
|
| 6 |
"python-depends": [],
|
build/torch211-cxx11-cu126-aarch64-linux/__init__.py
CHANGED
|
@@ -49,4 +49,8 @@ def apply_rotary_transformers(
|
|
| 49 |
return q_rotated, k_rotated
|
| 50 |
|
| 51 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 52 |
__all__ = ["apply_rotary", "apply_rotary_transformers"]
|
|
|
|
| 49 |
return q_rotated, k_rotated
|
| 50 |
|
| 51 |
|
| 52 |
+
# Add torch compile support for functions
|
| 53 |
+
apply_rotary_transformers.can_torch_compile = True
|
| 54 |
+
|
| 55 |
+
|
| 56 |
__all__ = ["apply_rotary", "apply_rotary_transformers"]
|
build/torch211-cxx11-cu126-aarch64-linux/_ops.py
CHANGED
|
@@ -1,9 +1,9 @@
|
|
| 1 |
import torch
|
| 2 |
-
from . import
|
| 3 |
-
ops = torch.ops.
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
-
return f"
|
|
|
|
| 1 |
import torch
|
| 2 |
+
from . import _rotary_cuda_52256cf
|
| 3 |
+
ops = torch.ops._rotary_cuda_52256cf
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
+
return f"_rotary_cuda_52256cf::{op_name}"
|
build/torch211-cxx11-cu126-aarch64-linux/{_rotary_cuda_725ebfc.abi3.so → _rotary_cuda_52256cf.abi3.so}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 8279200
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:2ee786ca03780536ed045a299f4be20cb73fbe888bf5d51970fd5a201efcbd9d
|
| 3 |
size 8279200
|
build/torch211-cxx11-cu126-aarch64-linux/metadata.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"name": "rotary",
|
| 3 |
-
"id": "
|
| 4 |
"version": 1,
|
| 5 |
"license": "BSD-3-Clause",
|
| 6 |
"python-depends": [],
|
|
|
|
| 1 |
{
|
| 2 |
"name": "rotary",
|
| 3 |
+
"id": "_rotary_cuda_52256cf",
|
| 4 |
"version": 1,
|
| 5 |
"license": "BSD-3-Clause",
|
| 6 |
"python-depends": [],
|
build/torch211-cxx11-cu128-aarch64-linux/__init__.py
CHANGED
|
@@ -49,4 +49,8 @@ def apply_rotary_transformers(
|
|
| 49 |
return q_rotated, k_rotated
|
| 50 |
|
| 51 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 52 |
__all__ = ["apply_rotary", "apply_rotary_transformers"]
|
|
|
|
| 49 |
return q_rotated, k_rotated
|
| 50 |
|
| 51 |
|
| 52 |
+
# Add torch compile support for functions
|
| 53 |
+
apply_rotary_transformers.can_torch_compile = True
|
| 54 |
+
|
| 55 |
+
|
| 56 |
__all__ = ["apply_rotary", "apply_rotary_transformers"]
|
build/torch211-cxx11-cu128-aarch64-linux/_ops.py
CHANGED
|
@@ -1,9 +1,9 @@
|
|
| 1 |
import torch
|
| 2 |
-
from . import
|
| 3 |
-
ops = torch.ops.
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
-
return f"
|
|
|
|
| 1 |
import torch
|
| 2 |
+
from . import _rotary_cuda_52256cf
|
| 3 |
+
ops = torch.ops._rotary_cuda_52256cf
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
+
return f"_rotary_cuda_52256cf::{op_name}"
|
build/torch211-cxx11-cu128-aarch64-linux/{_rotary_cuda_725ebfc.abi3.so → _rotary_cuda_52256cf.abi3.so}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 12015512
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f567b4fb5db7d67afd1e6218c145f35966c0e202e4547b796e11bf1c4b9fe1e4
|
| 3 |
size 12015512
|
build/torch211-cxx11-cu128-aarch64-linux/metadata.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"name": "rotary",
|
| 3 |
-
"id": "
|
| 4 |
"version": 1,
|
| 5 |
"license": "BSD-3-Clause",
|
| 6 |
"python-depends": [],
|
|
|
|
| 1 |
{
|
| 2 |
"name": "rotary",
|
| 3 |
+
"id": "_rotary_cuda_52256cf",
|
| 4 |
"version": 1,
|
| 5 |
"license": "BSD-3-Clause",
|
| 6 |
"python-depends": [],
|
build/torch211-cxx11-cu130-aarch64-linux/__init__.py
CHANGED
|
@@ -49,4 +49,8 @@ def apply_rotary_transformers(
|
|
| 49 |
return q_rotated, k_rotated
|
| 50 |
|
| 51 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 52 |
__all__ = ["apply_rotary", "apply_rotary_transformers"]
|
|
|
|
| 49 |
return q_rotated, k_rotated
|
| 50 |
|
| 51 |
|
| 52 |
+
# Add torch compile support for functions
|
| 53 |
+
apply_rotary_transformers.can_torch_compile = True
|
| 54 |
+
|
| 55 |
+
|
| 56 |
__all__ = ["apply_rotary", "apply_rotary_transformers"]
|
build/torch211-cxx11-cu130-aarch64-linux/_ops.py
CHANGED
|
@@ -1,9 +1,9 @@
|
|
| 1 |
import torch
|
| 2 |
-
from . import
|
| 3 |
-
ops = torch.ops.
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
-
return f"
|
|
|
|
| 1 |
import torch
|
| 2 |
+
from . import _rotary_cuda_52256cf
|
| 3 |
+
ops = torch.ops._rotary_cuda_52256cf
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
+
return f"_rotary_cuda_52256cf::{op_name}"
|
build/torch211-cxx11-cu130-aarch64-linux/{_rotary_cuda_725ebfc.abi3.so → _rotary_cuda_52256cf.abi3.so}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 10408368
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5467f9f40a5f222aed4c83630cf94c1e3cc3ce38c6d31d0703b94a7897030e76
|
| 3 |
size 10408368
|
build/torch211-cxx11-cu130-aarch64-linux/metadata.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"name": "rotary",
|
| 3 |
-
"id": "
|
| 4 |
"version": 1,
|
| 5 |
"license": "BSD-3-Clause",
|
| 6 |
"python-depends": [],
|
|
|
|
| 1 |
{
|
| 2 |
"name": "rotary",
|
| 3 |
+
"id": "_rotary_cuda_52256cf",
|
| 4 |
"version": 1,
|
| 5 |
"license": "BSD-3-Clause",
|
| 6 |
"python-depends": [],
|
build/torch212-cxx11-cu126-aarch64-linux/__init__.py
CHANGED
|
@@ -49,4 +49,8 @@ def apply_rotary_transformers(
|
|
| 49 |
return q_rotated, k_rotated
|
| 50 |
|
| 51 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 52 |
__all__ = ["apply_rotary", "apply_rotary_transformers"]
|
|
|
|
| 49 |
return q_rotated, k_rotated
|
| 50 |
|
| 51 |
|
| 52 |
+
# Add torch compile support for functions
|
| 53 |
+
apply_rotary_transformers.can_torch_compile = True
|
| 54 |
+
|
| 55 |
+
|
| 56 |
__all__ = ["apply_rotary", "apply_rotary_transformers"]
|
build/torch212-cxx11-cu126-aarch64-linux/_ops.py
CHANGED
|
@@ -1,9 +1,9 @@
|
|
| 1 |
import torch
|
| 2 |
-
from . import
|
| 3 |
-
ops = torch.ops.
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
-
return f"
|
|
|
|
| 1 |
import torch
|
| 2 |
+
from . import _rotary_cuda_52256cf
|
| 3 |
+
ops = torch.ops._rotary_cuda_52256cf
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
+
return f"_rotary_cuda_52256cf::{op_name}"
|
build/torch212-cxx11-cu126-aarch64-linux/{_rotary_cuda_725ebfc.abi3.so → _rotary_cuda_52256cf.abi3.so}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 8282736
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:600aa373decd690bd113de58af203730304021b49c7f0b95c4476d33385e1580
|
| 3 |
size 8282736
|
build/torch212-cxx11-cu126-aarch64-linux/metadata.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"name": "rotary",
|
| 3 |
-
"id": "
|
| 4 |
"version": 1,
|
| 5 |
"license": "BSD-3-Clause",
|
| 6 |
"python-depends": [],
|
|
|
|
| 1 |
{
|
| 2 |
"name": "rotary",
|
| 3 |
+
"id": "_rotary_cuda_52256cf",
|
| 4 |
"version": 1,
|
| 5 |
"license": "BSD-3-Clause",
|
| 6 |
"python-depends": [],
|
build/torch212-cxx11-cu130-aarch64-linux/__init__.py
CHANGED
|
@@ -49,4 +49,8 @@ def apply_rotary_transformers(
|
|
| 49 |
return q_rotated, k_rotated
|
| 50 |
|
| 51 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 52 |
__all__ = ["apply_rotary", "apply_rotary_transformers"]
|
|
|
|
| 49 |
return q_rotated, k_rotated
|
| 50 |
|
| 51 |
|
| 52 |
+
# Add torch compile support for functions
|
| 53 |
+
apply_rotary_transformers.can_torch_compile = True
|
| 54 |
+
|
| 55 |
+
|
| 56 |
__all__ = ["apply_rotary", "apply_rotary_transformers"]
|
build/torch212-cxx11-cu130-aarch64-linux/_ops.py
CHANGED
|
@@ -1,9 +1,9 @@
|
|
| 1 |
import torch
|
| 2 |
-
from . import
|
| 3 |
-
ops = torch.ops.
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
-
return f"
|
|
|
|
| 1 |
import torch
|
| 2 |
+
from . import _rotary_cuda_52256cf
|
| 3 |
+
ops = torch.ops._rotary_cuda_52256cf
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
+
return f"_rotary_cuda_52256cf::{op_name}"
|
build/torch212-cxx11-cu130-aarch64-linux/{_rotary_cuda_725ebfc.abi3.so → _rotary_cuda_52256cf.abi3.so}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 10411904
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d9895791dcca122324cc16b1d0a3163fd237039d36618ea62b8a4a70aaa5c4d5
|
| 3 |
size 10411904
|
build/torch212-cxx11-cu130-aarch64-linux/metadata.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"name": "rotary",
|
| 3 |
-
"id": "
|
| 4 |
"version": 1,
|
| 5 |
"license": "BSD-3-Clause",
|
| 6 |
"python-depends": [],
|
|
|
|
| 1 |
{
|
| 2 |
"name": "rotary",
|
| 3 |
+
"id": "_rotary_cuda_52256cf",
|
| 4 |
"version": 1,
|
| 5 |
"license": "BSD-3-Clause",
|
| 6 |
"python-depends": [],
|
build/torch212-cxx11-cu132-aarch64-linux/__init__.py
CHANGED
|
@@ -49,4 +49,8 @@ def apply_rotary_transformers(
|
|
| 49 |
return q_rotated, k_rotated
|
| 50 |
|
| 51 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 52 |
__all__ = ["apply_rotary", "apply_rotary_transformers"]
|
|
|
|
| 49 |
return q_rotated, k_rotated
|
| 50 |
|
| 51 |
|
| 52 |
+
# Add torch compile support for functions
|
| 53 |
+
apply_rotary_transformers.can_torch_compile = True
|
| 54 |
+
|
| 55 |
+
|
| 56 |
__all__ = ["apply_rotary", "apply_rotary_transformers"]
|
build/torch212-cxx11-cu132-aarch64-linux/_ops.py
CHANGED
|
@@ -1,9 +1,9 @@
|
|
| 1 |
import torch
|
| 2 |
-
from . import
|
| 3 |
-
ops = torch.ops.
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
-
return f"
|
|
|
|
| 1 |
import torch
|
| 2 |
+
from . import _rotary_cuda_52256cf
|
| 3 |
+
ops = torch.ops._rotary_cuda_52256cf
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
+
return f"_rotary_cuda_52256cf::{op_name}"
|
build/torch212-cxx11-cu132-aarch64-linux/{_rotary_cuda_725ebfc.abi3.so → _rotary_cuda_52256cf.abi3.so}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 10612968
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1d16e483132fff4fcf1edc6684f600ffc10e31870499da7086ad8bbe28f6830b
|
| 3 |
size 10612968
|
build/torch212-cxx11-cu132-aarch64-linux/metadata.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"name": "rotary",
|
| 3 |
-
"id": "
|
| 4 |
"version": 1,
|
| 5 |
"license": "BSD-3-Clause",
|
| 6 |
"python-depends": [],
|
|
|
|
| 1 |
{
|
| 2 |
"name": "rotary",
|
| 3 |
+
"id": "_rotary_cuda_52256cf",
|
| 4 |
"version": 1,
|
| 5 |
"license": "BSD-3-Clause",
|
| 6 |
"python-depends": [],
|