我想打PAC队的第一次评注 #16

Open
zbtrs2 wants to merge 43 commits from ssk015/mindspore2022:master into master
2 changed files with 4 additions and 0 deletions
Showing only changes of commit 76c458d27b - Show all commits

View File

@ -38,8 +38,11 @@ class GraphPartition {
std::vector<GraphSegmentPtr> Partition(const FuncGraphPtr &func_graph, bool *multi_target = nullptr);
private:
// To get the point can be cut or not
bool IsCut(const AnfNodePtr &node);
// New Primitives List After cutting
std::vector<PrimitivePtr> cut_list_;
// the key same as the name_ in the backend object
std::string backend_name_;
};

View File

@ -127,6 +127,7 @@ class BACKEND_EXPORT CompileGraphs {
FinalVMPtr CompileAndLink(const FuncGraphPtr &func_graph);
protected:
// A vector stores instructions and their types
InstSet insts_;
mindspore::HashMap<FuncGraphPtr, int64_t> mapping_;
CompileGraphPtr transform_;