Commit e6158f10 authored by Shucai Xiao's avatar Shucai Xiao
Browse files

add quantize_int8 python interace.

parents c63d2dd3 11a0f1a9
...@@ -22,6 +22,8 @@ void capture_arguments(program& prog, ...@@ -22,6 +22,8 @@ void capture_arguments(program& prog,
std::function<void(std::size_t, std::vector<argument>)> func); std::function<void(std::size_t, std::vector<argument>)> func);
void capture_arguments(program& prog, const std::vector<std::string>& ins_names); void capture_arguments(program& prog, const std::vector<std::string>& ins_names);
void quantize_int8(program& prog);
void quantize_int8(program& prog, const std::vector<std::string>& ins_names);
void quantize_int8(program& prog, void quantize_int8(program& prog,
const std::vector<std::string>& ins_names, const std::vector<std::string>& ins_names,
const std::vector<std::pair<float, float>>& quant_params); const std::vector<std::pair<float, float>>& quant_params);
......
...@@ -153,6 +153,7 @@ PYBIND11_MODULE(migraphx, m) ...@@ -153,6 +153,7 @@ PYBIND11_MODULE(migraphx, m)
py::class_<migraphx::target>(m, "target"); py::class_<migraphx::target>(m, "target");
py::class_<migraphx::program>(m, "program") py::class_<migraphx::program>(m, "program")
.def("clone", [](migraphx::program& p) { return *(new migraphx::program(p)); })
.def("get_parameter_shapes", &migraphx::program::get_parameter_shapes) .def("get_parameter_shapes", &migraphx::program::get_parameter_shapes)
.def("get_shape", &migraphx::program::get_shape) .def("get_shape", &migraphx::program::get_shape)
.def("compile", [](migraphx::program& p, const migraphx::target& t) { p.compile(t); }) .def("compile", [](migraphx::program& p, const migraphx::target& t) { p.compile(t); })
...@@ -192,6 +193,7 @@ PYBIND11_MODULE(migraphx, m) ...@@ -192,6 +193,7 @@ PYBIND11_MODULE(migraphx, m)
std::vector<std::pair<float, float>>& quant_params) { std::vector<std::pair<float, float>>& quant_params) {
migraphx::quantize_int8(p, ins_names, quant_params); migraphx::quantize_int8(p, ins_names, quant_params);
}); });
m.def("quantize_int8", [](migraphx::program& p) { migraphx::quantize_int8(p); });
m.def("capture_arguments", [](migraphx::program& p, const std::vector<std::string>& ins_names) { m.def("capture_arguments", [](migraphx::program& p, const std::vector<std::string>& ins_names) {
migraphx::capture_arguments(p, ins_names); migraphx::capture_arguments(p, ins_names);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment