mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!39327 [MS]: export pipeline interfaces when build 

Merge pull request !39327 from tinychong/export_pipeline_interfaces_for_master
This commit is contained in:
i-robot 2022-08-10 18:29:53 +00:00 committed by Gitee
parent 423791f298 2ea4ddb5c4
commit d06c50ad4b
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
4 changed files with 11 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/pipeline/jit/pass.h
View File

@ -22,6 +22,7 @@
#include <utility>
#include <string>
#include "pipeline/jit/resource.h"
#include "include/api/visible.h"
namespace mindspore {
namespace opt {
@ -50,7 +51,7 @@ bool MetaUnpackPreparePass(const ResourcePtr &resource);
void ReclaimOptimizer();
bool PynativeOptPass(const ResourcePtr &resource);
bool EliminateAdRelatedSpecialOpOptPass(const ResourcePtr &resource);
bool AutoMonadElimOptPass(const FuncGraphPtr &func_graph);
bool MS_API AutoMonadElimOptPass(const FuncGraphPtr &func_graph);
FuncGraphPtr PrimBpOptPassStep1(const opt::irpass::OptimizeIRPassLib &irpass, const ResourcePtr &resource);
FuncGraphPtr PrimBpOptPassStep2(const opt::irpass::OptimizeIRPassLib &irpass, const ResourcePtr &resource);
FuncGraphPtr BpropGraphFinalOptPass(const ResourcePtr &resource);

9
mindspore/ccsrc/pipeline/jit/pipeline.h
View File

@ -35,6 +35,7 @@
#include "backend/graph_compiler/transform.h"
#include "pipeline/jit/base.h"
#include "frontend/parallel/strategy.h"
#include "include/api/visible.h"
namespace mindspore {
extern const char kMsConvert[];
@ -178,7 +179,7 @@ uint32_t GetHcclRankId();
uint32_t GetHcclRankSize();
void InitPipeline();
void FinalizeBackend();
void ClearResAtexit();
void MS_API ClearResAtexit();
void CloseTsd(bool force = false);
void MemoryRecycle();
@ -186,9 +187,9 @@ FuncGraphPtr LoadMindIR(const std::string &file_name, const char *dec_key, const
const std::string &dec_mode, const py::object decrypt = py::none());
// init and exec dataset sub graph
bool InitExecDataset(const std::string &queue_name, int64_t iter_num, int64_t batch_size,
const std::vector<TypePtr> &types, const std::vector<std::vector<int64_t>> &shapes,
const std::vector<int64_t> &input_indexes, const std::string &phase, bool need_run);
bool MS_API InitExecDataset(const std::string &queue_name, int64_t iter_num, int64_t batch_size,
const std::vector<TypePtr> &types, const std::vector<std::vector<int64_t>> &shapes,
const std::vector<int64_t> &input_indexes, const std::string &phase, bool need_run);
// Build and run dataset subgraph for ms backend
bool InitExecDatasetVm(const std::string &queue_name, int64_t size, int64_t batch_size,

3
mindspore/ccsrc/pipeline/jit/static_analysis/auto_monad.h
View File

@ -19,11 +19,12 @@
#include "ir/anf.h"
#include "ir/func_graph.h"
#include "include/api/visible.h"
namespace mindspore::pipeline {
// Run auto-monad, handle side-effects, called from frontend pipeline.
bool AutoMonad(const FuncGraphPtr &func_graph);
bool MS_API AutoMonad(const FuncGraphPtr &func_graph);
// Run auto-monad after grad or Renormalize, handle side-effects, called from frontend opt pass.
bool ReAutoMonad(const FuncGraphPtr &func_graph);

3
mindspore/ccsrc/pipeline/jit/static_analysis/order_enforce.h
View File

@ -18,10 +18,11 @@
#define MINDSPORE_CCSRC_PIPELINE_JIT_ORDER_ENFORCE_H_
#include "ir/func_graph.h"
#include "include/api/visible.h"
namespace mindspore::pipeline {
// Enforce order of execution of the given graph.
void OrderEnforce(const FuncGraphPtr &func_graph);
void MS_API OrderEnforce(const FuncGraphPtr &func_graph);
} // namespace mindspore::pipeline
#endif // MINDSPORE_CCSRC_PIPELINE_JIT_ORDER_ENFORCE_H_