diff --git a/mlir/lib/Bindings/Python/DialectQuant.cpp b/mlir/lib/Bindings/Python/DialectQuant.cpp index af9cdc7bdd2d8..9a871f2c122d1 100644 --- a/mlir/lib/Bindings/Python/DialectQuant.cpp +++ b/mlir/lib/Bindings/Python/DialectQuant.cpp @@ -250,6 +250,7 @@ static void populateDialectQuantSubmodule(const py::module &m) { double scale = mlirUniformQuantizedPerAxisTypeGetScale(type, i); scales.push_back(scale); } + return scales; }, "The scales designate the difference between the real values " "corresponding to consecutive quantized values differing by 1. The ith " @@ -265,6 +266,7 @@ static void populateDialectQuantSubmodule(const py::module &m) { mlirUniformQuantizedPerAxisTypeGetZeroPoint(type, i); zeroPoints.push_back(zeroPoint); } + return zeroPoints; }, "the storage values corresponding to the real value 0 in the affine " "equation. The ith zero point corresponds to the ith slice in the " diff --git a/mlir/python/mlir/_mlir_libs/_mlir/dialects/quant.pyi b/mlir/python/mlir/_mlir_libs/_mlir/dialects/quant.pyi index a10bc693ba600..47168d49c5568 100644 --- a/mlir/python/mlir/_mlir_libs/_mlir/dialects/quant.pyi +++ b/mlir/python/mlir/_mlir_libs/_mlir/dialects/quant.pyi @@ -101,7 +101,7 @@ class UniformQuantizedPerAxisType(QuantizedType): def scales(self) -> list[float]: ... @property - def zero_points(self) -> list[float]: ... + def zero_points(self) -> list[int]: ... @property def quantized_dimension(self) -> int: ... diff --git a/mlir/test/python/dialects/quant.py b/mlir/test/python/dialects/quant.py index 0ee3327dec152..b1d6e85f519b5 100644 --- a/mlir/test/python/dialects/quant.py +++ b/mlir/test/python/dialects/quant.py @@ -108,9 +108,9 @@ def test_uniform_per_axis_type(): ), ) - # CHECK: scales: None + # CHECK: scales: [200.0, 0.99872] print(f"scales: {per_axis.scales}") - # CHECK: zero_points: None + # CHECK: zero_points: [0, 120] print(f"zero_points: {per_axis.zero_points}") # CHECK: quantized dim: 1 print(f"quantized dim: {per_axis.quantized_dimension}")