add an overload for function capture_arguments

src/include/migraphx/quantization.hpp

@@ -21,6 +21,7 @@ void capture_arguments(program& prog,
                        const std::vector<std::string>& ins_names,
                        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);
 
 void quantize_int8(program& prog);
 void quantize_int8(program& prog, const std::vector<std::string>& ins_names);

src/py/migraphx_py.cpp

@@ -202,6 +202,10 @@ PYBIND11_MODULE(migraphx, m)
         migraphx::capture_arguments(p, ins_names);
     });
 
+    m.def("capture_arguments", [](migraphx::program& p) {
+        migraphx::capture_arguments(p);
+    });
+
 #ifdef HAVE_GPU
     m.def("allocate_gpu", &migraphx::gpu::allocate_gpu, py::arg("s"), py::arg("host") = false);
     m.def("to_gpu", &migraphx::gpu::to_gpu, py::arg("arg"), py::arg("host") = false);

src/quantization.cpp

@@ -407,5 +407,11 @@ void capture_arguments(program& prog, const std::vector<std::string>& ins_names)
     capture_arguments(prog, ins_names, calc_quant_params);
 }
 
+void capture_arguments(program& prog)
+{
+    std::vector<std::string> ins_names = {"dot", "convolution"};
+    capture_arguments(prog, ins_names);
+}
+
 } // namespace MIGRAPHX_INLINE_NS
 } // namespace migraphx