Handle output

Author: TedThemistokleous

src/api/api.cpp

@@ -291,9 +291,9 @@ static void quantize_fp8_wrap(program& prog, const target& t, quantize_fp8_optio
 
 static size_t get_onnx_operators_size() { return migraphx::get_onnx_operators().size(); }
 
-static const char* get_onnx_operator_name_at_index(std::size_t index)
+static char* get_onnx_operator_name_at_index(std::size_t index)
 {
-    return get_onnx_operators().at(index).c_str();
+    return const_cast<char*>(get_onnx_operators().at(index).c_str());
 }
 
 #ifdef __clang__
@@ -2423,14 +2423,14 @@ extern "C" migraphx_status migraphx_quantize_fp8(migraphx_program_t prog,
     return api_error_result;
 }
 
-extern "C" migraphx_status migraphx_get_onnx_operator_name_at_index(const char** out, size_t index)
+extern "C" migraphx_status migraphx_get_onnx_operator_name_at_index(char** out, size_t index)
 {
     auto api_error_result =
         migraphx::try_([&] { *out = migraphx::get_onnx_operator_name_at_index((index)); });
     return api_error_result;
 }
 
-extern "C" migraphx_status migraphx_get_onnx_operators_size(std::size_t* out)
+extern "C" migraphx_status migraphx_get_onnx_operators_size(size_t* out)
 {
     auto api_error_result = migraphx::try_([&] { *out = migraphx::get_onnx_operators_size(); });
     return api_error_result;

src/api/include/migraphx/migraphx.h

@@ -659,10 +659,10 @@ 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_onnx_operator_name_at_index(const char** out,
+MIGRAPHX_C_EXPORT migraphx_status migraphx_get_onnx_operator_name_at_index(char** out,
                                                                            size_t index);
 
-MIGRAPHX_C_EXPORT migraphx_status migraphx_get_onnx_operators_size(std::size_t* out);
+MIGRAPHX_C_EXPORT migraphx_status migraphx_get_onnx_operators_size(size_t* out);
 
 MIGRAPHX_C_EXPORT migraphx_status migraphx_context_finish(const_migraphx_context_t context);
 

src/api/include/migraphx/migraphx.hpp

@@ -1643,15 +1643,18 @@ void register_experimental_custom_op(T& obj)
     op.register_op();
 }
 
-std::vector<const char *> get_onnx_operators()
+inline std::vector<std::string> get_onnx_operators()
 {
-    auto size = get_onnx_operators_size();
-    std::vector<const char *> result(size, "");
+    size_t size = 0;
+    migraphx_get_onnx_operators_size(&size);
+    std::vector<std::string> result(size, "");
 
     size_t index = 0;
     for(auto &name: result)
     {
-        name= get_onnx_operator_name_at_index(index);
+        char * name_op;
+        migraphx_get_onnx_operator_name_at_index(&name_op, index);
+        name = *name_op;
         index++;
     }
     return result;

src/api/migraphx.py

@@ -513,11 +513,11 @@ def quantize_fp8_options(h):
 api.add_function('migraphx_get_onnx_operator_name_at_index',
                  api.params(index='size_t'),
                  fname='migraphx::get_onnx_operator_name_at_index',
-                 returns='const char *')
+                 returns='char *')
 
 api.add_function('migraphx_get_onnx_operators_size',
                  fname='migraphx::get_onnx_operators_size',
-                 returns='std::size_t')
+                 returns='size_t')
 
 
 

tools/api/api.cpp

@@ -294,9 +294,9 @@ static size_t get_onnx_operators_size()
     return migraphx::get_onnx_operators().size();
 }
 
-static const char * get_onnx_operator_name_at_index(std::size_t index)
+static char * get_onnx_operator_name_at_index(std::size_t index)
 {
-    return get_onnx_operators().at(index).c_str();   
+    return const_cast<char*>(get_onnx_operators().at(index).c_str());   
 }
 
 #ifdef __clang__