Build uploaded using `kernels`.

build/torch27-cxx11-cu128-aarch64-linux/quantization_eetq/_ops.py

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

build/torch27-cxx11-cu128-aarch64-linux/quantization_eetq/{_quantization_eetq_6337ee0.abi3.so → _quantization_eetq_ee9ac42.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6022247112ebdaf553f6f92b459391e823fb10d28a224235e37be45dea3ed6a9
-size 37504664
+oid sha256:2c32dd6d144418fd7255d5555b634407890ad96b298c98a00bb8463d6f6dd9be
+size 38028936

build/torch28-cxx11-cu129-aarch64-linux/quantization_eetq/_ops.py

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

build/torch28-cxx11-cu129-aarch64-linux/quantization_eetq/{_quantization_eetq_1d4b892.abi3.so → _quantization_eetq_ee9ac42.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:6f9968edb45388c79bc28af40a3618d415c9149facdab15748d857f96b3be7ab
-size 38153728
+oid sha256:c3c53ac387f693a12937e3c6d8640832470cd801117269fcbc2f6a1a83382612
+size 38743552

build/torch29-cxx11-cu126-aarch64-linux/quantization_eetq/__init__.py

@@ -0,0 +1,3 @@
+from .custom_ops import w8_a16_gemm, w8_a16_gemm_, preprocess_weights, quant_weights
+
+__all__ = ["w8_a16_gemm", "w8_a16_gemm_", "preprocess_weights", "quant_weights"]

build/torch29-cxx11-cu126-aarch64-linux/quantization_eetq/_ops.py

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

build/torch29-cxx11-cu126-aarch64-linux/quantization_eetq/_quantization_eetq_ee9ac42.abi3.so

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

build/torch29-cxx11-cu126-aarch64-linux/quantization_eetq/custom_ops.py

@@ -0,0 +1,36 @@
+from typing import List
+import torch
+
+from ._ops import ops
+
+
+def w8_a16_gemm(
+    input: torch.Tensor, weight: torch.Tensor, scale: torch.Tensor
+) -> torch.Tensor:
+    return ops.w8_a16_gemm(input, weight, scale)
+
+
+def w8_a16_gemm_(
+    input: torch.Tensor,
+    weight: torch.Tensor,
+    scale: torch.Tensor,
+    output: torch.Tensor,
+    m: int,
+    n: int,
+    k: int,
+) -> torch.Tensor:
+    return ops.w8_a16_gemm_(input, weight, scale, output, m, n, k)
+
+
+def preprocess_weights(origin_weight: torch.Tensor, is_int4: bool) -> torch.Tensor:
+    return ops.preprocess_weights(origin_weight, is_int4)
+
+
+def quant_weights(
+    origin_weight: torch.Tensor,
+    quant_type: torch.dtype,
+    return_unprocessed_quantized_tensor: bool,
+) -> List[torch.Tensor]:
+    return ops.quant_weights(
+        origin_weight, quant_type, return_unprocessed_quantized_tensor
+    )

build/torch29-cxx11-cu128-aarch64-linux/quantization_eetq/__init__.py

@@ -0,0 +1,3 @@
+from .custom_ops import w8_a16_gemm, w8_a16_gemm_, preprocess_weights, quant_weights
+
+__all__ = ["w8_a16_gemm", "w8_a16_gemm_", "preprocess_weights", "quant_weights"]

build/torch29-cxx11-cu128-aarch64-linux/quantization_eetq/_ops.py

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

build/torch29-cxx11-cu128-aarch64-linux/quantization_eetq/_quantization_eetq_ee9ac42.abi3.so

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

build/torch29-cxx11-cu128-aarch64-linux/quantization_eetq/custom_ops.py

@@ -0,0 +1,36 @@
+from typing import List
+import torch
+
+from ._ops import ops
+
+
+def w8_a16_gemm(
+    input: torch.Tensor, weight: torch.Tensor, scale: torch.Tensor
+) -> torch.Tensor:
+    return ops.w8_a16_gemm(input, weight, scale)
+
+
+def w8_a16_gemm_(
+    input: torch.Tensor,
+    weight: torch.Tensor,
+    scale: torch.Tensor,
+    output: torch.Tensor,
+    m: int,
+    n: int,
+    k: int,
+) -> torch.Tensor:
+    return ops.w8_a16_gemm_(input, weight, scale, output, m, n, k)
+
+
+def preprocess_weights(origin_weight: torch.Tensor, is_int4: bool) -> torch.Tensor:
+    return ops.preprocess_weights(origin_weight, is_int4)
+
+
+def quant_weights(
+    origin_weight: torch.Tensor,
+    quant_type: torch.dtype,
+    return_unprocessed_quantized_tensor: bool,
+) -> List[torch.Tensor]:
+    return ops.quant_weights(
+        origin_weight, quant_type, return_unprocessed_quantized_tensor
+    )