api.cc 3.0 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13
// Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// 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.
14

Y
Yan Chunwei 已提交
15 16
#include <sstream>
#include "paddle/fluid/framework/commit.h"
17 18
#include "paddle/fluid/framework/lod_tensor.h"
#include "paddle/fluid/framework/scope.h"
19
#include "paddle/fluid/inference/api/paddle_inference_api.h"
20
#include "paddle/fluid/inference/api/paddle_pass_builder.h"
Y
Yan Chunwei 已提交
21
#include "paddle/fluid/platform/enforce.h"
22 23 24

namespace paddle {

25 26 27 28 29 30 31 32 33 34 35 36
int PaddleDtypeSize(PaddleDType dtype) {
  switch (dtype) {
    case PaddleDType::FLOAT32:
      return sizeof(float);
    case PaddleDType::INT64:
      return sizeof(int64_t);
    default:
      assert(false);
      return -1;
  }
}

37
PaddleBuf::PaddleBuf(PaddleBuf &&other)
38 39 40 41 42 43 44 45
    : data_(other.data_),
      length_(other.length_),
      memory_owned_(other.memory_owned_) {
  other.memory_owned_ = false;
  other.data_ = nullptr;
  other.length_ = 0;
}

46
PaddleBuf::PaddleBuf(const PaddleBuf &other) { *this = other; }
47

48
PaddleBuf &PaddleBuf::operator=(const PaddleBuf &other) {
49 50 51 52 53 54 55 56 57 58 59 60 61
  if (!other.memory_owned_) {
    data_ = other.data_;
    length_ = other.length_;
    memory_owned_ = other.memory_owned_;
  } else {
    Resize(other.length());
    memcpy(data_, other.data(), other.length());
    length_ = other.length();
    memory_owned_ = true;
  }
  return *this;
}

62
PaddleBuf &PaddleBuf::operator=(PaddleBuf &&other) {
63 64 65 66
  // only the buffer with external memory can be copied
  data_ = other.data_;
  length_ = other.length_;
  memory_owned_ = other.memory_owned_;
67 68 69
  other.data_ = nullptr;
  other.length_ = 0;
  other.memory_owned_ = false;
70 71 72 73 74
  return *this;
}

void PaddleBuf::Resize(size_t length) {
  // Only the owned memory can be reset, the external memory can't be changed.
Y
Yan Chunwei 已提交
75
  if (length_ >= length) return;
76 77
  if (memory_owned_) {
    Free();
Y
Yan Chunwei 已提交
78 79 80 81 82
    data_ = malloc(length);
    length_ = length;
    memory_owned_ = true;
  } else {
    PADDLE_THROW("The memory is allocated externally, can not Resized");
83
  }
84 85
}

86
void PaddleBuf::Reset(void *data, size_t length) {
87 88 89 90 91 92 93 94
  Free();
  memory_owned_ = false;
  data_ = data;
  length_ = length;
}

void PaddleBuf::Free() {
  if (memory_owned_ && data_) {
Y
Yan Chunwei 已提交
95
    PADDLE_ENFORCE_GT(length_, 0);
96
    free(static_cast<char *>(data_));
97 98 99 100 101
    data_ = nullptr;
    length_ = 0;
  }
}

Y
Yan Chunwei 已提交
102 103 104 105 106 107 108 109
std::string get_version() {
  std::stringstream ss;
  ss << "version: " << framework::paddle_version() << "\n";
  ss << "commit: " << framework::paddle_commit() << "\n";
  ss << "branch: " << framework::paddle_compile_branch() << "\n";
  return ss.str();
}

110
}  // namespace paddle