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

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

@@ -0,0 +1,13 @@
+from torch import Tensor
+from ._ops import ops
+
+def wvSpltK(in_a: Tensor, in_b: Tensor, out_c: Tensor, N_in: int, CuCount: int):
+    return ops.wvSpltK(in_a, in_b, out_c, N_in, CuCount)
+
+def LLMM1(in_a: Tensor, in_b: Tensor, out_c: Tensor, rows_per_block: int):
+    return ops.LLMM1(in_a, in_b, out_c, rows_per_block)
+
+__all__ = [
+    "wvSpltK",
+    "LLMM1",
+]

build/torch27-cxx11-rocm63-x86_64-linux/aiter_custom/_aiter_custom_a120ba4f5566b.so

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

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

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