Instructions to use replicate/paged-attention with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Kernels
How to use replicate/paged-attention with Kernels:
# !pip install kernels from kernels import get_kernel # a version (or an explicit revision) is required; see the "Files and versions" tab for the available ones kernel = get_kernel("replicate/paged-attention", version=1) - Notebooks
- Google Colab
- Kaggle
Uploaded using `kernel-builder`.
Browse files- build/torch210-cxx11-cu126-aarch64-linux/_ops.py +3 -3
- build/torch210-cxx11-cu126-aarch64-linux/{_paged_attention_cuda_0278991.abi3.so → _paged_attention_cuda_04f8f6e.abi3.so} +1 -1
- build/torch210-cxx11-cu128-aarch64-linux/_ops.py +3 -3
- build/torch210-cxx11-cu128-aarch64-linux/{_paged_attention_cuda_0278991.abi3.so → _paged_attention_cuda_04f8f6e.abi3.so} +1 -1
- build/torch210-cxx11-cu130-aarch64-linux/_ops.py +3 -3
- build/torch210-cxx11-cu130-aarch64-linux/{_paged_attention_cuda_0278991.abi3.so → _paged_attention_cuda_04f8f6e.abi3.so} +1 -1
- build/torch211-cxx11-cu126-aarch64-linux/_ops.py +3 -3
- build/torch211-cxx11-cu126-aarch64-linux/{_paged_attention_cuda_0278991.abi3.so → _paged_attention_cuda_04f8f6e.abi3.so} +1 -1
- build/torch211-cxx11-cu128-aarch64-linux/_ops.py +3 -3
- build/torch211-cxx11-cu128-aarch64-linux/{_paged_attention_cuda_0278991.abi3.so → _paged_attention_cuda_04f8f6e.abi3.so} +1 -1
- build/torch211-cxx11-cu130-aarch64-linux/_ops.py +3 -3
- build/torch211-cxx11-cu130-aarch64-linux/{_paged_attention_cuda_0278991.abi3.so → _paged_attention_cuda_04f8f6e.abi3.so} +1 -1
- build/torch29-cxx11-cu129-aarch64-linux/_ops.py +3 -3
- build/torch29-cxx11-cu129-aarch64-linux/{_paged_attention_cuda_0278991.abi3.so → _paged_attention_cuda_04f8f6e.abi3.so} +1 -1
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 _paged_attention_cuda_04f8f6e
|
| 3 |
+
ops = torch.ops._paged_attention_cuda_04f8f6e
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
+
return f"_paged_attention_cuda_04f8f6e::{op_name}"
|
build/torch210-cxx11-cu126-aarch64-linux/{_paged_attention_cuda_0278991.abi3.so → _paged_attention_cuda_04f8f6e.abi3.so}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 140013424
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9cff0e750a7ca10b79b6cde332e141d9071dd1a284f07c4ca970939415840d32
|
| 3 |
size 140013424
|
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 _paged_attention_cuda_04f8f6e
|
| 3 |
+
ops = torch.ops._paged_attention_cuda_04f8f6e
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
+
return f"_paged_attention_cuda_04f8f6e::{op_name}"
|
build/torch210-cxx11-cu128-aarch64-linux/{_paged_attention_cuda_0278991.abi3.so → _paged_attention_cuda_04f8f6e.abi3.so}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 167603936
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a8096fb7f93cb71471d6e3929956f856ad06aa530d439bebc8fa567f4450bcbd
|
| 3 |
size 167603936
|
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 _paged_attention_cuda_04f8f6e
|
| 3 |
+
ops = torch.ops._paged_attention_cuda_04f8f6e
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
+
return f"_paged_attention_cuda_04f8f6e::{op_name}"
|
build/torch210-cxx11-cu130-aarch64-linux/{_paged_attention_cuda_0278991.abi3.so → _paged_attention_cuda_04f8f6e.abi3.so}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 86069336
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:aba6af81e8938a05d1cdf70082d64f2aa4d4dcad93ab6d8e3161ac29e969c7ee
|
| 3 |
size 86069336
|
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 _paged_attention_cuda_04f8f6e
|
| 3 |
+
ops = torch.ops._paged_attention_cuda_04f8f6e
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
+
return f"_paged_attention_cuda_04f8f6e::{op_name}"
|
build/torch211-cxx11-cu126-aarch64-linux/{_paged_attention_cuda_0278991.abi3.so → _paged_attention_cuda_04f8f6e.abi3.so}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 140009304
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9d8d5d8bb2ab1608ead0f8f6a818783e7907f6b54a4d52df1b3f34ba827178d7
|
| 3 |
size 140009304
|
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 _paged_attention_cuda_04f8f6e
|
| 3 |
+
ops = torch.ops._paged_attention_cuda_04f8f6e
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
+
return f"_paged_attention_cuda_04f8f6e::{op_name}"
|
build/torch211-cxx11-cu128-aarch64-linux/{_paged_attention_cuda_0278991.abi3.so → _paged_attention_cuda_04f8f6e.abi3.so}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 167599784
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:68dbfaf653351a1efca0566d4556247438461f460249ae291307d3106e99d018
|
| 3 |
size 167599784
|
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 _paged_attention_cuda_04f8f6e
|
| 3 |
+
ops = torch.ops._paged_attention_cuda_04f8f6e
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
+
return f"_paged_attention_cuda_04f8f6e::{op_name}"
|
build/torch211-cxx11-cu130-aarch64-linux/{_paged_attention_cuda_0278991.abi3.so → _paged_attention_cuda_04f8f6e.abi3.so}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 86065184
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8563b5429b1188e108d8aa82c9eadebfd7f71ee86161860563a720f6dad7d786
|
| 3 |
size 86065184
|
build/torch29-cxx11-cu129-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 _paged_attention_cuda_04f8f6e
|
| 3 |
+
ops = torch.ops._paged_attention_cuda_04f8f6e
|
| 4 |
|
| 5 |
def add_op_namespace_prefix(op_name: str):
|
| 6 |
"""
|
| 7 |
Prefix op by namespace.
|
| 8 |
"""
|
| 9 |
+
return f"_paged_attention_cuda_04f8f6e::{op_name}"
|
build/torch29-cxx11-cu129-aarch64-linux/{_paged_attention_cuda_0278991.abi3.so → _paged_attention_cuda_04f8f6e.abi3.so}
RENAMED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 181953688
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a2281e8180dc6de2ef41a1ece7940d0b5b5af29b6a947777c43180c1fda1d99f
|
| 3 |
size 181953688
|