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_aiter_operator/_aiter_aiter_operator_a120ba4f5566b.so filter=lfs diff=lfs merge=lfs -text

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

@@ -0,0 +1,35 @@
+# This is a Python package for the kernel.
+import torch
+from ._ops import ops
+
+def add(input: torch.Tensor, other: torch.Tensor) -> torch.Tensor:
+    return ops.add(input, other)
+
+
+def sub(input: torch.Tensor, other: torch.Tensor) -> torch.Tensor:
+    return ops.sub(input, other)
+
+
+def mul(input: torch.Tensor, other: torch.Tensor) -> torch.Tensor:
+    return ops.mul(input, other)
+
+
+def div(input: torch.Tensor, other: torch.Tensor) -> torch.Tensor:
+    return ops.div(input, other)
+
+
+def add_(input: torch.Tensor, other: torch.Tensor) -> torch.Tensor:
+    return ops.add_(input, other)
+
+
+def sub_(input: torch.Tensor, other: torch.Tensor) -> torch.Tensor:
+    return ops.sub_(input, other)
+
+__all__ = [
+    "add",
+    "sub",
+    "mul",
+    "div",
+    "add_",
+    "sub_",
+]

build/torch27-cxx11-rocm63-x86_64-linux/aiter_aiter_operator/_aiter_aiter_operator_a120ba4f5566b.so

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

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

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