!8346 Pynative support dynamic graph by using ast parse

From: @joylvliang
Reviewed-by: 
Signed-off-by:

mindspore/ccsrc/pipeline/jit/parse/parse_base.h

@@ -90,6 +90,24 @@ const char PYTHON_PARSE_CLASS_ELLIPSIS[] = "create_ellipsis_obj";
 
 // define the common name
 const char NAMED_PRIMITIVE_LEN[] = "len";
+const char NAMED_PRIMITIVE_BODY[] = "body";
+const char NAMED_PRIMITIVE_ASSIGN[] = "Assign";
+const char NAMED_PRIMITIVE_FOR[] = "For";
+const char NAMED_PRIMITIVE_IF[] = "If";
+const char NAMED_PRIMITIVE_WHILE[] = "While";
+const char NAMED_PRIMITIVE_VALUE[] = "value";
+const char NAMED_PRIMITIVE_FUNC[] = "func";
+const char NAMED_PRIMITIVE_TEST[] = "test";
+const char NAMED_PRIMITIVE_LEFT[] = "left";
+const char NAMED_PRIMITIVE_CALL[] = "Call";
+const char NAMED_PRIMITIVE_SUBSCRIPT[] = "Subscript";
+const char NAMED_PRIMITIVE_ATTRIBUTE[] = "Attribute";
+const char NAMED_PRIMITIVE_COMPARE[] = "Compare";
+const char NAMED_PRIMITIVE_NAMECONSTANT[] = "NameConstant";
+const char NAMED_PRIMITIVE_COMPARATORS[] = "comparators";
+const char NAMED_PRIMITIVE_SLICE[] = "slice";
+const char NAMED_PRIMITIVE_NUM[] = "Num";
+const char NAMED_PRIMITIVE_STR[] = "Str";
 const char NAMED_PRIMITIVE_ITER[] = "iter";
 const char NAMED_PRIMITIVE_NEXT[] = "next";
 const char NAMED_PRIMITIVE_GETITEM[] = "getitem";
@@ -104,6 +122,7 @@ const char NAMED_METAGRAPH_UNPACKCALL[] = "unpack_call";
 
 // define NAMED_PRIMITIVE_GETATTR "getattr"
 // define python inline attr
+const char PYTHON_GET_METHOD_LEN[] = "__len__";
 const char PYTHON_GET_METHOD_SELF_CLASS[] = "__self__";
 const char PYTHON_GET_OBJ_DESC[] = "__str__";
 

mindspore/ccsrc/pipeline/pynative/base.h

@@ -28,6 +28,7 @@
 #include "pybind11/pybind11.h"
 #include "ir/anf.h"
 #include "pybind_api/ir/primitive_py.h"
+#include "pipeline/jit/parse/parse.h"
 #include "abstract/abstract_value.h"
 
 namespace mindspore {
@@ -65,6 +66,9 @@ OpExecInfoPtr GenerateOpExecInfo(const py::args &args);
 
 const std::set<std::string> ignore_infer_prim = {"make_ref", "mixed_precision_cast"};
 const std::set<std::string> force_infer_prim = {"TopK", "DropoutGenMask"};
+const std::set<std::string> unchanged_named_primitive = {parse::NAMED_PRIMITIVE_ATTRIBUTE,
+                                                         parse::NAMED_PRIMITIVE_NAMECONSTANT,
+                                                         parse::NAMED_PRIMITIVE_NUM, parse::NAMED_PRIMITIVE_STR};
 }  // namespace pynative
 }  // namespace mindspore
 

mindspore/ccsrc/pipeline/pynative/pynative_execute.h

@@ -100,6 +100,14 @@ class PynativeExecutor : public std::enable_shared_from_this<PynativeExecutor> {
  private:
   PynativeExecutor() = default;
 
+  // check cell struct
+  bool IsDynamicCell(const py::object &cell);
+  bool ParseIfWhileExprNode(const std::shared_ptr<parse::ParseAst> &ast, const py::object &node);
+  bool ParseAssignExprNode(const std::shared_ptr<parse::ParseAst> &ast, const py::object &node);
+  bool ParseForExprNode(const std::shared_ptr<parse::ParseAst> &ast, const py::object &node);
+  std::string ParseNodeName(const std::shared_ptr<parse::ParseAst> &ast, const py::object &node,
+                            parse::AstMainType type);
+
   // run op
   AnfNodePtr GetInput(const py::object &obj, bool op_mask);
   MsBackendPolicy InitEnv(const OpExecInfoPtr &op_exec_info);
@@ -158,9 +166,9 @@ class PynativeExecutor : public std::enable_shared_from_this<PynativeExecutor> {
   static std::mutex instance_lock_;
   static int64_t graph_id_;
   bool grad_flag_{false};
+  bool dynamic_cell_{false};
   bool first_grad_step_{false};
   bool grad_is_running{false};
-  bool dynamic_shape{false};
 
   // Used for construct grad graph
   FuncGraphPtr top_g_{nullptr};
@@ -173,6 +181,7 @@ class PynativeExecutor : public std::enable_shared_from_this<PynativeExecutor> {
 
   // record all info of a graph
   std::unordered_map<FuncGraphPtr, GraphInfo> graph_info_map_;
+  std::unordered_map<std::string, bool> cell_dynamic_map_;
   std::unordered_map<std::string, ResourcePtr> cell_resource_map_;
   std::unordered_map<std::string, std::pair<FuncGraphPtr, bool>> cell_graph_map_;
   // key: cell_id, value: (send_id, weigths_id), cache for sens and weight change