“b72beee4f0a9521e97d7f8a5c6ead51f38c05522”上不存在“paddle/legacy/gserver/layers/CudnnConvBaseLayer.h”
variable_visitor.cc 4.7 KB
Newer Older
Y
Yu Yang 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
//   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.

#include "paddle/fluid/framework/details/variable_visitor.h"
#include "paddle/fluid/framework/selected_rows.h"
namespace paddle {
namespace framework {
namespace details {
template <typename Func>
C
chengduoZH 已提交
21
static void VisitVariable(Variable* var, Func* func) {
Y
Yu Yang 已提交
22
  if (var->IsType<LoDTensor>()) {
C
chengduoZH 已提交
23
    (*func)(var->GetMutable<LoDTensor>());
Y
Yu Yang 已提交
24
  } else if (var->IsType<SelectedRows>()) {
C
chengduoZH 已提交
25
    (*func)(var->GetMutable<SelectedRows>());
Y
Yu Yang 已提交
26 27 28 29 30 31
  } else {
    PADDLE_THROW("Not supported type %s", var->Type().name());
  }
}

template <typename Func>
C
chengduoZH 已提交
32
static void VisitVariable(const Variable& var, Func* func) {
Y
Yu Yang 已提交
33
  if (var.IsType<LoDTensor>()) {
C
chengduoZH 已提交
34
    (*func)(var.Get<LoDTensor>());
Y
Yu Yang 已提交
35
  } else if (var.IsType<SelectedRows>()) {
C
chengduoZH 已提交
36
    (*func)(var.Get<SelectedRows>());
Y
Yu Yang 已提交
37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58
  } else {
    PADDLE_THROW("Not supported type %s", var.Type().name());
  }
}

struct TensorVisitor {
  Tensor* result_{nullptr};

  void operator()(LoDTensor* tensor) { result_ = tensor; }

  void operator()(SelectedRows* selected_rows) {
    result_ = selected_rows->mutable_value();
  }

  template <typename T>
  void operator()() {
    PADDLE_THROW("Not Support to get LoDTensor from %s", typeid(T).name());
  }
};

Tensor& VariableVisitor::GetMutableTensor(Variable* var) {
  TensorVisitor vistor;
C
chengduoZH 已提交
59
  VisitVariable(var, &vistor);
Y
Yu Yang 已提交
60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87
  return *vistor.result_;
}

struct ShareDimsAndLoDVisitor {
  Variable* trg_;
  void operator()(const LoDTensor& val) {
    auto* tensor = trg_->GetMutable<LoDTensor>();
    tensor->set_layout(val.layout());
    tensor->set_lod(val.lod());
    tensor->Resize(val.dims());
  }

  void operator()(const SelectedRows& val) {
    auto* selected_rows = trg_->GetMutable<SelectedRows>();
    selected_rows->set_rows(val.rows());
    selected_rows->set_height(val.height());
    selected_rows->mutable_value()->Resize(val.value().dims());
  }

  template <typename T>
  void operator()(const T&) {
    PADDLE_ENFORCE("ShareDimsAndLoD is not supported by type %s",
                   typeid(T).name());
  }
};

void VariableVisitor::ShareDimsAndLoD(const Variable& src, Variable* trg) {
  ShareDimsAndLoDVisitor visitor{trg};
C
chengduoZH 已提交
88
  VisitVariable(src, &visitor);
Y
Yu Yang 已提交
89 90
}

C
chengduoZH 已提交
91
struct EnforceShapeAndDTypeEQVisitor {
C
chengduoZH 已提交
92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132
  const Variable* trg_;

  void operator()(const LoDTensor& src) {
    auto& tensor = trg_->Get<LoDTensor>();
    PADDLE_ENFORCE_EQ(
        src.place().which(), tensor.place().which(),
        "The Places of the two Variable must be all on CPU or all on GPU.");
    PADDLE_ENFORCE_EQ(src.type(), tensor.type(),
                      "The dtype of the two Variable is not equal.");
    PADDLE_ENFORCE_EQ(src.dims(), tensor.dims(),
                      "The dims of the two Variable is not equal.");
    PADDLE_ENFORCE_EQ(src.lod(), tensor.lod(),
                      "The lod of the two Variable is not equal.");
    PADDLE_ENFORCE_EQ(src.layout(), tensor.layout(),
                      "The layout of the two Variable's tensor is not equal.");
  }

  void operator()(const SelectedRows& src) {
    auto& selected_rows = trg_->Get<SelectedRows>();
    PADDLE_ENFORCE_EQ(
        src.place().which(), selected_rows.place().which(),
        "The Places of the two Variable must be all on CPU or all on GPU.");
    PADDLE_ENFORCE_EQ(src.value().type(), selected_rows.value().type(),
                      "The dtype of the two Variable is not equal.");
    PADDLE_ENFORCE_EQ(src.value().layout(), selected_rows.value().layout(),
                      "The layout of the two Variable's tensor is not equal.");
    PADDLE_ENFORCE_EQ(src.height(), selected_rows.height(),
                      "The height of the two Variable is not equal.");
    PADDLE_ENFORCE_EQ(src.GetCompleteDims(), selected_rows.GetCompleteDims(),
                      "The dims of the two Variable is not equal.");
  }

  template <typename T>
  void operator()(const T&) {
    PADDLE_ENFORCE("EnforceShapeAndDTypeEQ is not supported by type %s",
                   typeid(T).name());
  }
};

void VariableVisitor::EnforceShapeAndDTypeEQ(const Variable& var1,
                                             const Variable& var2) {
C
chengduoZH 已提交
133
  EnforceShapeAndDTypeEQVisitor visitor{&var1};
C
chengduoZH 已提交
134 135 136
  VisitVariable(var2, &visitor);
}

Y
Yu Yang 已提交
137 138 139
}  // namespace details
}  // namespace framework
}  // namespace paddle