add a python interface for int8 quantization.

09740556 · Shucai Xiao · 4cf9bd01 · 09740556
Commit 09740556 authored Jun 06, 2019 by Shucai Xiao
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

src/py/migraphx_py.cpp src/py/migraphx_py.cpp +3 -0

No files found.
--- a/src/py/migraphx_py.cpp
+++ b/src/py/migraphx_py.cpp
@@ -187,6 +187,9 @@ PYBIND11_MODULE(migraphx, m)
        migraphx::quantize(p, ins_names);
    });
    m.def("quantize", [](migraphx::program& p) { migraphx::quantize(p, {"all"}); });
+    m.def("quantize_int8", [](migraphx::program& p, std::vector<std::string>& ins_names) {
+        migraphx::quantize_int8(p, ins_names);
+    });
    m.def("quantize_int8",
          [](migraphx::program& p,
             std::vector<std::string>& ins_names,