Upload 3 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+build/torch27-cxx11-rocm63-x86_64-linux/aiter_pa/_aiter_pa_a120ba4f5566b.so filter=lfs diff=lfs merge=lfs -text

build/torch27-cxx11-rocm63-x86_64-linux/aiter_pa/__init__.py

@@ -0,0 +1,52 @@
+# This is a Python package for the kernel.
+from typing import Optional
+import torch
+
+from ._ops import ops
+
+def paged_attention_rocm(
+    out: torch.Tensor,
+    exp_sums: torch.Tensor,
+    max_logits: torch.Tensor,
+    tmp_out: torch.Tensor,
+    query: torch.Tensor,
+    key_cache: torch.Tensor,
+    value_cache: torch.Tensor,
+    num_kv_heads: int,
+    scale: float,
+    block_tables: torch.Tensor,
+    context_lens: torch.Tensor,
+    block_size: int,
+    max_context_len: int,
+    alibi_slopes: Optional[torch.Tensor],
+    kv_cache_dtype: str,
+    k_scale: float,
+    v_scale: float,
+    fp8_out_scale: Optional[torch.Tensor],
+    partition_size: int,
+):
+    return ops.paged_attention_rocm(
+        out,
+        exp_sums,
+        max_logits,
+        tmp_out,
+        query,
+        key_cache,
+        value_cache,
+        num_kv_heads,
+        scale,
+        block_tables,
+        context_lens,
+        block_size,
+        max_context_len,
+        alibi_slopes,
+        kv_cache_dtype,
+        k_scale,
+        v_scale,
+        fp8_out_scale,
+        partition_size,
+    )
+
+__all__ = [
+    "paged_attention_rocm",
+]

build/torch27-cxx11-rocm63-x86_64-linux/aiter_pa/_aiter_pa_a120ba4f5566b.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cbe37179bb4344cdd65280558bd853fd6f489de7725e2df72347a620e70fee8e
+size 2919776

build/torch27-cxx11-rocm63-x86_64-linux/aiter_pa/_ops.py

@@ -0,0 +1,9 @@
+import torch
+from . import _aiter_pa_a120ba4f5566b
+ops = torch.ops._aiter_pa_a120ba4f5566b
+
+def add_op_namespace_prefix(op_name: str):
+    """
+    Prefix op by namespace.
+    """
+    return f"_aiter_pa_a120ba4f5566b::{op_name}"