mindspore 2.0.0rc1__cp38-cp38-manylinux1_x86_64.whl → 2.2.0__cp38-cp38-manylinux1_x86_64.whl

mindspore/include/c_api/ms/context.h

@@ -0,0 +1,102 @@
+/**
+ * Copyright 2022 Huawei Technologies Co., Ltd
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#ifndef MINDSPORE_CCSRC_C_API_INCLUDE_CONTEXT_H_
+#define MINDSPORE_CCSRC_C_API_INCLUDE_CONTEXT_H_
+
+#include <stdint.h>
+#include <stdlib.h>
+#include "include/c_api/ms/base/macros.h"
+#include "include/c_api/ms/base/status.h"
+#include "include/c_api/ms/base/handle_types.h"
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+/// \brief Create a resource manager. Tracking allocated objects(i.e. FuncGraph, Nodes...).
+MIND_C_API ResMgrHandle MSResourceManagerCreate();
+
+/// \brief Release a resource manager and the enclosed objects.
+///
+/// \param[in] res_mgr Manager Handle that manages the resource of the funcGraph.
+MIND_C_API void MSResourceManagerDestroy(ResMgrHandle res_mgr);
+
+/// \brief Set context to Graph or pynative mood.
+///
+/// \param[in] eager_mode True: Pynative, False: Graph.
+MIND_C_API void MSSetEagerMode(bool eager_mode);
+
+/// \brief Select MindRT, VM or GE as Backend Policy.
+///
+/// \param[in] policy select one from {"ge", "vm", "ms", "ge_only", "vm_prior"}.
+///
+/// \return Error code indicates whether the function executed successfully.
+MIND_C_API STATUS MSSetBackendPolicy(const char *policy);
+
+/// \brief Get backend policy context.
+///
+/// \param[in] str_buf The char array to contain the backend policy string.
+/// \param[in] str_len The size of the char array.
+///
+/// \return name of the device under current context
+MIND_C_API STATUS MSGetBackendPolicy(char str_buf[], size_t str_len);
+
+/// \brief set device type context.
+///
+/// \param[in] device type name of the device i.e. CPU, GPU, Ascend.
+MIND_C_API void MSSetDeviceTarget(const char *device);
+
+/// \brief Get device type context.
+///
+/// \param[in] str_buf The char array to contain the device target string.
+/// \param[in] str_len The size of the char array.
+///
+/// \return name of the device under current context
+MIND_C_API STATUS MSGetDeviceTarget(char str_buf[], size_t str_len);
+
+/// \brief Set Device ID context.
+///
+/// \param[in] deviceId Device ID.
+MIND_C_API void MSSetDeviceId(uint32_t deviceId);
+
+/// \brief Set flag for saving the graph.
+///
+/// \param[in] save_mode The flag to control the amount of saved graph files. There are 3 levels:
+/// 1: Basic level. Only save frontend IR graph and some common backend IR graphs.
+/// 2: Advanced level. Save all graphs except for verbose IR graphs and dot files.
+/// 3: Full level. Save all IR graphs.
+MIND_C_API void MSSetIRGraphsSaveMode(int save_mode);
+
+/// \brief Set save path of IRGraphs.
+///
+/// \param[in] save_path The save path.
+MIND_C_API void MSSetIRGraphsSavePath(const char *save_path);
+
+/// \brief Set flag for auto shape and type infer.
+///
+/// \param[in] res_mgr Manager Handle that manages the resource of the funcGraph.
+/// \param[in] infer True: Use inner auto infer, False: Not use auto infer.
+MIND_C_API void MSSetInfer(ResMgrHandle res_mgr, bool infer);
+
+/// \brief Get flag for auto shape and type infer.
+///
+/// \param[in] res_mgr Manager Handle that manages the resource of the funcGraph.
+MIND_C_API bool MSGetInfer(ResMgrHandle res_mgr);
+
+#ifdef __cplusplus
+}
+#endif
+#endif  // MINDSPORE_CCSRC_C_API_INCLUDE_CONTEXT_H_

mindspore/include/c_api/ms/graph.h

@@ -0,0 +1,160 @@
+/**
+ * Copyright 2022 Huawei Technologies Co., Ltd
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#ifndef MINDSPORE_CCSRC_C_API_INCLUDE_GRAPH_H_
+#define MINDSPORE_CCSRC_C_API_INCLUDE_GRAPH_H_
+
+#include <stdbool.h>
+#include <stdlib.h>
+#include "include/c_api/ms/node.h"
+#include "include/c_api/ms/base/macros.h"
+#include "include/c_api/ms/base/status.h"
+#include "include/c_api/ms/context.h"
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+/// \brief Creates an empty function graph.
+///
+/// \param[in] res_mgr Resource manager that saves allocated instance resources.
+///
+/// \return The created function graph.
+MIND_C_API GraphHandle MSFuncGraphCreate(ResMgrHandle res_mgr);
+
+/// \brief Load function graph from MINDIR.
+///
+/// \param[in] res_mgr Resource manager that saves allocated instance resources.
+/// \param[in] file_path The path of MINDIR.
+///
+/// \return The loaded function graph.
+MIND_C_API GraphHandle MSFuncGraphLoad(ResMgrHandle res_mgr, const char *file_path);
+
+/// \brief Get the input node of the function graph.
+///
+/// \param[in] res_mgr Resource manager that saves allocated instance resources.
+/// \param[in] graph The given function graph pointer handle.
+/// \param[in] i Index of the input node.
+///
+/// \return The created function graph.
+MIND_C_API NodeHandle MSFuncGraphGetInput(ResMgrHandle res_mgr, ConstGraphHandle graph, size_t i);
+
+/// \brief Get the inputs number of the function graph.
+///
+/// \param[in] res_mgr Resource manager that saves allocated instance resources.
+/// \param[in] graph The given function graph pointer handle.
+/// \param[in] error Records error code that indicate whether the functions executed successfully.
+///
+/// \return the inputs number of the function graph.
+MIND_C_API size_t MSFuncGraphGetInputNum(ResMgrHandle res_mgr, ConstGraphHandle graph, STATUS *error);
+
+/// \brief Get all inputs of the function graph.
+///
+/// \param[in] res_mgr Resource manager that saves allocated instance resources.
+/// \param[in] graph The given function graph pointer handle.
+/// \param[in] inputs The array to contain input nodes.
+/// \param[in] input_num The length of the array.
+///
+/// \return Error code that indicate whether the functions executed successfully.
+MIND_C_API STATUS MSFuncGraphGetInputs(ResMgrHandle res_mgr, ConstGraphHandle graph, NodeHandle inputs[],
+                                       size_t input_num);
+
+/// \brief Set the output node.
+///
+/// \param[in] res_mgr Resource manager that saves allocated instance resources.
+/// \param[in] graph The given function graph pointer handle.
+/// \param[in] op_node The output operator node to be set.
+/// \param[in] force_new_ret If true, a new return node is always created.
+///
+/// \return Error code that indicate whether the functions executed successfully.
+MIND_C_API STATUS MSFuncGraphSetOutput(ResMgrHandle res_mgr, GraphHandle graph, ConstNodeHandle op_node,
+                                       bool force_new_ret);
+
+/// \brief Set the output node.
+///
+/// \param[in] res_mgr Resource manager that saves allocated instance resources.
+/// \param[in] graph The given function graph pointer handle.
+/// \param[in] outputs The array of output operator nodes.
+/// \param[in] force_new_ret If true, a new return node is always created.
+///
+/// \return Error code that indicate whether the functions executed successfully.
+MIND_C_API STATUS MSFuncGraphSetOutputs(ResMgrHandle res_mgr, GraphHandle graph, Handle const outputs[],
+                                        size_t output_num, bool force_new_ret);
+
+/// \brief Get the output node according to the index.
+///
+/// \param[in] res_mgr Resource manager that saves allocated instance resources.
+/// \param[in] graph The given function graph pointer handle.
+/// \param[in] i The index to get the output. If there is only one output for graph, the i should be 0;
+///
+/// \return The output node, nullptr if output not set.
+MIND_C_API NodeHandle MSFuncGraphGetOutput(ResMgrHandle res_mgr, ConstGraphHandle graph, size_t i);
+
+/// \brief Get the outputs number of the function graph.
+///
+/// \param[in] res_mgr Resource manager that saves allocated instance resources.
+/// \param[in] graph The given function graph pointer handle.
+/// \param[in] error Records error code that indicate whether the functions executed successfully.
+///
+/// \return the outputs number of the function graph.
+MIND_C_API size_t MSFuncGraphGetOutputNum(ResMgrHandle res_mgr, ConstGraphHandle graph, STATUS *error);
+
+/// \brief Get all outputs of the function graph.
+///
+/// \param[in] res_mgr Resource manager that saves allocated instance resources.
+/// \param[in] graph The given function graph pointer handle.
+/// \param[in] outputs The array to contain input nodes.
+/// \param[in] output_num The length of the array.
+///
+/// \return Error code that indicate whether the functions executed successfully.
+MIND_C_API STATUS MSFuncGraphGetOutputs(ResMgrHandle res_mgr, ConstGraphHandle graph, NodeHandle outputs[],
+                                        size_t output_num);
+
+/// \brief Replace a node in a function graph.
+///
+/// \param[in] res_mgr Resource manager that saves allocated instance resources.
+/// \param[in] graph The given function graph pointer handle.
+/// \param[in] new_node The node needs to be replaced.
+/// \param[in] new_node The replace node.
+///
+/// \return Error code that indicate whether the functions executed successfully.
+MIND_C_API STATUS MSFuncGraphReplace(ResMgrHandle res_mgr, GraphHandle graph, ConstNodeHandle old_node,
+                                     ConstNodeHandle new_node);
+
+/// \brief Compile the function graph.
+///
+/// \param[in] res_mgr Resource manager that saves allocated instance resources.
+/// \param[in] graph The given function graph pointer handle.
+///
+/// \return Error code that indicate whether the function graph compiled successfully.
+MIND_C_API STATUS MSFuncGraphCompile(ResMgrHandle res_mgr, GraphHandle graph, OptPassID *opt_pass, size_t pass_num);
+
+/// \brief Run the function graph.
+///
+/// \param[in] res_mgr Resource manager that saves allocated instance resources.
+/// \param[in] graph The given function graph pointer handle.
+/// \param[in] inputs The function graph (model) inputs which can be Values or Tensors.
+/// \param[in] input_num The input size.
+/// \param[in] outputs The function graph (model) outputs in Tensor form.
+/// \param[in] outputs_num The output size.
+///
+/// \return Error code that indicate whether the function graph executed successfully.
+MIND_C_API STATUS MSFuncGraphRun(ResMgrHandle res_mgr, GraphHandle graph, Handle const inputs[], size_t input_num,
+                                 TensorHandle outputs[], size_t outputs_num);
+#ifdef __cplusplus
+}
+#endif
+#endif  // MINDSPORE_CCSRC_C_API_INCLUDE_GRAPH_H_