[PIR save/load]Refine dirs and migrate apis

paddle/fluid/pir/serialize_deserialize/include/interface.h

@@ -13,6 +13,8 @@
 // limitations under the License.
 #pragma once
 
+#include <string>
+#include "paddle/phi/core/dense_tensor.h"
 #include "paddle/pir/include/core/program.h"
 namespace pir {
 /**
@@ -60,4 +62,28 @@ void WriteModule(const pir::Program& program,
 void ReadModule(const std::string& file_path,
                 pir::Program* program,
                 const uint64_t& pir_version);
+
+void SaveFunction(const phi::DenseTensor& x,
+                  const std::string& name,
+                  const std::string& file_path,
+                  bool overwrite,
+                  bool save_as_fp16);
+
+void SaveCombineFunction(const std::vector<const phi::DenseTensor*>& x,
+                         const std::vector<std::string>& names,
+                         const std::string& file_path,
+                         bool overwrite,
+                         bool save_as_fp16,
+                         bool save_to_memory);
+
+void LoadFunction(const std::string& file_path,
+                  int64_t seek,
+                  const std::vector<int64_t>& shape,
+                  bool load_as_fp16,
+                  phi::DenseTensor* out);
+
+void LoadCombineFunction(const std::string& file_path,
+                         const std::vector<std::string>& names,
+                         std::vector<phi::DenseTensor*>* out,
+                         bool load_as_fp16);
 }  // namespace pir

paddle/fluid/pir/serialize_deserialize/src/save_load_parameters.cc

@@ -9,14 +9,13 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License. */
 
-#include "paddle/fluid/pir/serialize_deserialize/include/save_load_parameters.h"
-
 #include <cstdint>
 #include <fstream>
 #include <numeric>
 
 #include "glog/logging.h"
 #include "paddle/fluid/framework/lod_tensor.h"
+#include "paddle/fluid/pir/serialize_deserialize/include/interface.h"
 #include "paddle/phi/common/port.h"
 #include "paddle/phi/kernels/funcs/data_type_transform.h"
 
@@ -157,7 +156,7 @@ void LoadFunction(const std::string& file_path,
                           "seek with tensor must great than or equal to 0"));
     paddle::framework::DeserializeFromStream(fin, out, *dev_ctx, seek, shape);
   } else {
-    paddle::framework::DeserializeFromStream(fin, out);
+    paddle::framework::DeserializeFromStream(fin, out, *dev_ctx);
   }
 
   auto in_dtype = out->dtype();
@@ -189,7 +188,7 @@ void LoadCombineFunction(const std::string& file_path,
   const phi::DeviceContext* dev_ctx = GetDeviceContext(*(out->at(0)));
   for (size_t i = 0; i < names.size(); i++) {
     auto tensor = out->at(i);
-    paddle::framework::DeserializeFromStream(fin, tensor);
+    paddle::framework::DeserializeFromStream(fin, tensor, *dev_ctx);
 
     auto in_dtype = tensor->dtype();
     auto out_dtype = load_as_fp16 ? phi::DataType::FLOAT16 : in_dtype;

paddle/fluid/pybind/io.cc

@@ -18,7 +18,6 @@ limitations under the License. */
 #include "paddle/fluid/framework/lod_tensor.h"
 #include "paddle/fluid/framework/selected_rows_utils.h"
 #include "paddle/fluid/pir/serialize_deserialize/include/interface.h"
-#include "paddle/fluid/pir/serialize_deserialize/include/save_load_parameters.h"
 #include "paddle/fluid/platform/enforce.h"
 #include "paddle/fluid/pybind/pybind_variant_caster.h"
 #include "paddle/utils/pybind.h"