Build uploaded using `kernels`.

build/torch27-cxx11-cu118-x86_64-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_92f3139
+ops = torch.ops._quantization_eetq_92f3139
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_quantization_eetq_6337ee0::{op_name}"
+    return f"_quantization_eetq_92f3139::{op_name}"

build/torch27-cxx11-cu118-x86_64-linux/quantization_eetq/{_quantization_eetq_6337ee0.abi3.so → _quantization_eetq_92f3139.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:f5d18745cdd32c963b92bd1fd88a10a0ad815eaa8aa393668c26515ce82e5b64
-size 31355280
+oid sha256:a87d3bf237f1e897b2b58d95a686ee4b5584329b6d04243eaf970681be00bb67
+size 31977872

build/torch27-cxx11-cu126-x86_64-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_92f3139
+ops = torch.ops._quantization_eetq_92f3139
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_quantization_eetq_6337ee0::{op_name}"
+    return f"_quantization_eetq_92f3139::{op_name}"

build/torch27-cxx11-cu126-x86_64-linux/quantization_eetq/{_quantization_eetq_6337ee0.abi3.so → _quantization_eetq_92f3139.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:40e09f6ca563b4eefabed226279812577c5303f265c5e51717d3993b48d6805d
-size 31029608
+oid sha256:3a0d40ed3d499de55afde610e6bdb0c08e3ed40128c6f82984a1fdfbd2a5baae
+size 31639912

build/torch27-cxx11-cu128-x86_64-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_92f3139
+ops = torch.ops._quantization_eetq_92f3139
 
 def add_op_namespace_prefix(op_name: str):
     """
     Prefix op by namespace.
     """
-    return f"_quantization_eetq_6337ee0::{op_name}"
+    return f"_quantization_eetq_92f3139::{op_name}"

build/torch27-cxx11-cu128-x86_64-linux/quantization_eetq/{_quantization_eetq_6337ee0.abi3.so → _quantization_eetq_92f3139.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:78e412e8c6fa92e7ab034f902c6369eb907ca3b861e04c5e8224922d8edf9a30
-size 37624552
+oid sha256:eb04c8f5d3cb36f8566d702e184f66277061bbbc2a72693bdfdb1182e2516125
+size 38054616

build/torch28-cxx11-cu126-x86_64-linux/quantization_eetq/_ops.py

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

build/torch28-cxx11-cu126-x86_64-linux/quantization_eetq/{_quantization_eetq_0c60fb4.abi3.so → _quantization_eetq_92f3139.abi3.so}

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:854740b7499076f0f7c54be036dc9a66bdf4aa0bab322f8f2ef7e92723f76f93
-size 30881144
+oid sha256:12d4712f6937fc8abaf1c30461318ac52913e3b836430b28e60b8ee07bd5490b
+size 31593848

build/torch28-cxx11-cu128-x86_64-linux/quantization_eetq/_ops.py

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

build/torch28-cxx11-cu128-x86_64-linux/quantization_eetq/_quantization_eetq_0c60fb4.abi3.so

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

build/torch28-cxx11-cu128-x86_64-linux/quantization_eetq/_quantization_eetq_92f3139.abi3.so

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

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

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

build/torch28-cxx11-cu129-x86_64-linux/quantization_eetq/_quantization_eetq_0c60fb4.abi3.so

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

build/torch28-cxx11-cu129-x86_64-linux/quantization_eetq/_quantization_eetq_92f3139.abi3.so

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

build/torch29-cxx11-cu126-x86_64-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-x86_64-linux/quantization_eetq/_ops.py

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

build/torch29-cxx11-cu126-x86_64-linux/quantization_eetq/_quantization_eetq_92f3139.abi3.so

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

build/torch29-cxx11-cu126-x86_64-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-x86_64-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-x86_64-linux/quantization_eetq/_ops.py

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

build/torch29-cxx11-cu128-x86_64-linux/quantization_eetq/_quantization_eetq_92f3139.abi3.so

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

build/torch29-cxx11-cu128-x86_64-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
+    )