api.cc 3.3 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
int PaddleDtypeSize(PaddleDType dtype) {
  switch (dtype) {
    case PaddleDType::FLOAT32:
      return sizeof(float);
    case PaddleDType::INT64:
      return sizeof(int64_t);
31 32
    case PaddleDType::INT32:
      return sizeof(int32_t);
33 34 35 36 37 38
    default:
      assert(false);
      return -1;
  }
}

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

48
PaddleBuf::PaddleBuf(const PaddleBuf &other) { *this = other; }
49

50
PaddleBuf &PaddleBuf::operator=(const PaddleBuf &other) {
51 52 53 54 55 56
  if (!other.memory_owned_) {
    data_ = other.data_;
    length_ = other.length_;
    memory_owned_ = other.memory_owned_;
  } else {
    Resize(other.length());
57 58 59 60 61 62 63 64 65
    // if other.length() == 0 or other.data() == nullptr, then the memcpy
    // behavior is undefined
    if (other.length() && other.data())
      memcpy(data_, other.data(), other.length());
    else if (other.length())
      PADDLE_THROW(
          "Invalid argument, null pointer data with length %u is passed",
          other.length());

66 67 68 69 70 71
    length_ = other.length();
    memory_owned_ = true;
  }
  return *this;
}

72
PaddleBuf &PaddleBuf::operator=(PaddleBuf &&other) {
73 74 75 76
  // only the buffer with external memory can be copied
  data_ = other.data_;
  length_ = other.length_;
  memory_owned_ = other.memory_owned_;
77 78 79
  other.data_ = nullptr;
  other.length_ = 0;
  other.memory_owned_ = false;
80 81 82 83 84
  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 已提交
85
  if (length_ >= length) return;
86 87
  if (memory_owned_) {
    Free();
88
    data_ = new char[length];
Y
Yan Chunwei 已提交
89 90 91 92
    length_ = length;
    memory_owned_ = true;
  } else {
    PADDLE_THROW("The memory is allocated externally, can not Resized");
93
  }
94 95
}

96
void PaddleBuf::Reset(void *data, size_t length) {
97 98 99 100 101 102 103 104
  Free();
  memory_owned_ = false;
  data_ = data;
  length_ = length;
}

void PaddleBuf::Free() {
  if (memory_owned_ && data_) {
T
tensor-tang 已提交
105
    PADDLE_ENFORCE_GT(length_, 0UL);
106
    delete[] static_cast<char *>(data_);
107 108 109 110 111
    data_ = nullptr;
    length_ = 0;
  }
}

Y
Yan Chunwei 已提交
112 113 114 115 116 117 118 119
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();
}

120
}  // namespace paddle