Initial changes to API to get onnx names out

Author: TedThemistokleous

src/api/api.cpp

@@ -289,6 +289,11 @@ static void quantize_fp8_wrap(program& prog, const target& t, quantize_fp8_optio
     migraphx::quantize_fp8(prog, t, options.calibration);
 }
 
+static std::vector<std::string> get_supported_onnx_operators()
+{
+    return migraphx::get_onnx_operators();
+}
+
 #ifdef __clang__
 #pragma clang diagnostic push
 #pragma clang diagnostic ignored "-Wformat-nonliteral"
@@ -733,6 +738,17 @@ struct migraphx_quantize_fp8_options
     migraphx::quantize_fp8_options object;
 };
 
+extern "C" struct migraphx_get_supported_onnx_operators;
+struct migraphx_get_supported_onnx_operators
+{
+    template <class... Ts>
+    migraphx_get_supported_onnx_operators(Ts&&... xs)
+        : object(std::forward<Ts>(xs)...) // NOLINT(readability-redundant-member-init)
+    {
+    }
+    migraphx::get_supported_onnx_operators object;
+};
+
 extern "C" struct migraphx_context;
 struct migraphx_context
 {
@@ -2418,6 +2434,32 @@ extern "C" migraphx_status migraphx_quantize_fp8(migraphx_program_t prog,
     return api_error_result;
 }
 
+extern "C" migraphx_status migraphx_get_supported_onnx_operators_destroy(
+    migraphx_get_supported_onnx_operators_t get_supported_onnx_operators)
+{
+    auto api_error_result = migraphx::try_([&] { destroy((get_supported_onnx_operators)); });
+    return api_error_result;
+}
+
+extern "C" migraphx_status
+migraphx_get_supported_onnx_operators_assign_to(migraphx_get_supported_onnx_operators_t output,
+                                                const_migraphx_get_supported_onnx_operators_t input)
+{
+    auto api_error_result = migraphx::try_([&] { *output = *input; });
+    return api_error_result;
+}
+
+extern "C" migraphx_status migraphx_get_supported_onnx_operators_create(
+    migraphx_get_supported_onnx_operators_t* get_supported_onnx_operators)
+{
+    auto api_error_result = migraphx::try_([&] {
+        *get_supported_onnx_operators = object_cast<migraphx_get_supported_onnx_operators_t>(
+            allocate<migraphx::get_supported_onnx_operators>(
+                migraphx::get_supported_onnx_operators()));
+    });
+    return api_error_result;
+}
+
 extern "C" migraphx_status migraphx_context_finish(const_migraphx_context_t context)
 {
     auto api_error_result = migraphx::try_([&] {

src/api/include/migraphx/migraphx.h

@@ -145,6 +145,10 @@ typedef const struct migraphx_quantize_int8_options* const_migraphx_quantize_int
 typedef struct migraphx_quantize_fp8_options* migraphx_quantize_fp8_options_t;
 typedef const struct migraphx_quantize_fp8_options* const_migraphx_quantize_fp8_options_t;
 
+typedef struct migraphx_get_supported_onnx_operators* migraphx_get_supported_onnx_operators_t;
+typedef const struct migraphx_get_supported_onnx_operators*
+    const_migraphx_get_supported_onnx_operators_t;
+
 typedef struct migraphx_context* migraphx_context_t;
 typedef const struct migraphx_context* const_migraphx_context_t;
 
@@ -659,6 +663,16 @@ MIGRAPHX_C_EXPORT migraphx_status migraphx_quantize_fp8(migraphx_program_t prog,
                                                         migraphx_target_t target,
                                                         migraphx_quantize_fp8_options_t options);
 
+MIGRAPHX_C_EXPORT migraphx_status migraphx_get_supported_onnx_operators_destroy(
+    migraphx_get_supported_onnx_operators_t get_supported_onnx_operators);
+
+MIGRAPHX_C_EXPORT migraphx_status migraphx_get_supported_onnx_operators_assign_to(
+    migraphx_get_supported_onnx_operators_t output,
+    const_migraphx_get_supported_onnx_operators_t input);
+
+MIGRAPHX_C_EXPORT migraphx_status migraphx_get_supported_onnx_operators_create(
+    migraphx_get_supported_onnx_operators_t* get_supported_onnx_operators);
+
 MIGRAPHX_C_EXPORT migraphx_status migraphx_context_finish(const_migraphx_context_t context);
 
 MIGRAPHX_C_EXPORT migraphx_status migraphx_context_get_queue(void** out,

src/api/include/migraphx/migraphx.hpp

@@ -1603,6 +1603,19 @@ quantize_fp8(const program& prog, const target& ptarget, const quantize_fp8_opti
          options.get_handle_ptr());
 }
 
+struct supported_onnx_ops_options : MIGRPAHX_HANDLE_BASE(supported_onnx_ops_options)
+{
+    supported_onnx_ops-options() {this->make_handle(&migraphx_supported_onnx_ops_options_create); }
+
+    MIGRAPHX_HANDLE_CONSTRUCTOR(supported_onnx_op_options)
+};
+
+inline void get_supported_onnx_operators(supported_onnx_ops_options& options)
+{
+    call(&migraphx_get_onnx_operators,
+         options.get_handle_ptr());
+}
+
 struct experimental_custom_op_base
 {
     experimental_custom_op_base()                                              = default;
@@ -1618,6 +1631,7 @@ struct experimental_custom_op_base
     virtual bool runs_on_offload_target() const = 0;
 };
 
+
 struct experimental_custom_op : interface_base<MIGRAPHX_HANDLE_BASE(experimental_custom_op)>
 {
     template <class T>

src/api/migraphx.py

@@ -510,6 +510,11 @@ def quantize_fp8_options(h):
                             options='migraphx::quantize_fp8_options'),
                  fname='migraphx::quantize_fp8_wrap')
 
+@auto_handle()
+def get_supported_onnx_operators(h):
+    h.constructor('create',
+                  fname='migraphx::get_supported_onnx_operators')
+
 
 @auto_handle(ref=True)
 def context(h):

src/py/migraphx_py.cpp

@@ -621,6 +621,7 @@ MIGRAPHX_PYBIND11_MODULE(migraphx, m)
         py::arg("map_input_dims") = std::unordered_map<std::string, std::vector<std::size_t>>(),
         py::arg("output_names")   = std::vector<std::string>());
 
+    m.def("get_supported_onnx_operators", []{return migraphx::get_onnx_operators();});
     m.def(
         "parse_onnx",
         [](const std::string& filename,

tools/api/api.cpp

@@ -289,6 +289,11 @@ static void quantize_fp8_wrap(program& prog, const target& t, quantize_fp8_optio
     migraphx::quantize_fp8(prog, t, options.calibration);
 }
 
+static std::vector<std::string> get_supported_onnx_operators()
+{
+    return migraphx::get_onnx_operators();
+}
+
 #ifdef __clang__
 #pragma clang diagnostic push
 #pragma clang diagnostic ignored "-Wformat-nonliteral"