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

add python interface.

parent d8f5cf8e
......@@ -3,6 +3,7 @@
#include <pybind11/stl.h>
#include <migraphx/program.hpp>
#include <migraphx/quantization.hpp>
#include <migraphx/quantize_ins.hpp>
#include <migraphx/generate.hpp>
#include <migraphx/cpu/target.hpp>
#include <migraphx/stringutils.hpp>
......@@ -183,6 +184,7 @@ PYBIND11_MODULE(migraphx, m)
m.def("generate_argument", &migraphx::generate_argument, py::arg("s"), py::arg("seed") = 0);
m.def("quantize", &migraphx::quantize);
m.def("quantize_ins", &migraphx::quantize_ins);
#ifdef HAVE_GPU
m.def("allocate_gpu", &migraphx::gpu::allocate_gpu, py::arg("s"), py::arg("host") = false);
......
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