diff --git a/paddle/fluid/operators/math/math_function.cc b/paddle/fluid/operators/math/math_function.cc index 1efddc4818671c089a7f86f3e1e4ce16590f01c4..ec21524b0b880edf24a394b653158773a5a1008c 100644 --- a/paddle/fluid/operators/math/math_function.cc +++ b/paddle/fluid/operators/math/math_function.cc @@ -29,7 +29,7 @@ limitations under the License. */ #include "paddle/fluid/operators/math/math_function_impl.h" #include "paddle/fluid/platform/bfloat16.h" #include "paddle/fluid/platform/float16.h" -#include "paddle/pten/kernels/hybird/eigen/common.h" +#include "paddle/pten/kernels/funcs/eigen/common.h" #include "unsupported/Eigen/CXX11/Tensor" namespace paddle { diff --git a/paddle/fluid/operators/math/math_function.cu b/paddle/fluid/operators/math/math_function.cu index a692246a06b1cb9a7424cb6dd1c34ac45667f0a5..378f0426ddfb7669e0b7e0c6325a359c797a9dd1 100644 --- a/paddle/fluid/operators/math/math_function.cu +++ b/paddle/fluid/operators/math/math_function.cu @@ -21,7 +21,7 @@ limitations under the License. */ #include "paddle/fluid/operators/math/math_function_impl.h" #include "paddle/fluid/platform/bfloat16.h" #include "paddle/fluid/platform/float16.h" -#include "paddle/pten/kernels/hybird/eigen/common.h" +#include "paddle/pten/kernels/funcs/eigen/common.h" namespace paddle { namespace operators { diff --git a/paddle/pten/kernels/CMakeLists.txt b/paddle/pten/kernels/CMakeLists.txt index fc04cd797f4a5713766b69329e4e6e4270b1e35e..45724e5d22abde9e218fd0a9decf31631979892a 100644 --- a/paddle/pten/kernels/CMakeLists.txt +++ b/paddle/pten/kernels/CMakeLists.txt @@ -6,8 +6,8 @@ file(WRITE ${kernel_declare_file} "// Generated by the paddle/pten/kernels/CMake # kernel primitive api add_subdirectory(primitive) -# pten hybird functors and functions called by kernels -add_subdirectory(hybird) +# pten functors and functions called by kernels +add_subdirectory(funcs) add_subdirectory(cpu) if(WITH_GPU OR WITH_ROCM) diff --git a/paddle/pten/kernels/cpu/elementwise.h b/paddle/pten/kernels/cpu/elementwise.h index 5a421de1173d8e58a4e07b09deef56e515b2417a..97db997a164788447b58884ce8a1355216b0819f 100644 --- a/paddle/pten/kernels/cpu/elementwise.h +++ b/paddle/pten/kernels/cpu/elementwise.h @@ -19,7 +19,7 @@ limitations under the License. */ #include "paddle/pten/kernels/funcs/elementwise_base.h" #include "paddle/fluid/operators/math/blas.h" -#include "paddle/pten/kernels/hybird/eigen/common.h" +#include "paddle/pten/kernels/funcs/eigen/common.h" namespace pten { diff --git a/paddle/pten/kernels/cpu/reduce.h b/paddle/pten/kernels/cpu/reduce.h index fa603b2163055c4b02a6c6a93d78801cfe87c096..1e9c1e885f44d68cf39a8c9214aa20a9519a90cc 100644 --- a/paddle/pten/kernels/cpu/reduce.h +++ b/paddle/pten/kernels/cpu/reduce.h @@ -22,8 +22,8 @@ #include "paddle/pten/api/lib/utils/storage.h" #include "paddle/pten/core/dense_tensor.h" -#include "paddle/pten/kernels/hybird/eigen/common.h" -#include "paddle/pten/kernels/hybird/transpose.h" +#include "paddle/pten/kernels/funcs/eigen/common.h" +#include "paddle/pten/kernels/funcs/transpose.h" // See Note [ Why still include the fluid headers? ] #include "paddle/fluid/operators/eigen/eigen_function.h" namespace pten { diff --git a/paddle/pten/kernels/hybird/CMakeLists.txt b/paddle/pten/kernels/funcs/CMakeLists.txt similarity index 92% rename from paddle/pten/kernels/hybird/CMakeLists.txt rename to paddle/pten/kernels/funcs/CMakeLists.txt index 5d04bae2eae82014f3147d1c1710f110a5bd530b..32bdc94b95d52c16b19b0bbd6e8747693c3d1e96 100644 --- a/paddle/pten/kernels/hybird/CMakeLists.txt +++ b/paddle/pten/kernels/funcs/CMakeLists.txt @@ -1,5 +1,4 @@ add_subdirectory(eigen) -add_subdirectory(general) cc_library(pten_transpose_cpu SRCS transpose.cc DEPS dense_tensor pten_context) if(WITH_GPU) diff --git a/paddle/pten/kernels/hybird/cuda/CMakeLists.txt b/paddle/pten/kernels/funcs/eigen/CMakeLists.txt similarity index 100% rename from paddle/pten/kernels/hybird/cuda/CMakeLists.txt rename to paddle/pten/kernels/funcs/eigen/CMakeLists.txt diff --git a/paddle/pten/kernels/hybird/eigen/common.h b/paddle/pten/kernels/funcs/eigen/common.h similarity index 100% rename from paddle/pten/kernels/hybird/eigen/common.h rename to paddle/pten/kernels/funcs/eigen/common.h diff --git a/paddle/pten/kernels/hybird/transpose.cc b/paddle/pten/kernels/funcs/transpose.cc similarity index 98% rename from paddle/pten/kernels/hybird/transpose.cc rename to paddle/pten/kernels/funcs/transpose.cc index d1d4350c931617b759d432299a1b219467739fe5..5a40abbd1b7e8ce0836389b6ea58eb6a8e163116 100644 --- a/paddle/pten/kernels/hybird/transpose.cc +++ b/paddle/pten/kernels/funcs/transpose.cc @@ -12,7 +12,7 @@ // See the License for the specific language governing permissions and // limitations under the License. -#include "paddle/pten/kernels/hybird/transpose.h" +#include "paddle/pten/kernels/funcs/transpose.h" #include "paddle/fluid/framework/ddim.h" #include "paddle/pten/backends/cpu/cpu_context.h" #include "paddle/pten/core/dense_tensor.h" diff --git a/paddle/pten/kernels/hybird/transpose.cu b/paddle/pten/kernels/funcs/transpose.cu similarity index 98% rename from paddle/pten/kernels/hybird/transpose.cu rename to paddle/pten/kernels/funcs/transpose.cu index 6ea5e36e106d54fba6e72ce9edfd2fb74f4ed813..e03c538e386823daec5ad0204d95ca3a9cc2c371 100644 --- a/paddle/pten/kernels/hybird/transpose.cu +++ b/paddle/pten/kernels/funcs/transpose.cu @@ -16,7 +16,7 @@ #include "paddle/fluid/memory/memcpy.h" #include "paddle/pten/backends/gpu/gpu_context.h" #include "paddle/pten/core/dense_tensor.h" -#include "paddle/pten/kernels/hybird/transpose.h" +#include "paddle/pten/kernels/funcs/transpose.h" // See Note [ Why still include the fluid headers? ] #include "paddle/fluid/platform/bfloat16.h" diff --git a/paddle/pten/kernels/hybird/transpose.h b/paddle/pten/kernels/funcs/transpose.h similarity index 97% rename from paddle/pten/kernels/hybird/transpose.h rename to paddle/pten/kernels/funcs/transpose.h index 17f52c74a13441ce4b856629fd40de4aa613323e..d0e4dafe2c3b8de56b0feaecd7ddd6e442d9c1a5 100644 --- a/paddle/pten/kernels/hybird/transpose.h +++ b/paddle/pten/kernels/funcs/transpose.h @@ -18,7 +18,7 @@ #include "paddle/pten/core/dense_tensor.h" #include "paddle/fluid/operators/eigen/eigen_function.h" -#include "paddle/pten/kernels/hybird/eigen/common.h" +#include "paddle/pten/kernels/funcs/eigen/common.h" namespace pten { diff --git a/paddle/pten/kernels/gpu/dot_kernel.cu b/paddle/pten/kernels/gpu/dot_kernel.cu index 1f9e7aa3f1cfd54e797cc63063a225389fba7bd2..08d8f83c408dea5a0fff6d7131b77a50f11c9827 100644 --- a/paddle/pten/kernels/gpu/dot_kernel.cu +++ b/paddle/pten/kernels/gpu/dot_kernel.cu @@ -16,7 +16,7 @@ #include "paddle/pten/backends/gpu/gpu_context.h" #include "paddle/pten/core/kernel_registry.h" -#include "paddle/pten/kernels/hybird/eigen/common.h" +#include "paddle/pten/kernels/funcs/eigen/common.h" // See Note [ Why still include the fluid headers? ] #include "paddle/fluid/operators/eigen/eigen_function.h" diff --git a/paddle/pten/kernels/hybird/eigen/CMakeLists.txt b/paddle/pten/kernels/hybird/eigen/CMakeLists.txt deleted file mode 100644 index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0000000000000000000000000000000000000000 diff --git a/paddle/pten/kernels/hybird/eigen/sign.h b/paddle/pten/kernels/hybird/eigen/sign.h deleted file mode 100644 index 0beebda4f39e81dd108568485014383747b580a6..0000000000000000000000000000000000000000 --- a/paddle/pten/kernels/hybird/eigen/sign.h +++ /dev/null @@ -1,41 +0,0 @@ -/* Copyright (c) 2021 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. */ - -#pragma once - -#include "paddle/pten/core/dense_tensor.h" -#include "paddle/pten/kernels/hybird/eigen/common.h" - -// See Note [ Why still include the fluid headers? ] -#include "paddle/fluid/operators/eigen/eigen_function.h" - -namespace pten { -namespace eigen { - -template -void Sign(const DevCtx& dev_ctx, const DenseTensor& x, DenseTensor* out) { - out->mutable_data(); - // TODO(chenweihang): if we design new tensor, we should support - // the low-level calc functor use new tensor as input, - // which may be a big project! - auto eigen_out = pten::EigenVector::Flatten(*out); - auto eigen_x = pten::EigenVector::Flatten(x); - - auto& dev = *dev_ctx.eigen_device(); - paddle::operators::EigenSign, T>::Eval( - dev, eigen_out, eigen_x); -} - -} // namespace eigen -} // namespace pten diff --git a/paddle/pten/kernels/hybird/general/CMakeLists.txt b/paddle/pten/kernels/hybird/general/CMakeLists.txt deleted file mode 100644 index e69de29bb2d1d6434b8b29ae775ad8c2e48c5391..0000000000000000000000000000000000000000 diff --git a/paddle/pten/kernels/impl/dot_grad_kernel_impl.h b/paddle/pten/kernels/impl/dot_grad_kernel_impl.h index 16c87bbab474abe5c0d26394e6b5476d8e9dfa14..39cdbad5146deb3235a78d823fbbcf51369ead18 100644 --- a/paddle/pten/kernels/impl/dot_grad_kernel_impl.h +++ b/paddle/pten/kernels/impl/dot_grad_kernel_impl.h @@ -15,7 +15,7 @@ limitations under the License. */ #pragma once #include "paddle/pten/core/dense_tensor.h" -#include "paddle/pten/kernels/hybird/eigen/common.h" +#include "paddle/pten/kernels/funcs/eigen/common.h" #include "paddle/pten/kernels/complex_kernel.h" diff --git a/paddle/pten/kernels/impl/full_kernel_impl.h b/paddle/pten/kernels/impl/full_kernel_impl.h index 9be40e22a0360ee23050f59b8a7598333def38ad..79ca63c9b0669b79b63f7fb4799d91d13263a0fe 100644 --- a/paddle/pten/kernels/impl/full_kernel_impl.h +++ b/paddle/pten/kernels/impl/full_kernel_impl.h @@ -17,7 +17,7 @@ limitations under the License. */ #include "paddle/pten/common/scalar.h" #include "paddle/pten/common/scalar_array.h" #include "paddle/pten/core/dense_tensor.h" -#include "paddle/pten/kernels/hybird/eigen/common.h" +#include "paddle/pten/kernels/funcs/eigen/common.h" // See Note [ Why still include the fluid headers? ] #include "paddle/fluid/operators/eigen/eigen_function.h" diff --git a/paddle/pten/kernels/impl/scale_kernel_impl.h b/paddle/pten/kernels/impl/scale_kernel_impl.h index 937b3115e63b37003d34cca35aa8b6c949fcfdc6..2e0b158b36b8dd9ce10f447e54c9049f48919dba 100644 --- a/paddle/pten/kernels/impl/scale_kernel_impl.h +++ b/paddle/pten/kernels/impl/scale_kernel_impl.h @@ -16,7 +16,7 @@ limitations under the License. */ #include "paddle/pten/common/scalar.h" #include "paddle/pten/core/dense_tensor.h" -#include "paddle/pten/kernels/hybird/eigen/common.h" +#include "paddle/pten/kernels/funcs/eigen/common.h" // See Note [ Why still include the fluid headers? ] #include "paddle/fluid/operators/eigen/eigen_function.h" diff --git a/paddle/pten/kernels/impl/sign_kernel_impl.h b/paddle/pten/kernels/impl/sign_kernel_impl.h index 306206f1fc3fa070496cb316fcf0a66662e99ad9..655cda762ee1a03c451981d3890cd3176390004f 100644 --- a/paddle/pten/kernels/impl/sign_kernel_impl.h +++ b/paddle/pten/kernels/impl/sign_kernel_impl.h @@ -15,7 +15,7 @@ limitations under the License. */ #pragma once #include "paddle/pten/core/dense_tensor.h" -#include "paddle/pten/kernels/hybird/eigen/common.h" +#include "paddle/pten/kernels/funcs/eigen/common.h" // See Note [ Why still include the fluid headers? ] #include "paddle/fluid/operators/eigen/eigen_function.h"