我想打PAC队的第一次评注 #16
|
|
@ -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_;
|
||||
};
|
||||
|
||||
|
|
|
|||
|
|
@ -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_;
|
||||
|
|
|
|||
Loading…
Reference in New Issue