chore: import upstream snapshot with attribution
This commit is contained in:
@@ -0,0 +1,83 @@
|
||||
// Copyright (c) 2019 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 "gtest/gtest.h"
|
||||
#include "paddle/common/ddim.h"
|
||||
#include "paddle/fluid/framework/op_registry.h"
|
||||
#include "paddle/phi/common/place.h"
|
||||
#include "paddle/phi/core/kernel_registry.h"
|
||||
#include "test/cpp/fluid/elementwise/test_elementwise_op_grad_grad.h"
|
||||
|
||||
USE_OP_ITSELF(elementwise_add);
|
||||
PD_DECLARE_KERNEL(add_double_grad, CPU, ALL_LAYOUT);
|
||||
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
|
||||
PD_DECLARE_KERNEL(add_double_grad, GPU, ALL_LAYOUT);
|
||||
#endif
|
||||
|
||||
namespace paddle {
|
||||
namespace operators {
|
||||
|
||||
template <typename T>
|
||||
class TestElementwiseAddGradGradWithoutDDX
|
||||
: public TestElementwiseOpGradGrad<T> {
|
||||
public:
|
||||
TestElementwiseAddGradGradWithoutDDX(const phi::Place &place,
|
||||
const phi::DDim &dims)
|
||||
: TestElementwiseOpGradGrad<T>("elementwise_add_grad_grad",
|
||||
place,
|
||||
dims,
|
||||
{"Y", "DOut", "DDY"},
|
||||
{"DDOut"}) {}
|
||||
|
||||
using TestElementwiseOpGradGrad<T>::feed_datas_;
|
||||
using TestElementwiseOpGradGrad<T>::expected_outs_;
|
||||
using TestElementwiseOpGradGrad<T>::dims_;
|
||||
void ComputeExpectedOuts() override {
|
||||
size_t numel = static_cast<size_t>(common::product(dims_));
|
||||
std::vector<T> dy(numel);
|
||||
std::vector<T> ddout(numel);
|
||||
for (size_t i = 0; i < numel; ++i) {
|
||||
// ddOut = ddX + ddY = ddY if ddX empty
|
||||
ddout[i] = feed_datas_["DDY"][i];
|
||||
}
|
||||
expected_outs_["DDOut"] = ddout;
|
||||
}
|
||||
|
||||
std::unique_ptr<framework::OperatorBase> CreateTestOp() override {
|
||||
auto op = framework::OpRegistry::CreateOp(
|
||||
this->op_type_,
|
||||
{{"Y", {"Y"}}, {"DOut", {"DOut"}}, {"DDY", {"DDY"}}},
|
||||
{{"DDOut", {"DDOut"}}},
|
||||
{{"use_onednn", false}, {"axis", 0}});
|
||||
return op;
|
||||
}
|
||||
};
|
||||
|
||||
TEST(test_elementwise_add_grad_grad_without_ddx, cpu_place) {
|
||||
phi::DDim dims({32, 64});
|
||||
phi::CPUPlace p;
|
||||
TestElementwiseAddGradGradWithoutDDX<float> test(p, dims);
|
||||
ASSERT_TRUE(test.Check());
|
||||
}
|
||||
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
|
||||
TEST(test_elementwise_add_grad_grad_without_ddx, gpu_place) {
|
||||
phi::DDim dims({32, 64});
|
||||
phi::GPUPlace p(0);
|
||||
TestElementwiseAddGradGradWithoutDDX<float> test(p, dims);
|
||||
ASSERT_TRUE(test.Check());
|
||||
}
|
||||
#endif
|
||||
|
||||
} // namespace operators
|
||||
} // namespace paddle
|
||||
Reference in New Issue
Block a user