PaddlePaddle
diff --git a/‎paddle/fluid/pir/utils/name_analysis.cc
Copy file name to clipboard
+14Lines changed: 14 additions & 0 deletions b/‎paddle/fluid/pir/utils/name_analysis.cc
Copy file name to clipboard
+14Lines changed: 14 additions & 0 deletions
diff --git a/‎paddle/fluid/pir/utils/name_analysis.h
Copy file name to clipboard
+3Lines changed: 3 additions & 0 deletions b/‎paddle/fluid/pir/utils/name_analysis.h
Copy file name to clipboard
+3Lines changed: 3 additions & 0 deletions
diff --git a/‎paddle/fluid/pybind/pir.cc
Copy file name to clipboard
+15Lines changed: 15 additions & 0 deletions b/‎paddle/fluid/pybind/pir.cc
Copy file name to clipboard
+15Lines changed: 15 additions & 0 deletions
diff --git a/‎paddle/phi/infermeta/binary.cc
Copy file name to clipboard
+1Lines changed: 1 addition & 0 deletions b/‎paddle/phi/infermeta/binary.cc
Copy file name to clipboard
+1Lines changed: 1 addition & 0 deletions
diff --git a/‎python/paddle/base/framework.py
Copy file name to clipboard
+3-1Lines changed: 3 additions & 1 deletion b/‎python/paddle/base/framework.py
Copy file name to clipboard
+3-1Lines changed: 3 additions & 1 deletion
@@ -59,6 +59,20 @@ pir::Value GetParameterValueByName(const pir::Program &program,
   return value;
 }
 
+std::unordered_map<std::string, pir::Value> GetAllParameterValues(
+    const pir::Program &program) {
+  auto &block = *program.block();
+  std::unordered_map<std::string, pir::Value> values;
+  for (auto &op : block) {
+    if (op.isa<pir::ParameterOp>()) {
+      values[op.attribute("parameter_name")
+                 .dyn_cast<pir::StrAttribute>()
+                 .AsString()] = op.result(0);
+    }
+  }
+  return values;
+}
+
 void SetValueName(pir::Value value, const std::string name) {
   pir::Operation *define_op = value.defining_op();
   if (define_op->isa<pir::ParameterOp>()) {
 
@@ -29,6 +29,9 @@ std::string GetValueFirstName(pir::Value value);
 std::optional<std::string> TryGetValueFirstName(pir::Value value);
 pir::Value GetParameterValueByName(const pir::Program &program,
                                    const std::string &name);
+std::unordered_map<std::string, pir::Value> GetAllParameterValues(
+    const pir::Program &program);
+
 void SetValueName(pir::Value value, const std::string name);
 
 std::map<std::string, std::string> RenameValue(Value value,
 
@@ -358,6 +358,15 @@ void PruneWithInput(const std::vector<pir::Value> &input_vars,
   }
 }
 
+void SetIsTestAttr(const std::shared_ptr<Program> &prog) {
+  for (auto &op : prog->block()->ops()) {
+    if (op->HasAttribute("is_test")) {
+      op->set_attribute(
+          "is_test", pir::BoolAttribute::get(pir::IrContext::Instance(), true));
+    }
+  }
+}
+
 void BindProgram(py::module *m) {
   static int64_t global_prog_seed = 0;
   py::class_<Program, std::shared_ptr<Program>> program(
@@ -422,6 +431,8 @@ void BindProgram(py::module *m) {
            [](const std::shared_ptr<Program> &self) {
              return self->parameters_num();
            })
+      .def("set_is_test_attr",
+           [](const std::shared_ptr<Program> &self) { SetIsTestAttr(self); })
       .def("set_parameters_from",
            [](const std::shared_ptr<Program> &self,
               const std::shared_ptr<Program> &other) {
@@ -507,6 +518,10 @@ void BindProgram(py::module *m) {
            [](Program &self, const std::string &name) {
              return name_analysis::GetParameterValueByName(self, name);
            })
+      .def("get_all_parameter_values",
+           [](Program &self) {
+             return name_analysis::GetAllParameterValues(self);
+           })
       .def("num_ops", [](Program &self) { return self.num_ops(); })
       .def(
           "state_dict",
 
@@ -2150,6 +2150,7 @@ void GatherTreeMeta(const MetaTensor& ids,
                         "The shape of Input(Parents) must be same with the "
                         "shape of Input(Ids)."));
   out->set_dims(ids_dims);
+  out->set_dtype(ids.dtype());
 }
 
 void GridSampleBaseInferMeta(const MetaTensor& x,
 
@@ -789,7 +789,9 @@ def _dygraph_tracer():
 
 def _current_expected_place_():
     global _global_expected_place_
-    if _global_expected_place_ is None:
+    if _global_expected_place_ is None or isinstance(
+        _global_expected_place_, core.Place
+    ):
         if core.is_compiled_with_cuda():
             try:
                 device_count = core.get_cuda_device_count()
Original file line number	Diff line number	Diff line change
`@@ -2150,6 +2150,7 @@ void GatherTreeMeta(const MetaTensor& ids,`
`2150`	`2150`	`"The shape of Input(Parents) must be same with the "`
`2151`	`2151`	`"shape of Input(Ids)."));`
`2152`	`2152`	`out->set_dims(ids_dims);`
	`2153`	`+ out->set_dtype(ids.dtype());`
`2153`	`2154`	`}`
`2154`	`2155`
`2155`	`2156`	`void GridSampleBaseInferMeta(const MetaTensor& x,`