Skip to content

P
Paddle

PaddlePaddle / Paddle
大约 2 年 前同步成功

通知 2325
Star 20933
Fork 5424
P
Paddle
未验证 提交 4a338796 编写于 作者: C Chen Weihang 提交者： GitHub
浏览文件
操作

[PTen->Phi PR2] Rename PT_REGISTER macro to PD_REGISTER (#39790) 

* unify register macro

* rename declare macro

* fix infrt error
上级 73bf9673
隐藏空白更改
内联 并排
Showing with 354 addition and 352 deletion
cmake/pten.cmake
浏览文件 @ 4a338796
...@@ -58,26 +58,26 @@ endfunction() ...@@ -58,26 +58,26 @@ endfunction()
function(kernel_declare TARGET_LIST) function(kernel_declare TARGET_LIST)
foreach(kernel_path ${TARGET_LIST}) foreach(kernel_path ${TARGET_LIST})
file(READ ${kernel_path} kernel_impl) file(READ ${kernel_path} kernel_impl)
# TODO(chenweihang): rename PT_REGISTER_KERNEL to PT_REGISTER_KERNEL # TODO(chenweihang): rename PD_REGISTER_KERNEL to PD_REGISTER_KERNEL
# NOTE(chenweihang): now we don't recommend to use digit in kernel name # NOTE(chenweihang): now we don't recommend to use digit in kernel name
string(REGEX MATCH "(PT_REGISTER_KERNEL|PT_REGISTER_GENERAL_KERNEL)\\([ \t\r\n]*[a-z0-9_]*," first_registry "${kernel_impl}") string(REGEX MATCH "(PD_REGISTER_KERNEL|PD_REGISTER_GENERAL_KERNEL)\\([ \t\r\n]*[a-z0-9_]*," first_registry "${kernel_impl}")
if (NOT first_registry STREQUAL "") if (NOT first_registry STREQUAL "")
# parse the first kernel name # parse the first kernel name
string(REPLACE "PT_REGISTER_KERNEL(" "" kernel_name "${first_registry}") string(REPLACE "PD_REGISTER_KERNEL(" "" kernel_name "${first_registry}")
string(REPLACE "PT_REGISTER_GENERAL_KERNEL(" "" kernel_name "${kernel_name}") string(REPLACE "PD_REGISTER_GENERAL_KERNEL(" "" kernel_name "${kernel_name}")
string(REPLACE "," "" kernel_name "${kernel_name}") string(REPLACE "," "" kernel_name "${kernel_name}")
string(REGEX REPLACE "[ \t\r\n]+" "" kernel_name "${kernel_name}") string(REGEX REPLACE "[ \t\r\n]+" "" kernel_name "${kernel_name}")
# append kernel declare into declarations.h # append kernel declare into declarations.h
# TODO(chenweihang): default declare ALL_LAYOUT for each kernel # TODO(chenweihang): default declare ALL_LAYOUT for each kernel
if (${kernel_path} MATCHES "./cpu\/") if (${kernel_path} MATCHES "./cpu\/")
file(APPEND ${kernel_declare_file} "PT_DECLARE_KERNEL(${kernel_name}, CPU, ALL_LAYOUT);\n") file(APPEND ${kernel_declare_file} "PD_DECLARE_KERNEL(${kernel_name}, CPU, ALL_LAYOUT);\n")
elseif (${kernel_path} MATCHES "./gpu\/") elseif (${kernel_path} MATCHES "./gpu\/")
file(APPEND ${kernel_declare_file} "PT_DECLARE_KERNEL(${kernel_name}, GPU, ALL_LAYOUT);\n") file(APPEND ${kernel_declare_file} "PD_DECLARE_KERNEL(${kernel_name}, GPU, ALL_LAYOUT);\n")
elseif (${kernel_path} MATCHES "./xpu\/") elseif (${kernel_path} MATCHES "./xpu\/")
file(APPEND ${kernel_declare_file} "PT_DECLARE_KERNEL(${kernel_name}, XPU, ALL_LAYOUT);\n") file(APPEND ${kernel_declare_file} "PD_DECLARE_KERNEL(${kernel_name}, XPU, ALL_LAYOUT);\n")
else () else ()
# deal with device independent kernel, now we use CPU temporaary # deal with device independent kernel, now we use CPU temporaary
file(APPEND ${kernel_declare_file} "PT_DECLARE_KERNEL(${kernel_name}, CPU, ALL_LAYOUT);\n") file(APPEND ${kernel_declare_file} "PD_DECLARE_KERNEL(${kernel_name}, CPU, ALL_LAYOUT);\n")
endif() endif()
endif() endif()
endforeach() endforeach()
...@@ -285,9 +285,9 @@ endfunction() ...@@ -285,9 +285,9 @@ endfunction()
function(append_op_util_declare TARGET) function(append_op_util_declare TARGET)
file(READ ${CMAKE_CURRENT_SOURCE_DIR}/${TARGET} target_content) file(READ ${CMAKE_CURRENT_SOURCE_DIR}/${TARGET} target_content)
string(REGEX MATCH "(PT_REGISTER_BASE_KERNEL_NAME|PT_REGISTER_ARG_MAPPING_FN)\\([ \t\r\n]*[a-z0-9_]*" util_registrar "${target_content}") string(REGEX MATCH "(PD_REGISTER_BASE_KERNEL_NAME|PD_REGISTER_ARG_MAPPING_FN)\\([ \t\r\n]*[a-z0-9_]*" util_registrar "${target_content}")
string(REPLACE "PT_REGISTER_ARG_MAPPING_FN" "PT_DECLARE_ARG_MAPPING_FN" util_declare "${util_registrar}") string(REPLACE "PD_REGISTER_ARG_MAPPING_FN" "PD_DECLARE_ARG_MAPPING_FN" util_declare "${util_registrar}")
string(REPLACE "PT_REGISTER_BASE_KERNEL_NAME" "PT_DECLARE_BASE_KERNEL_NAME" util_declare "${util_declare}") string(REPLACE "PD_REGISTER_BASE_KERNEL_NAME" "PD_DECLARE_BASE_KERNEL_NAME" util_declare "${util_declare}")
string(APPEND util_declare ");\n") string(APPEND util_declare ");\n")
file(APPEND ${op_utils_header} "${util_declare}") file(APPEND ${op_utils_header} "${util_declare}")
endfunction() endfunction()
......
paddle/fluid/framework/infershape_utils_test.cc
浏览文件 @ 4a338796
...@@ -118,7 +118,7 @@ REGISTER_OPERATOR(infer_shape_utils_test, ...@@ -118,7 +118,7 @@ REGISTER_OPERATOR(infer_shape_utils_test,
paddle::framework::InferShapeUtilsTestOpMaker, paddle::framework::InferShapeUtilsTestOpMaker,
InferShapeUtilsTestInferShapeFunctor); InferShapeUtilsTestInferShapeFunctor);
PT_REGISTER_KERNEL(infer_shape_utils_test, CPU, ALL_LAYOUT, PD_REGISTER_KERNEL(infer_shape_utils_test, CPU, ALL_LAYOUT,
paddle::framework::InferShapeUtilsTestKernel, int) {} paddle::framework::InferShapeUtilsTestKernel, int) {}
TEST(InferShapeUtilsTest, ALL) { TEST(InferShapeUtilsTest, ALL) {
......
paddle/phi/api/ext/op_kernel_info.h
浏览文件 @ 4a338796
...@@ -630,16 +630,16 @@ class PADDLE_API OpKernelInfoBuilder { ...@@ -630,16 +630,16 @@ class PADDLE_API OpKernelInfoBuilder {
}; };
/////////////////////// Custom kernel register API ///////////////////////// /////////////////////// Custom kernel register API /////////////////////////
// For inference: compile directly with framework // For inference: compile directly with framework
// Call after PD_REGISTER_KERNEL(...) // Call after PD_REGISTER_BUILTIN_KERNEL(...)
void RegisterAllCustomKernel(); void RegisterAllCustomKernel();
//////////////// Custom kernel register macro ///////////////////// //////////////// Custom kernel register macro /////////////////////
// Refer to paddle/phi/core/kernel_registry.h, we can not use // Refer to paddle/phi/core/kernel_registry.h, we can not use
// PT_REGISTER_KERNEL directly, common macros and functions are // PD_REGISTER_KERNEL directly, common macros and functions are
// not ready for custom kernel now. // not ready for custom kernel now.
// Difference: custom_kernel stores all kernels' info into global // Difference: custom_kernel stores all kernels' info into global
// g_custom_kernel_info_map before loading and registering into // g_custom_kernel_info_map before loading and registering into
// pten kernel management. Only providing PD_REGISTER_KERNEL which // pten kernel management. Only providing PD_REGISTER_BUILTIN_KERNEL which
// supports 2 template arguments. // supports 2 template arguments.
#define PD_BACKEND(arg__) phi::Backend::arg__ #define PD_BACKEND(arg__) phi::Backend::arg__
...@@ -666,11 +666,12 @@ void RegisterAllCustomKernel(); ...@@ -666,11 +666,12 @@ void RegisterAllCustomKernel();
#define PD_ID __LINE__ #define PD_ID __LINE__
#endif #endif
#define PD_REGISTER_KERNEL(kernel_name, backend, layout, func, cpp_dtype, ...) \ #define PD_REGISTER_BUILTIN_KERNEL( \
STATIC_ASSERT_GLOBAL_NAMESPACE( \ kernel_name, backend, layout, func, cpp_dtype, ...) \
_reg_custom_kernel_ns_check_##kernel_name##_##backend##_##layout, \ STATIC_ASSERT_GLOBAL_NAMESPACE( \
"PD_REGISTER_KERNEL must be called in global namespace."); \ _reg_custom_kernel_ns_check_##kernel_name##_##backend##_##layout, \
_PD_REGISTER_2TA_KERNEL( \ "PD_REGISTER_BUILTIN_KERNEL must be called in global namespace."); \
_PD_REGISTER_2TA_KERNEL( \
kernel_name, backend, layout, func, cpp_dtype, ##__VA_ARGS__) kernel_name, backend, layout, func, cpp_dtype, ##__VA_ARGS__)
// WIN32 is not supported // WIN32 is not supported
......
paddle/phi/api/lib/api_declare.h
浏览文件 @ 4a338796
...@@ -17,6 +17,6 @@ limitations under the License. */ ...@@ -17,6 +17,6 @@ limitations under the License. */
// api symbols declare, remove in the future // api symbols declare, remove in the future
#include "paddle/phi/api/lib/api_registry.h" #include "paddle/phi/api/lib/api_registry.h"
PT_DECLARE_API(Math); PD_DECLARE_API(Math);
PT_DECLARE_API(Utils); PD_DECLARE_API(Utils);
PT_DECLARE_API(SparseApi); PD_DECLARE_API(SparseApi);
paddle/phi/api/lib/api_registry.h
浏览文件 @ 4a338796
...@@ -36,10 +36,10 @@ namespace experimental { ...@@ -36,10 +36,10 @@ namespace experimental {
*/ */
// use to declare symbol // use to declare symbol
#define PT_REGISTER_API(name) \ #define PD_REGISTER_API(name) \
PADDLE_API int RegisterSymbolsFor##name() { return 0; } PADDLE_API int RegisterSymbolsFor##name() { return 0; }
#define PT_DECLARE_API(name) \ #define PD_DECLARE_API(name) \
extern PADDLE_API int RegisterSymbolsFor##name(); \ extern PADDLE_API int RegisterSymbolsFor##name(); \
UNUSED static int use_pten_api_##name = RegisterSymbolsFor##name() UNUSED static int use_pten_api_##name = RegisterSymbolsFor##name()
......
paddle/phi/api/lib/manual_api.cc
浏览文件 @ 4a338796
...@@ -27,15 +27,15 @@ limitations under the License. */ ...@@ -27,15 +27,15 @@ limitations under the License. */
#include "paddle/phi/core/meta_tensor.h" #include "paddle/phi/core/meta_tensor.h"
#include "paddle/phi/infermeta/unary.h" #include "paddle/phi/infermeta/unary.h"
PT_DECLARE_KERNEL(copy, CPU, ALL_LAYOUT); PD_DECLARE_KERNEL(copy, CPU, ALL_LAYOUT);
PT_DECLARE_KERNEL(split, CPU, ALL_LAYOUT); PD_DECLARE_KERNEL(split, CPU, ALL_LAYOUT);
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP) #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
PT_DECLARE_KERNEL(copy, GPU, ALL_LAYOUT); PD_DECLARE_KERNEL(copy, GPU, ALL_LAYOUT);
#endif #endif
#ifdef PADDLE_WITH_XPU #ifdef PADDLE_WITH_XPU
PT_DECLARE_KERNEL(copy, XPU, ALL_LAYOUT); PD_DECLARE_KERNEL(copy, XPU, ALL_LAYOUT);
#endif #endif
namespace paddle { namespace paddle {
...@@ -147,4 +147,4 @@ PADDLE_API std::vector<Tensor> split(const Tensor& x, ...@@ -147,4 +147,4 @@ PADDLE_API std::vector<Tensor> split(const Tensor& x,
} // namespace experimental } // namespace experimental
} // namespace paddle } // namespace paddle
PT_REGISTER_API(Utils); PD_REGISTER_API(Utils);
paddle/phi/api/lib/op_kernel_info.cc
浏览文件 @ 4a338796
...@@ -86,7 +86,7 @@ OpKernelInfoBuilder& OpKernelInfoBuilder::ArgsDef(CustomKernelArgsDefFn func) { ...@@ -86,7 +86,7 @@ OpKernelInfoBuilder& OpKernelInfoBuilder::ArgsDef(CustomKernelArgsDefFn func) {
/////////////////////// Op register API ///////////////////////// /////////////////////// Op register API /////////////////////////
// For inference: compile directly with framework // For inference: compile directly with framework
// Call after PD_REGISTER_KERNEL(...) // Call after PD_REGISTER_BUILTIN_KERNEL(...)
void RegisterAllCustomKernel() { void RegisterAllCustomKernel() {
auto& op_kernel_info_map = OpKernelInfoMap::Instance(); auto& op_kernel_info_map = OpKernelInfoMap::Instance();
framework::RegisterKernelWithMetaInfoMap(op_kernel_info_map); framework::RegisterKernelWithMetaInfoMap(op_kernel_info_map);
......
paddle/phi/api/lib/sparse_api.cc
浏览文件 @ 4a338796
...@@ -22,20 +22,20 @@ limitations under the License. */ ...@@ -22,20 +22,20 @@ limitations under the License. */
#include "paddle/phi/core/kernel_registry.h" #include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/infermeta/unary.h" #include "paddle/phi/infermeta/unary.h"
PT_DECLARE_KERNEL(dense_to_sparse_coo, CPU, ALL_LAYOUT); PD_DECLARE_KERNEL(dense_to_sparse_coo, CPU, ALL_LAYOUT);
PT_DECLARE_KERNEL(sparse_csr_to_coo, CPU, ALL_LAYOUT); PD_DECLARE_KERNEL(sparse_csr_to_coo, CPU, ALL_LAYOUT);
PT_DECLARE_KERNEL(dense_to_sparse_csr, CPU, ALL_LAYOUT); PD_DECLARE_KERNEL(dense_to_sparse_csr, CPU, ALL_LAYOUT);
PT_DECLARE_KERNEL(sparse_coo_to_csr, CPU, ALL_LAYOUT); PD_DECLARE_KERNEL(sparse_coo_to_csr, CPU, ALL_LAYOUT);
PT_DECLARE_KERNEL(sparse_coo_to_dense, CPU, ALL_LAYOUT); PD_DECLARE_KERNEL(sparse_coo_to_dense, CPU, ALL_LAYOUT);
PT_DECLARE_KERNEL(sparse_csr_to_dense, CPU, ALL_LAYOUT); PD_DECLARE_KERNEL(sparse_csr_to_dense, CPU, ALL_LAYOUT);
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP) #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
PT_DECLARE_KERNEL(dense_to_sparse_coo, GPU, ALL_LAYOUT); PD_DECLARE_KERNEL(dense_to_sparse_coo, GPU, ALL_LAYOUT);
PT_DECLARE_KERNEL(sparse_csr_to_coo, GPU, ALL_LAYOUT); PD_DECLARE_KERNEL(sparse_csr_to_coo, GPU, ALL_LAYOUT);
PT_DECLARE_KERNEL(dense_to_sparse_csr, GPU, ALL_LAYOUT); PD_DECLARE_KERNEL(dense_to_sparse_csr, GPU, ALL_LAYOUT);
PT_DECLARE_KERNEL(sparse_coo_to_csr, GPU, ALL_LAYOUT); PD_DECLARE_KERNEL(sparse_coo_to_csr, GPU, ALL_LAYOUT);
PT_DECLARE_KERNEL(sparse_coo_to_dense, GPU, ALL_LAYOUT); PD_DECLARE_KERNEL(sparse_coo_to_dense, GPU, ALL_LAYOUT);
PT_DECLARE_KERNEL(sparse_csr_to_dense, GPU, ALL_LAYOUT); PD_DECLARE_KERNEL(sparse_csr_to_dense, GPU, ALL_LAYOUT);
#endif #endif
namespace paddle { namespace paddle {
...@@ -228,4 +228,4 @@ PADDLE_API Tensor to_dense(const Tensor& x, Backend backend) { ...@@ -228,4 +228,4 @@ PADDLE_API Tensor to_dense(const Tensor& x, Backend backend) {
} // namespace experimental } // namespace experimental
} // namespace paddle } // namespace paddle
PT_REGISTER_API(SparseApi); PD_REGISTER_API(SparseApi);
paddle/phi/common/backend.h
浏览文件 @ 4a338796
...@@ -71,17 +71,17 @@ enum class Backend : uint8_t { ...@@ -71,17 +71,17 @@ enum class Backend : uint8_t {
* Of course, we have also considered solving this problem through different * Of course, we have also considered solving this problem through different
* named macros, for example, if we define * named macros, for example, if we define
* *
* PT_REGISTER_KERNEL_FOR_ALL_BACKEND * PD_REGISTER_KERNEL_FOR_ALL_BACKEND
* *
* Based on this design pattern, the dtype and layout also have the same * Based on this design pattern, the dtype and layout also have the same
* requirements, this cause we need to define a series of macros * requirements, this cause we need to define a series of macros
* *
* PT_REGISTER_KERNEL_FOR_ALL_DTYPE * PD_REGISTER_KERNEL_FOR_ALL_DTYPE
* PT_REGISTER_KERNEL_FOR_ALL_LAYOUT * PD_REGISTER_KERNEL_FOR_ALL_LAYOUT
* PT_REGISTER_KERNEL_FOR_ALL_BACKEND_AND_LAYOUT * PD_REGISTER_KERNEL_FOR_ALL_BACKEND_AND_LAYOUT
* PT_REGISTER_KERNEL_FOR_ALL_BACKEND_AND_DTYPE * PD_REGISTER_KERNEL_FOR_ALL_BACKEND_AND_DTYPE
* PT_REGISTER_KERNEL_FOR_ALL_LAYOUT_AND_DTYPE * PD_REGISTER_KERNEL_FOR_ALL_LAYOUT_AND_DTYPE
* PT_REGISTER_KERNEL_FOR_ALL_BACKEND_AND_LAYOUT_AND_DTYPE * PD_REGISTER_KERNEL_FOR_ALL_BACKEND_AND_LAYOUT_AND_DTYPE
* *
* It makes the system of registering macros more complicated, we think * It makes the system of registering macros more complicated, we think
* this is not a simple design, so we still adopt the design of providing * this is not a simple design, so we still adopt the design of providing
......
paddle/phi/core/compat/op_utils.h
浏览文件 @ 4a338796
...@@ -164,34 +164,34 @@ struct ArgumentMappingFnRegistrar { ...@@ -164,34 +164,34 @@ struct ArgumentMappingFnRegistrar {
} }
}; };
#define PT_REGISTER_BASE_KERNEL_NAME(op_type, base_kernel_name) \ #define PD_REGISTER_BASE_KERNEL_NAME(op_type, base_kernel_name) \
PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \ PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \
pt_register_base_kernel_name_ns_check_##op_type, \ PD_REGISTER_base_kernel_name_ns_check_##op_type, \
"PT_REGISTER_BASE_KERNEL_NAME must be called in global namespace."); \ "PD_REGISTER_BASE_KERNEL_NAME must be called in global namespace."); \
static const ::phi::BaseKernelNameRegistrar \ static const ::phi::BaseKernelNameRegistrar \
__registrar_base_kernel_name_for_##op_type(#op_type, #base_kernel_name); \ __registrar_base_kernel_name_for_##op_type(#op_type, #base_kernel_name); \
int TouchBaseKernelNameSymbol_##op_type() { return 0; } int TouchBaseKernelNameSymbol_##op_type() { return 0; }
#define PT_DECLARE_BASE_KERNEL_NAME(op_type) \ #define PD_DECLARE_BASE_KERNEL_NAME(op_type) \
PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \ PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \
pt_declare_ai_name_ns_check_##op_type, \ PD_DECLARE_ai_name_ns_check_##op_type, \
"PT_DECLARE_BASE_KERNEL_NAME must be called in global namespace."); \ "PD_DECLARE_BASE_KERNEL_NAME must be called in global namespace."); \
extern int TouchBaseKernelNameSymbol_##op_type(); \ extern int TouchBaseKernelNameSymbol_##op_type(); \
UNUSED static int __declare_base_kernel_name_symbol_for_##op_type = \ UNUSED static int __declare_base_kernel_name_symbol_for_##op_type = \
TouchBaseKernelNameSymbol_##op_type() TouchBaseKernelNameSymbol_##op_type()
#define PT_REGISTER_ARG_MAPPING_FN(op_type, arg_mapping_fn) \ #define PD_REGISTER_ARG_MAPPING_FN(op_type, arg_mapping_fn) \
PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \ PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \
pt_register_arg_map_fn_ns_check_##op_type, \ PD_REGISTER_arg_map_fn_ns_check_##op_type, \
"PT_REGISTER_ARG_MAPPING_FN must be called in global namespace."); \ "PD_REGISTER_ARG_MAPPING_FN must be called in global namespace."); \
static const ::phi::ArgumentMappingFnRegistrar \ static const ::phi::ArgumentMappingFnRegistrar \
__registrar_arg_map_fn_for_##op_type(#op_type, arg_mapping_fn); \ __registrar_arg_map_fn_for_##op_type(#op_type, arg_mapping_fn); \
int TouchArgumentMappingFnSymbol_##op_type() { return 0; } int TouchArgumentMappingFnSymbol_##op_type() { return 0; }
#define PT_DECLARE_ARG_MAPPING_FN(op_type) \ #define PD_DECLARE_ARG_MAPPING_FN(op_type) \
PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \ PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \
pt_declare_arg_map_fn_ns_check_##op_type, \ PD_DECLARE_arg_map_fn_ns_check_##op_type, \
"PT_DECLARE_ARG_MAPPING_FN must be called in global namespace."); \ "PD_DECLARE_ARG_MAPPING_FN must be called in global namespace."); \
extern int TouchArgumentMappingFnSymbol_##op_type(); \ extern int TouchArgumentMappingFnSymbol_##op_type(); \
UNUSED static int __declare_arg_map_fn_symbol_for_##op_type = \ UNUSED static int __declare_arg_map_fn_symbol_for_##op_type = \
TouchArgumentMappingFnSymbol_##op_type() TouchArgumentMappingFnSymbol_##op_type()
......
paddle/phi/core/infermeta_utils.h
浏览文件 @ 4a338796
...@@ -282,10 +282,10 @@ struct InferMetaFnRegistrar { ...@@ -282,10 +282,10 @@ struct InferMetaFnRegistrar {
} }
}; };
#define PT_REGISTER_INFER_META_FN(kernel_name_prefix, variadic_infer_meta_fn) \ #define PD_REGISTER_INFER_META_FN(kernel_name_prefix, variadic_infer_meta_fn) \
PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \ PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \
pt_register_infer_meta_fn_ns_check_##kernel_name_prefix, \ PD_REGISTER_infer_meta_fn_ns_check_##kernel_name_prefix, \
"PT_REGISTER_INFER_META_FN must be called in global namespace."); \ "PD_REGISTER_INFER_META_FN must be called in global namespace."); \
static const ::phi::InferMetaFnRegistrar \ static const ::phi::InferMetaFnRegistrar \
__registrar_arg_map_fn_for_##kernel_name_prefix( \ __registrar_arg_map_fn_for_##kernel_name_prefix( \
#kernel_name_prefix, PT_INFER_META(variadic_infer_meta_fn)) #kernel_name_prefix, PT_INFER_META(variadic_infer_meta_fn))
......
paddle/phi/core/kernel_registry.h
浏览文件 @ 4a338796
...@@ -234,7 +234,7 @@ struct KernelRegistrar { ...@@ -234,7 +234,7 @@ struct KernelRegistrar {
#define _PT_ARG_N(args) _PT_ARG_N_EXPAND args #define _PT_ARG_N(args) _PT_ARG_N_EXPAND args
#define _PT_RESQ_N() 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0 #define _PT_RESQ_N() 15, 14, 13, 12, 11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1, 0
/** PT_REGISTER_KERNEL /** PD_REGISTER_KERNEL
* *
* The most frequently used kernel registration macro, used for kernel * The most frequently used kernel registration macro, used for kernel
* registration with only data type as template parameter, and the function * registration with only data type as template parameter, and the function
...@@ -243,8 +243,8 @@ struct KernelRegistrar { ...@@ -243,8 +243,8 @@ struct KernelRegistrar {
* *
* Note: `2TA` means `2 template argument` * Note: `2TA` means `2 template argument`
*/ */
#define PT_REGISTER_KERNEL(kernel_name, backend, layout, meta_kernel_fn, ...) \ #define PD_REGISTER_KERNEL(kernel_name, backend, layout, meta_kernel_fn, ...) \
_PT_REGISTER_KERNEL(::phi::RegType::BUILTIN, \ _PD_REGISTER_KERNEL(::phi::RegType::BUILTIN, \
kernel_name, \ kernel_name, \
backend, \ backend, \
::phi::backend##Context, \ ::phi::backend##Context, \
...@@ -252,12 +252,12 @@ struct KernelRegistrar { ...@@ -252,12 +252,12 @@ struct KernelRegistrar {
meta_kernel_fn, \ meta_kernel_fn, \
__VA_ARGS__) __VA_ARGS__)
#define _PT_REGISTER_KERNEL( \ #define _PD_REGISTER_KERNEL( \
reg_type, kernel_name, backend, context, layout, meta_kernel_fn, ...) \ reg_type, kernel_name, backend, context, layout, meta_kernel_fn, ...) \
PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \ PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \
pt_register_tp_kernel_ns_check_##kernel_name##_##backend##_##layout, \ PD_REGISTER_tp_kernel_ns_check_##kernel_name##_##backend##_##layout, \
"PT_REGISTER_KERNEL must be called in global namespace."); \ "PD_REGISTER_KERNEL must be called in global namespace."); \
PT_EXPAND(_PT_REGISTER_2TA_KERNEL(reg_type, \ PT_EXPAND(_PD_REGISTER_2TA_KERNEL(reg_type, \
kernel_name, \ kernel_name, \
backend, \ backend, \
context, \ context, \
...@@ -266,7 +266,7 @@ struct KernelRegistrar { ...@@ -266,7 +266,7 @@ struct KernelRegistrar {
__VA_ARGS__)) __VA_ARGS__))
#ifndef _WIN32 #ifndef _WIN32
#define _PT_REGISTER_2TA_KERNEL( \ #define _PD_REGISTER_2TA_KERNEL( \
reg_type, kernel_name, backend, context, layout, meta_kernel_fn, ...) \ reg_type, kernel_name, backend, context, layout, meta_kernel_fn, ...) \
PT_KERNEL_INSTANTIATION(meta_kernel_fn, backend, context, __VA_ARGS__); \ PT_KERNEL_INSTANTIATION(meta_kernel_fn, backend, context, __VA_ARGS__); \
static void __PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \ static void __PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \
...@@ -295,7 +295,7 @@ struct KernelRegistrar { ...@@ -295,7 +295,7 @@ struct KernelRegistrar {
* *
* And msvc can work without template instantiation * And msvc can work without template instantiation
*/ */
#define _PT_REGISTER_2TA_KERNEL( \ #define _PD_REGISTER_2TA_KERNEL( \
reg_type, kernel_name, backend, context, layout, meta_kernel_fn, ...) \ reg_type, kernel_name, backend, context, layout, meta_kernel_fn, ...) \
static void __PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \ static void __PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \
const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel); \ const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel); \
...@@ -909,27 +909,27 @@ struct KernelRegistrar { ...@@ -909,27 +909,27 @@ struct KernelRegistrar {
args_def_fn, \ args_def_fn, \
meta_kernel_fn, \ meta_kernel_fn, \
__VA_ARGS__)) __VA_ARGS__))
/** PT_REGISTER_GENERAL_KERNEL /** PD_REGISTER_GENERAL_KERNEL
* *
* Basic Kernel register marco, used to register a instantiated kernel function * Basic Kernel register marco, used to register a instantiated kernel function
* with one template argument. * with one template argument.
*/ */
#define PT_REGISTER_GENERAL_KERNEL( \ #define PD_REGISTER_GENERAL_KERNEL( \
kernel_name, backend, layout, kernel_fn, dtype) \ kernel_name, backend, layout, kernel_fn, dtype) \
_PT_REGISTER_GENERAL_KERNEL( \ _PD_REGISTER_GENERAL_KERNEL( \
::phi::RegType::BUILTIN, kernel_name, backend, layout, kernel_fn, dtype) ::phi::RegType::BUILTIN, kernel_name, backend, layout, kernel_fn, dtype)
#define _PT_REGISTER_GENERAL_KERNEL( \ #define _PD_REGISTER_GENERAL_KERNEL( \
reg_type, kernel_name, backend, layout, kernel_fn, dtype) \ reg_type, kernel_name, backend, layout, kernel_fn, dtype) \
PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \ PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \
pt_register_no_t_kernel_ns_check_##kernel_name##_##backend##_##layout, \ PD_REGISTER_no_t_kernel_ns_check_##kernel_name##_##backend##_##layout, \
"PT_REGISTER_NO_TEMPLATE_KERNEL must be called in global namespace."); \ "PD_REGISTER_NO_TEMPLATE_KERNEL must be called in global namespace."); \
__PT_REGISTER_GENERAL_KERNEL( \ __PD_REGISTER_GENERAL_KERNEL( \
reg_type, kernel_name, backend, layout, kernel_fn, dtype) reg_type, kernel_name, backend, layout, kernel_fn, dtype)
#ifndef _WIN32 #ifndef _WIN32
#define __PT_REGISTER_GENERAL_KERNEL( \ #define __PD_REGISTER_GENERAL_KERNEL( \
reg_type, kernel_name, backend, layout, kernel_fn, dtype) \ reg_type, kernel_name, backend, layout, kernel_fn, dtype) \
template decltype(kernel_fn) kernel_fn; \ template decltype(kernel_fn) kernel_fn; \
static void __PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \ static void __PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \
...@@ -950,7 +950,7 @@ struct KernelRegistrar { ...@@ -950,7 +950,7 @@ struct KernelRegistrar {
void __PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \ void __PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \
const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel) const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel)
#else #else
#define __PT_REGISTER_GENERAL_KERNEL( \ #define __PD_REGISTER_GENERAL_KERNEL( \
reg_type, kernel_name, backend, layout, kernel_fn, dtype) \ reg_type, kernel_name, backend, layout, kernel_fn, dtype) \
static void __PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \ static void __PT_KERNEL_args_def_FN_##kernel_name##_##backend##_##layout( \
const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel); \ const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel); \
...@@ -971,42 +971,43 @@ struct KernelRegistrar { ...@@ -971,42 +971,43 @@ struct KernelRegistrar {
const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel) const ::phi::KernelKey& kernel_key, ::phi::Kernel* kernel)
#endif #endif
/** PT_DECLARE_KERNEL /** PD_DECLARE_KERNEL
* *
* Used to export the symbols of the file where the kernel is located, * Used to export the symbols of the file where the kernel is located,
* to avoid being removed by linker * to avoid being removed by linker
*/ */
#define PT_DECLARE_KERNEL(kernel_name, backend, layout) \ #define PD_DECLARE_KERNEL(kernel_name, backend, layout) \
PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \ PT_STATIC_ASSERT_GLOBAL_NAMESPACE( \
pt_declare_tp_kernel_ns_check_##kernel_name##_##backend##_##layout, \ PD_DECLARE_tp_kernel_ns_check_##kernel_name##_##backend##_##layout, \
"PT_DECLARE_KERNEL must be called in global namespace."); \ "PD_DECLARE_KERNEL must be called in global namespace."); \
extern int TouchKernelSymbolFor_##kernel_name##_##backend##_##layout(); \ extern int TouchKernelSymbolFor_##kernel_name##_##backend##_##layout(); \
UNUSED static int \ UNUSED static int \
__declare_kernel_symbol_for_##kernel_name##_##backend##_##layout = \ __declare_kernel_symbol_for_##kernel_name##_##backend##_##layout = \
TouchKernelSymbolFor_##kernel_name##_##backend##_##layout() TouchKernelSymbolFor_##kernel_name##_##backend##_##layout()
/** PD_REGISTER_KERNEL /** PD_REGISTER_BUILTIN_KERNEL
* *
* Used to register kernels for built-in backends. * Used to register kernels for built-in backends.
* Support CPU GPU XPU. * Support CPU GPU XPU.
*/ */
#define PD_REGISTER_KERNEL(kernel_name, backend, layout, meta_kernel_fn, ...) \ #define PD_REGISTER_BUILTIN_KERNEL( \
_PT_REGISTER_KERNEL(::phi::RegType::PLUGIN, \ kernel_name, backend, layout, meta_kernel_fn, ...) \
kernel_name, \ _PD_REGISTER_KERNEL(::phi::RegType::PLUGIN, \
backend, \ kernel_name, \
::phi::backend##Context, \ backend, \
layout, \ ::phi::backend##Context, \
meta_kernel_fn, \ layout, \
meta_kernel_fn, \
__VA_ARGS__) __VA_ARGS__)
/** PD_REGISTER_CUSTOM_KERNEL /** PD_REGISTER_PLUGIN_KERNEL
* *
* Used to register kernels for plug-in backends. * Used to register kernels for plug-in backends.
* Support user-defined backend such as 'Ascend910'. * Support user-defined backend such as 'Ascend910'.
*/ */
#define PD_REGISTER_CUSTOM_KERNEL( \ #define PD_REGISTER_PLUGIN_KERNEL( \
kernel_name, backend, layout, meta_kernel_fn, ...) \ kernel_name, backend, layout, meta_kernel_fn, ...) \
_PT_REGISTER_KERNEL(::phi::RegType::PLUGIN, \ _PD_REGISTER_KERNEL(::phi::RegType::PLUGIN, \
kernel_name, \ kernel_name, \
backend, \ backend, \
::phi::CustomContext, \ ::phi::CustomContext, \
......
paddle/phi/infermeta/unary.cc
浏览文件 @ 4a338796
...@@ -539,5 +539,5 @@ void TraceInferMeta( ...@@ -539,5 +539,5 @@ void TraceInferMeta(
} // namespace phi } // namespace phi
PT_REGISTER_INFER_META_FN(copy_to, phi::CopyToInferMeta); PD_REGISTER_INFER_META_FN(copy_to, phi::CopyToInferMeta);
PT_REGISTER_INFER_META_FN(split, phi::SplitInferMeta); PD_REGISTER_INFER_META_FN(split, phi::SplitInferMeta);
paddle/phi/kernels/cpu/abs_grad_kernel.cc
浏览文件 @ 4a338796
...@@ -19,7 +19,7 @@ ...@@ -19,7 +19,7 @@
using phi::dtype::complex; using phi::dtype::complex;
PT_REGISTER_KERNEL(abs_grad, PD_REGISTER_KERNEL(abs_grad,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::AbsGradKernel, phi::AbsGradKernel,
...@@ -29,7 +29,7 @@ PT_REGISTER_KERNEL(abs_grad, ...@@ -29,7 +29,7 @@ PT_REGISTER_KERNEL(abs_grad,
int64_t, int64_t,
complex<float>, complex<float>,
complex<double>) {} complex<double>) {}
PT_REGISTER_KERNEL(abs_double_grad, PD_REGISTER_KERNEL(abs_double_grad,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::AbsDoubleGradKernel, phi::AbsDoubleGradKernel,
......
paddle/phi/kernels/cpu/abs_kernel.cc
浏览文件 @ 4a338796
...@@ -36,7 +36,7 @@ void AbsKernel(const Context& ctx, const DenseTensor& x, DenseTensor* out) { ...@@ -36,7 +36,7 @@ void AbsKernel(const Context& ctx, const DenseTensor& x, DenseTensor* out) {
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(abs, PD_REGISTER_KERNEL(abs,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::AbsKernel, phi::AbsKernel,
......
paddle/phi/kernels/cpu/bernoulli_kernel.cc
浏览文件 @ 4a338796
...@@ -51,5 +51,5 @@ void BernoulliKernel(const Context& ctx, ...@@ -51,5 +51,5 @@ void BernoulliKernel(const Context& ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL( PD_REGISTER_KERNEL(
bernoulli, CPU, ALL_LAYOUT, phi::BernoulliKernel, float, double) {} bernoulli, CPU, ALL_LAYOUT, phi::BernoulliKernel, float, double) {}
paddle/phi/kernels/cpu/cast_kernel.cc
浏览文件 @ 4a338796
...@@ -58,7 +58,7 @@ void CastKernel(const Context& dev_ctx, ...@@ -58,7 +58,7 @@ void CastKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(cast, PD_REGISTER_KERNEL(cast,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::CastKernel, phi::CastKernel,
......
paddle/phi/kernels/cpu/complex_kernel.cc
浏览文件 @ 4a338796
...@@ -21,7 +21,7 @@ ...@@ -21,7 +21,7 @@
// See Note [ Why still include the fluid headers? ] // See Note [ Why still include the fluid headers? ]
#include "paddle/phi/common/complex.h" #include "paddle/phi/common/complex.h"
PT_REGISTER_KERNEL(conj, PD_REGISTER_KERNEL(conj,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ConjKernel, phi::ConjKernel,
......
paddle/phi/kernels/cpu/concat_kernel.cc
浏览文件 @ 4a338796
...@@ -110,7 +110,7 @@ void ConcatKernel(const Context& dev_ctx, ...@@ -110,7 +110,7 @@ void ConcatKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(concat, PD_REGISTER_KERNEL(concat,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ConcatKernel, phi::ConcatKernel,
......
paddle/phi/kernels/cpu/copy_kernel.cc
浏览文件 @ 4a338796
...@@ -56,5 +56,5 @@ void Copy(const Context& dev_ctx, ...@@ -56,5 +56,5 @@ void Copy(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_GENERAL_KERNEL( PD_REGISTER_GENERAL_KERNEL(
copy, CPU, ALL_LAYOUT, phi::Copy<phi::CPUContext>, ALL_DTYPE) {} copy, CPU, ALL_LAYOUT, phi::Copy<phi::CPUContext>, ALL_DTYPE) {}
paddle/phi/kernels/cpu/diagonal_grad_kernel.cc
浏览文件 @ 4a338796
...@@ -82,7 +82,7 @@ void DiagonalGradKernel(const Context& dev_ctx, ...@@ -82,7 +82,7 @@ void DiagonalGradKernel(const Context& dev_ctx,
} }
} }
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(diagonal_grad, PD_REGISTER_KERNEL(diagonal_grad,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::DiagonalGradKernel, phi::DiagonalGradKernel,
......
paddle/phi/kernels/cpu/diagonal_kernel.cc
浏览文件 @ 4a338796
...@@ -79,7 +79,7 @@ void DiagonalKernel(const Context& dev_ctx, ...@@ -79,7 +79,7 @@ void DiagonalKernel(const Context& dev_ctx,
} }
} }
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(diagonal, PD_REGISTER_KERNEL(diagonal,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::DiagonalKernel, phi::DiagonalKernel,
......
paddle/phi/kernels/cpu/digamma_grad_kernel.cc
浏览文件 @ 4a338796
...@@ -19,5 +19,5 @@ ...@@ -19,5 +19,5 @@
#include "paddle/phi/core/kernel_registry.h" #include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/impl/digamma_grad_kernel_impl.h" #include "paddle/phi/kernels/impl/digamma_grad_kernel_impl.h"
PT_REGISTER_KERNEL( PD_REGISTER_KERNEL(
digamma_grad, CPU, ALL_LAYOUT, phi::DigammaGradKernel, float, double) {} digamma_grad, CPU, ALL_LAYOUT, phi::DigammaGradKernel, float, double) {}
paddle/phi/kernels/cpu/digamma_kernel.cc
浏览文件 @ 4a338796
...@@ -19,5 +19,5 @@ ...@@ -19,5 +19,5 @@
#include "paddle/phi/core/kernel_registry.h" #include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/impl/digamma_kernel_impl.h" #include "paddle/phi/kernels/impl/digamma_kernel_impl.h"
PT_REGISTER_KERNEL( PD_REGISTER_KERNEL(
digamma, CPU, ALL_LAYOUT, phi::DigammaKernel, float, double) {} digamma, CPU, ALL_LAYOUT, phi::DigammaKernel, float, double) {}
paddle/phi/kernels/cpu/dot_grad_kernel.cc
浏览文件 @ 4a338796
...@@ -20,7 +20,7 @@ ...@@ -20,7 +20,7 @@
#include "paddle/phi/common/complex.h" #include "paddle/phi/common/complex.h"
PT_REGISTER_KERNEL(dot_grad, PD_REGISTER_KERNEL(dot_grad,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::DotGradKernel, phi::DotGradKernel,
......
paddle/phi/kernels/cpu/dot_kernel.cc
浏览文件 @ 4a338796
...@@ -49,7 +49,7 @@ void DotKernel(const Context& dev_ctx, ...@@ -49,7 +49,7 @@ void DotKernel(const Context& dev_ctx,
using complex64 = ::phi::dtype::complex<float>; using complex64 = ::phi::dtype::complex<float>;
using complex128 = ::phi::dtype::complex<double>; using complex128 = ::phi::dtype::complex<double>;
PT_REGISTER_KERNEL(dot, PD_REGISTER_KERNEL(dot,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::DotKernel, phi::DotKernel,
......
paddle/phi/kernels/cpu/elementwise_grad_kernel.cc
浏览文件 @ 4a338796
...@@ -125,7 +125,7 @@ void SubtractDoubleGradKernel(const Context& dev_ctx, ...@@ -125,7 +125,7 @@ void SubtractDoubleGradKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(add_grad, PD_REGISTER_KERNEL(add_grad,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::AddGradKernel, phi::AddGradKernel,
...@@ -137,7 +137,7 @@ PT_REGISTER_KERNEL(add_grad, ...@@ -137,7 +137,7 @@ PT_REGISTER_KERNEL(add_grad,
phi::dtype::complex<float>, phi::dtype::complex<float>,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {}
PT_REGISTER_KERNEL(add_double_grad, PD_REGISTER_KERNEL(add_double_grad,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::AddDoubleGradKernel, phi::AddDoubleGradKernel,
...@@ -149,7 +149,7 @@ PT_REGISTER_KERNEL(add_double_grad, ...@@ -149,7 +149,7 @@ PT_REGISTER_KERNEL(add_double_grad,
phi::dtype::complex<float>, phi::dtype::complex<float>,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {}
PT_REGISTER_KERNEL(add_triple_grad, PD_REGISTER_KERNEL(add_triple_grad,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::AddTripleGradKernel, phi::AddTripleGradKernel,
...@@ -161,7 +161,7 @@ PT_REGISTER_KERNEL(add_triple_grad, ...@@ -161,7 +161,7 @@ PT_REGISTER_KERNEL(add_triple_grad,
phi::dtype::complex<float>, phi::dtype::complex<float>,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {}
PT_REGISTER_KERNEL(subtract_grad, PD_REGISTER_KERNEL(subtract_grad,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::SubtractGradKernel, phi::SubtractGradKernel,
...@@ -173,7 +173,7 @@ PT_REGISTER_KERNEL(subtract_grad, ...@@ -173,7 +173,7 @@ PT_REGISTER_KERNEL(subtract_grad,
phi::dtype::complex<float>, phi::dtype::complex<float>,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {}
PT_REGISTER_KERNEL(subtract_double_grad, PD_REGISTER_KERNEL(subtract_double_grad,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::SubtractDoubleGradKernel, phi::SubtractDoubleGradKernel,
......
paddle/phi/kernels/cpu/expand_grad_kernel.cc
浏览文件 @ 4a338796
...@@ -19,7 +19,7 @@ ...@@ -19,7 +19,7 @@
#include "paddle/phi/core/kernel_registry.h" #include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/impl/expand_grad_kernel_impl.h" #include "paddle/phi/kernels/impl/expand_grad_kernel_impl.h"
PT_REGISTER_KERNEL(expand_grad, PD_REGISTER_KERNEL(expand_grad,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ExpandGradKernel, phi::ExpandGradKernel,
......
paddle/phi/kernels/cpu/expand_kernel.cc
浏览文件 @ 4a338796
...@@ -19,7 +19,7 @@ ...@@ -19,7 +19,7 @@
#include "paddle/phi/core/kernel_registry.h" #include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/impl/expand_kernel_impl.h" #include "paddle/phi/kernels/impl/expand_kernel_impl.h"
PT_REGISTER_KERNEL(expand, PD_REGISTER_KERNEL(expand,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ExpandKernel, phi::ExpandKernel,
......
paddle/phi/kernels/cpu/full_kernel.cc
浏览文件 @ 4a338796
...@@ -73,7 +73,7 @@ void FullLikeKernel(const Context& dev_ctx, ...@@ -73,7 +73,7 @@ void FullLikeKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(full, PD_REGISTER_KERNEL(full,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::FullKernel, phi::FullKernel,
...@@ -89,7 +89,7 @@ PT_REGISTER_KERNEL(full, ...@@ -89,7 +89,7 @@ PT_REGISTER_KERNEL(full,
phi::dtype::complex<float>, phi::dtype::complex<float>,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {}
PT_REGISTER_KERNEL(full_like, PD_REGISTER_KERNEL(full_like,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::FullLikeKernel, phi::FullLikeKernel,
......
paddle/phi/kernels/cpu/histogram_kernel.cc
浏览文件 @ 4a338796
...@@ -77,7 +77,7 @@ void HistogramKernel(const Context& dev_ctx, ...@@ -77,7 +77,7 @@ void HistogramKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(histogram, PD_REGISTER_KERNEL(histogram,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::HistogramKernel, phi::HistogramKernel,
......
paddle/phi/kernels/cpu/huber_loss_grad_kernel.cc
浏览文件 @ 4a338796
...@@ -17,6 +17,6 @@ ...@@ -17,6 +17,6 @@
#include "paddle/phi/core/kernel_registry.h" #include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/impl/huber_loss_grad_kernel_impl.h" #include "paddle/phi/kernels/impl/huber_loss_grad_kernel_impl.h"
PT_REGISTER_KERNEL( PD_REGISTER_KERNEL(
huber_loss_grad, CPU, ALL_LAYOUT, phi::HuberLossGradKernel, float, double) { huber_loss_grad, CPU, ALL_LAYOUT, phi::HuberLossGradKernel, float, double) {
} }
paddle/phi/kernels/cpu/huber_loss_kernel.cc
浏览文件 @ 4a338796
...@@ -17,5 +17,5 @@ ...@@ -17,5 +17,5 @@
#include "paddle/phi/core/kernel_registry.h" #include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/impl/huber_loss_kernel_impl.h" #include "paddle/phi/kernels/impl/huber_loss_kernel_impl.h"
PT_REGISTER_KERNEL( PD_REGISTER_KERNEL(
huber_loss, CPU, ALL_LAYOUT, phi::HuberLossKernel, float, double) {} huber_loss, CPU, ALL_LAYOUT, phi::HuberLossKernel, float, double) {}
paddle/phi/kernels/cpu/lerp_grad_kernel.cc
浏览文件 @ 4a338796
...@@ -17,5 +17,5 @@ ...@@ -17,5 +17,5 @@
#include "paddle/phi/core/kernel_registry.h" #include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/impl/lerp_grad_kernel_impl.h" #include "paddle/phi/kernels/impl/lerp_grad_kernel_impl.h"
PT_REGISTER_KERNEL( PD_REGISTER_KERNEL(
lerp_grad, CPU, ALL_LAYOUT, phi::LerpGradKernel, float, double) {} lerp_grad, CPU, ALL_LAYOUT, phi::LerpGradKernel, float, double) {}
paddle/phi/kernels/cpu/lerp_kernel.cc
浏览文件 @ 4a338796
...@@ -17,4 +17,4 @@ ...@@ -17,4 +17,4 @@
#include "paddle/phi/core/kernel_registry.h" #include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/impl/lerp_kernel_impl.h" #include "paddle/phi/kernels/impl/lerp_kernel_impl.h"
PT_REGISTER_KERNEL(lerp, CPU, ALL_LAYOUT, phi::LerpKernel, float, double) {} PD_REGISTER_KERNEL(lerp, CPU, ALL_LAYOUT, phi::LerpKernel, float, double) {}
paddle/phi/kernels/cpu/masked_select_grad_kernel.cc
浏览文件 @ 4a338796
...@@ -43,7 +43,7 @@ void MaskedSelectGradKernel(const Context& dev_ctx, ...@@ -43,7 +43,7 @@ void MaskedSelectGradKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(masked_select_grad, PD_REGISTER_KERNEL(masked_select_grad,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::MaskedSelectGradKernel, phi::MaskedSelectGradKernel,
......
paddle/phi/kernels/cpu/masked_select_kernel.cc
浏览文件 @ 4a338796
...@@ -61,7 +61,7 @@ void MaskedSelectKernel(const Context& dev_ctx, ...@@ -61,7 +61,7 @@ void MaskedSelectKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(masked_select, PD_REGISTER_KERNEL(masked_select,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::MaskedSelectKernel, phi::MaskedSelectKernel,
......
paddle/phi/kernels/cpu/math_kernel.cc
浏览文件 @ 4a338796
...@@ -118,7 +118,7 @@ using complex128 = ::phi::dtype::complex<double>; ...@@ -118,7 +118,7 @@ using complex128 = ::phi::dtype::complex<double>;
// NOTE(chenweihang): using bfloat16 will cause redefine with xpu bfloat16 // NOTE(chenweihang): using bfloat16 will cause redefine with xpu bfloat16
// using bfloat16 = ::phi::dtype::bfloat16; // using bfloat16 = ::phi::dtype::bfloat16;
PT_REGISTER_KERNEL(add_raw, PD_REGISTER_KERNEL(add_raw,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::AddRawKernel, phi::AddRawKernel,
...@@ -129,7 +129,7 @@ PT_REGISTER_KERNEL(add_raw, ...@@ -129,7 +129,7 @@ PT_REGISTER_KERNEL(add_raw,
int64_t, int64_t,
complex64, complex64,
complex128) {} complex128) {}
PT_REGISTER_KERNEL(subtract_raw, PD_REGISTER_KERNEL(subtract_raw,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::SubtractRawKernel, phi::SubtractRawKernel,
...@@ -140,7 +140,7 @@ PT_REGISTER_KERNEL(subtract_raw, ...@@ -140,7 +140,7 @@ PT_REGISTER_KERNEL(subtract_raw,
int64_t, int64_t,
complex64, complex64,
complex128) {} complex128) {}
PT_REGISTER_KERNEL(divide_raw, PD_REGISTER_KERNEL(divide_raw,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::DivideRawKernel, phi::DivideRawKernel,
...@@ -150,7 +150,7 @@ PT_REGISTER_KERNEL(divide_raw, ...@@ -150,7 +150,7 @@ PT_REGISTER_KERNEL(divide_raw,
int64_t, int64_t,
complex64, complex64,
complex128) {} complex128) {}
PT_REGISTER_KERNEL(multiply_raw, PD_REGISTER_KERNEL(multiply_raw,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::MultiplyRawKernel, phi::MultiplyRawKernel,
...@@ -161,7 +161,7 @@ PT_REGISTER_KERNEL(multiply_raw, ...@@ -161,7 +161,7 @@ PT_REGISTER_KERNEL(multiply_raw,
bool, bool,
complex64, complex64,
complex128) {} complex128) {}
PT_REGISTER_KERNEL(sum_raw, PD_REGISTER_KERNEL(sum_raw,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::SumRawKernel, phi::SumRawKernel,
...@@ -176,5 +176,5 @@ PT_REGISTER_KERNEL(sum_raw, ...@@ -176,5 +176,5 @@ PT_REGISTER_KERNEL(sum_raw,
complex128) { complex128) {
kernel->OutputAt(0).SetDataType(paddle::experimental::DataType::UNDEFINED); kernel->OutputAt(0).SetDataType(paddle::experimental::DataType::UNDEFINED);
} }
PT_REGISTER_KERNEL( PD_REGISTER_KERNEL(
mean_raw, CPU, ALL_LAYOUT, phi::MeanRawKernel, float, double, bool) {} mean_raw, CPU, ALL_LAYOUT, phi::MeanRawKernel, float, double, bool) {}
paddle/phi/kernels/cpu/matmul_grad_kernel.cc
浏览文件 @ 4a338796
...@@ -19,7 +19,7 @@ limitations under the License. */ ...@@ -19,7 +19,7 @@ limitations under the License. */
#include "paddle/phi/kernels/impl/matmul_grad_kernel_impl.h" #include "paddle/phi/kernels/impl/matmul_grad_kernel_impl.h"
PT_REGISTER_KERNEL(matmul_grad, PD_REGISTER_KERNEL(matmul_grad,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::MatmulGradKernel, phi::MatmulGradKernel,
...@@ -28,7 +28,7 @@ PT_REGISTER_KERNEL(matmul_grad, ...@@ -28,7 +28,7 @@ PT_REGISTER_KERNEL(matmul_grad,
phi::dtype::complex<float>, phi::dtype::complex<float>,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {}
PT_REGISTER_KERNEL(matmul_double_grad, PD_REGISTER_KERNEL(matmul_double_grad,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::MatmulDoubleGradKernel, phi::MatmulDoubleGradKernel,
...@@ -37,7 +37,7 @@ PT_REGISTER_KERNEL(matmul_double_grad, ...@@ -37,7 +37,7 @@ PT_REGISTER_KERNEL(matmul_double_grad,
phi::dtype::complex<float>, phi::dtype::complex<float>,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {}
PT_REGISTER_KERNEL(matmul_triple_grad, PD_REGISTER_KERNEL(matmul_triple_grad,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::MatmulTripleGradKernel, phi::MatmulTripleGradKernel,
......
paddle/phi/kernels/cpu/matmul_kernel.cc
浏览文件 @ 4a338796
...@@ -20,7 +20,7 @@ limitations under the License. */ ...@@ -20,7 +20,7 @@ limitations under the License. */
#include "paddle/phi/common/complex.h" #include "paddle/phi/common/complex.h"
#include "paddle/phi/kernels/impl/matmul_kernel_impl.h" #include "paddle/phi/kernels/impl/matmul_kernel_impl.h"
PT_REGISTER_KERNEL(matmul, PD_REGISTER_KERNEL(matmul,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::MatmulKernel, phi::MatmulKernel,
......
paddle/phi/kernels/cpu/norm_grad_kernel.cc
浏览文件 @ 4a338796
...@@ -83,5 +83,5 @@ void NormGradKernel(const Context& ctx, ...@@ -83,5 +83,5 @@ void NormGradKernel(const Context& ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL( PD_REGISTER_KERNEL(
norm_grad, CPU, ALL_LAYOUT, phi::NormGradKernel, float, double) {} norm_grad, CPU, ALL_LAYOUT, phi::NormGradKernel, float, double) {}
paddle/phi/kernels/cpu/norm_kernel.cc
浏览文件 @ 4a338796
...@@ -76,4 +76,4 @@ void NormKernel(const Context& ctx, ...@@ -76,4 +76,4 @@ void NormKernel(const Context& ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(norm, CPU, ALL_LAYOUT, phi::NormKernel, float, double) {} PD_REGISTER_KERNEL(norm, CPU, ALL_LAYOUT, phi::NormKernel, float, double) {}
paddle/phi/kernels/cpu/scale_kernel.cc
浏览文件 @ 4a338796
...@@ -51,7 +51,7 @@ void ScaleKernel(const Context& dev_ctx, ...@@ -51,7 +51,7 @@ void ScaleKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(scale, PD_REGISTER_KERNEL(scale,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ScaleKernel, phi::ScaleKernel,
......
paddle/phi/kernels/cpu/sign_kernel.cc
浏览文件 @ 4a338796
...@@ -21,4 +21,4 @@ limitations under the License. */ ...@@ -21,4 +21,4 @@ limitations under the License. */
// See Note [ Why still include the fluid headers? ] // See Note [ Why still include the fluid headers? ]
#include "paddle/phi/common/bfloat16.h" #include "paddle/phi/common/bfloat16.h"
PT_REGISTER_KERNEL(sign, CPU, ALL_LAYOUT, phi::SignKernel, float, double) {} PD_REGISTER_KERNEL(sign, CPU, ALL_LAYOUT, phi::SignKernel, float, double) {}
paddle/phi/kernels/cpu/split_kernel.cc
浏览文件 @ 4a338796
...@@ -60,7 +60,7 @@ void SplitKernel(const Context& dev_ctx, ...@@ -60,7 +60,7 @@ void SplitKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(split, PD_REGISTER_KERNEL(split,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::SplitKernel, phi::SplitKernel,
......
paddle/phi/kernels/cpu/trace_grad_kernel.cc
浏览文件 @ 4a338796
...@@ -18,7 +18,7 @@ ...@@ -18,7 +18,7 @@
#include "paddle/phi/core/kernel_registry.h" #include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/impl/trace_grad_kernel_impl.h" #include "paddle/phi/kernels/impl/trace_grad_kernel_impl.h"
PT_REGISTER_KERNEL(trace_grad, PD_REGISTER_KERNEL(trace_grad,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::TraceGradKernel, phi::TraceGradKernel,
......
paddle/phi/kernels/cpu/trace_kernel.cc
浏览文件 @ 4a338796
...@@ -45,7 +45,7 @@ void TraceKernel(const Context& dev_ctx, ...@@ -45,7 +45,7 @@ void TraceKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(trace, PD_REGISTER_KERNEL(trace,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::TraceKernel, phi::TraceKernel,
......
paddle/phi/kernels/cpu/trunc_grad_kernel.cc
浏览文件 @ 4a338796
...@@ -30,7 +30,7 @@ void TruncGradKernel(const Context& dev_ctx, ...@@ -30,7 +30,7 @@ void TruncGradKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(trunc_grad, PD_REGISTER_KERNEL(trunc_grad,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::TruncGradKernel, phi::TruncGradKernel,
......
paddle/phi/kernels/cpu/trunc_kernel.cc
浏览文件 @ 4a338796
...@@ -35,5 +35,5 @@ void TruncKernel(const Context& dev_ctx, ...@@ -35,5 +35,5 @@ void TruncKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL( PD_REGISTER_KERNEL(
trunc, CPU, ALL_LAYOUT, phi::TruncKernel, float, double, int, int64_t) {} trunc, CPU, ALL_LAYOUT, phi::TruncKernel, float, double, int, int64_t) {}
paddle/phi/kernels/empty_kernel.cc
浏览文件 @ 4a338796
...@@ -38,7 +38,7 @@ void EmptyLikeKernel(const Context& dev_ctx, ...@@ -38,7 +38,7 @@ void EmptyLikeKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(empty, PD_REGISTER_KERNEL(empty,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::EmptyKernel, phi::EmptyKernel,
...@@ -54,7 +54,7 @@ PT_REGISTER_KERNEL(empty, ...@@ -54,7 +54,7 @@ PT_REGISTER_KERNEL(empty,
phi::dtype::complex<float>, phi::dtype::complex<float>,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {}
PT_REGISTER_KERNEL(empty_like, PD_REGISTER_KERNEL(empty_like,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::EmptyLikeKernel, phi::EmptyLikeKernel,
...@@ -71,7 +71,7 @@ PT_REGISTER_KERNEL(empty_like, ...@@ -71,7 +71,7 @@ PT_REGISTER_KERNEL(empty_like,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {}
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP) #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
PT_REGISTER_KERNEL(empty, PD_REGISTER_KERNEL(empty,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::EmptyKernel, phi::EmptyKernel,
...@@ -86,7 +86,7 @@ PT_REGISTER_KERNEL(empty, ...@@ -86,7 +86,7 @@ PT_REGISTER_KERNEL(empty,
phi::dtype::complex<float>, phi::dtype::complex<float>,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {}
PT_REGISTER_KERNEL(empty_like, PD_REGISTER_KERNEL(empty_like,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::EmptyLikeKernel, phi::EmptyLikeKernel,
......
paddle/phi/kernels/flatten_grad_kernel.cc
浏览文件 @ 4a338796
...@@ -32,7 +32,7 @@ void FlattenGradKernel(const Context& dev_ctx, ...@@ -32,7 +32,7 @@ void FlattenGradKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(flatten_grad, PD_REGISTER_KERNEL(flatten_grad,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::FlattenGradKernel, phi::FlattenGradKernel,
...@@ -44,7 +44,7 @@ PT_REGISTER_KERNEL(flatten_grad, ...@@ -44,7 +44,7 @@ PT_REGISTER_KERNEL(flatten_grad,
int64_t) {} int64_t) {}
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP) #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
PT_REGISTER_KERNEL(flatten_grad, PD_REGISTER_KERNEL(flatten_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::FlattenGradKernel, phi::FlattenGradKernel,
...@@ -59,7 +59,7 @@ PT_REGISTER_KERNEL(flatten_grad, ...@@ -59,7 +59,7 @@ PT_REGISTER_KERNEL(flatten_grad,
#endif #endif
#ifdef PADDLE_WITH_XPU #ifdef PADDLE_WITH_XPU
PT_REGISTER_KERNEL(flatten_grad, PD_REGISTER_KERNEL(flatten_grad,
XPU, XPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::FlattenGradKernel, phi::FlattenGradKernel,
......
paddle/phi/kernels/flatten_kernel.cc
浏览文件 @ 4a338796
...@@ -48,7 +48,7 @@ void FlattenWithXShape(const Context& dev_ctx, ...@@ -48,7 +48,7 @@ void FlattenWithXShape(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(flatten, PD_REGISTER_KERNEL(flatten,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::FlattenKernel, phi::FlattenKernel,
...@@ -60,7 +60,7 @@ PT_REGISTER_KERNEL(flatten, ...@@ -60,7 +60,7 @@ PT_REGISTER_KERNEL(flatten,
int, int,
int64_t) {} int64_t) {}
PT_REGISTER_KERNEL(flatten_with_xshape, PD_REGISTER_KERNEL(flatten_with_xshape,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::FlattenWithXShape, phi::FlattenWithXShape,
...@@ -73,7 +73,7 @@ PT_REGISTER_KERNEL(flatten_with_xshape, ...@@ -73,7 +73,7 @@ PT_REGISTER_KERNEL(flatten_with_xshape,
int64_t) {} int64_t) {}
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP) #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
PT_REGISTER_KERNEL(flatten, PD_REGISTER_KERNEL(flatten,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::FlattenKernel, phi::FlattenKernel,
...@@ -86,7 +86,7 @@ PT_REGISTER_KERNEL(flatten, ...@@ -86,7 +86,7 @@ PT_REGISTER_KERNEL(flatten,
int, int,
int64_t) {} int64_t) {}
PT_REGISTER_KERNEL(flatten_with_xshape, PD_REGISTER_KERNEL(flatten_with_xshape,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::FlattenWithXShape, phi::FlattenWithXShape,
...@@ -101,7 +101,7 @@ PT_REGISTER_KERNEL(flatten_with_xshape, ...@@ -101,7 +101,7 @@ PT_REGISTER_KERNEL(flatten_with_xshape,
#endif #endif
#ifdef PADDLE_WITH_XPU #ifdef PADDLE_WITH_XPU
PT_REGISTER_KERNEL(flatten, PD_REGISTER_KERNEL(flatten,
XPU, XPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::FlattenKernel, phi::FlattenKernel,
...@@ -112,7 +112,7 @@ PT_REGISTER_KERNEL(flatten, ...@@ -112,7 +112,7 @@ PT_REGISTER_KERNEL(flatten,
int, int,
int64_t) {} int64_t) {}
PT_REGISTER_KERNEL(flatten_with_xshape, PD_REGISTER_KERNEL(flatten_with_xshape,
XPU, XPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::FlattenWithXShape, phi::FlattenWithXShape,
......
paddle/phi/kernels/gpu/abs_grad_kernel.cu
浏览文件 @ 4a338796
...@@ -20,7 +20,7 @@ ...@@ -20,7 +20,7 @@
using phi::dtype::complex; using phi::dtype::complex;
PT_REGISTER_KERNEL(abs_grad, PD_REGISTER_KERNEL(abs_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::AbsGradKernel, phi::AbsGradKernel,
...@@ -31,7 +31,7 @@ PT_REGISTER_KERNEL(abs_grad, ...@@ -31,7 +31,7 @@ PT_REGISTER_KERNEL(abs_grad,
phi::dtype::float16, phi::dtype::float16,
complex<float>, complex<float>,
complex<double>) {} complex<double>) {}
PT_REGISTER_KERNEL(abs_double_grad, PD_REGISTER_KERNEL(abs_double_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::AbsDoubleGradKernel, phi::AbsDoubleGradKernel,
......
paddle/phi/kernels/gpu/abs_kernel.cu
浏览文件 @ 4a338796
...@@ -52,7 +52,7 @@ void AbsKernel(const Context& ctx, const DenseTensor& x, DenseTensor* out) { ...@@ -52,7 +52,7 @@ void AbsKernel(const Context& ctx, const DenseTensor& x, DenseTensor* out) {
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(abs, PD_REGISTER_KERNEL(abs,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::AbsKernel, phi::AbsKernel,
......
paddle/phi/kernels/gpu/bernoulli_kernel.cu
浏览文件 @ 4a338796
...@@ -73,5 +73,5 @@ void BernoulliKernel(const Context& ctx, ...@@ -73,5 +73,5 @@ void BernoulliKernel(const Context& ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL( PD_REGISTER_KERNEL(
bernoulli, GPU, ALL_LAYOUT, phi::BernoulliKernel, float, double) {} bernoulli, GPU, ALL_LAYOUT, phi::BernoulliKernel, float, double) {}
paddle/phi/kernels/gpu/cast_kernel.cu
浏览文件 @ 4a338796
...@@ -61,7 +61,7 @@ void CastKernel(const Context& dev_ctx, ...@@ -61,7 +61,7 @@ void CastKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
#define PTEN_REGISTER_CAST_CUDA_BASE_TYPE(op_name, ...) \ #define PTEN_REGISTER_CAST_CUDA_BASE_TYPE(op_name, ...) \
PT_REGISTER_KERNEL(cast, \ PD_REGISTER_KERNEL(cast, \
GPU, \ GPU, \
ALL_LAYOUT, \ ALL_LAYOUT, \
phi::CastKernel, \ phi::CastKernel, \
......
paddle/phi/kernels/gpu/complex_kernel.cu
浏览文件 @ 4a338796
...@@ -21,7 +21,7 @@ ...@@ -21,7 +21,7 @@
// See Note [ Why still include the fluid headers? ] // See Note [ Why still include the fluid headers? ]
#include "paddle/phi/common/complex.h" #include "paddle/phi/common/complex.h"
PT_REGISTER_KERNEL(conj, PD_REGISTER_KERNEL(conj,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ConjKernel, phi::ConjKernel,
......
paddle/phi/kernels/gpu/concat_kernel.cu
浏览文件 @ 4a338796
...@@ -110,7 +110,7 @@ void ConcatKernel(const Context& dev_ctx, ...@@ -110,7 +110,7 @@ void ConcatKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(concat, PD_REGISTER_KERNEL(concat,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ConcatKernel, phi::ConcatKernel,
......
paddle/phi/kernels/gpu/copy_kernel.cu
浏览文件 @ 4a338796
...@@ -207,5 +207,5 @@ void Copy(const Context& dev_ctx, ...@@ -207,5 +207,5 @@ void Copy(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_GENERAL_KERNEL( PD_REGISTER_GENERAL_KERNEL(
copy, GPU, ALL_LAYOUT, phi::Copy<phi::GPUContext>, ALL_DTYPE) {} copy, GPU, ALL_LAYOUT, phi::Copy<phi::GPUContext>, ALL_DTYPE) {}
paddle/phi/kernels/gpu/diagonal_grad_kernel.cu
浏览文件 @ 4a338796
...@@ -158,7 +158,7 @@ void DiagonalGradKernel(const Context& dev_ctx, ...@@ -158,7 +158,7 @@ void DiagonalGradKernel(const Context& dev_ctx,
} }
} }
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(diagonal_grad, PD_REGISTER_KERNEL(diagonal_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::DiagonalGradKernel, phi::DiagonalGradKernel,
......
paddle/phi/kernels/gpu/diagonal_kernel.cu
浏览文件 @ 4a338796
...@@ -154,7 +154,7 @@ void DiagonalKernel(const Context& dev_ctx, ...@@ -154,7 +154,7 @@ void DiagonalKernel(const Context& dev_ctx,
} }
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(diagonal, PD_REGISTER_KERNEL(diagonal,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::DiagonalKernel, phi::DiagonalKernel,
......
paddle/phi/kernels/gpu/digamma_grad_kernel.cu
浏览文件 @ 4a338796
...@@ -18,5 +18,5 @@ ...@@ -18,5 +18,5 @@
#include "paddle/phi/kernels/digamma_grad_kernel.h" #include "paddle/phi/kernels/digamma_grad_kernel.h"
#include "paddle/phi/kernels/impl/digamma_grad_kernel_impl.h" #include "paddle/phi/kernels/impl/digamma_grad_kernel_impl.h"
PT_REGISTER_KERNEL( PD_REGISTER_KERNEL(
digamma_grad, GPU, ALL_LAYOUT, phi::DigammaGradKernel, float, double) {} digamma_grad, GPU, ALL_LAYOUT, phi::DigammaGradKernel, float, double) {}
paddle/phi/kernels/gpu/digamma_kernel.cu
浏览文件 @ 4a338796
...@@ -19,5 +19,5 @@ ...@@ -19,5 +19,5 @@
#include "paddle/phi/kernels/digamma_kernel.h" #include "paddle/phi/kernels/digamma_kernel.h"
#include "paddle/phi/kernels/impl/digamma_kernel_impl.h" #include "paddle/phi/kernels/impl/digamma_kernel_impl.h"
PT_REGISTER_KERNEL( PD_REGISTER_KERNEL(
digamma, GPU, ALL_LAYOUT, phi::DigammaKernel, float, double) {} digamma, GPU, ALL_LAYOUT, phi::DigammaKernel, float, double) {}
paddle/phi/kernels/gpu/dot_grad_kernel.cu
浏览文件 @ 4a338796
...@@ -20,7 +20,7 @@ limitations under the License. */ ...@@ -20,7 +20,7 @@ limitations under the License. */
#include "paddle/phi/common/complex.h" #include "paddle/phi/common/complex.h"
PT_REGISTER_KERNEL(dot_grad, PD_REGISTER_KERNEL(dot_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::DotGradKernel, phi::DotGradKernel,
......
paddle/phi/kernels/gpu/dot_kernel.cu
浏览文件 @ 4a338796
...@@ -52,7 +52,7 @@ void DotKernel(const Context& dev_ctx, ...@@ -52,7 +52,7 @@ void DotKernel(const Context& dev_ctx,
using complex64 = ::phi::dtype::complex<float>; using complex64 = ::phi::dtype::complex<float>;
using complex128 = ::phi::dtype::complex<double>; using complex128 = ::phi::dtype::complex<double>;
PT_REGISTER_KERNEL(dot, PD_REGISTER_KERNEL(dot,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::DotKernel, phi::DotKernel,
......
paddle/phi/kernels/gpu/elementwise_grad_kernel.cu
浏览文件 @ 4a338796
...@@ -119,7 +119,7 @@ void SubtractDoubleGradKernel(const Context& dev_ctx, ...@@ -119,7 +119,7 @@ void SubtractDoubleGradKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(add_grad, PD_REGISTER_KERNEL(add_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::AddGradKernel, phi::AddGradKernel,
...@@ -131,7 +131,7 @@ PT_REGISTER_KERNEL(add_grad, ...@@ -131,7 +131,7 @@ PT_REGISTER_KERNEL(add_grad,
phi::dtype::complex<float>, phi::dtype::complex<float>,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {}
PT_REGISTER_KERNEL(add_double_grad, PD_REGISTER_KERNEL(add_double_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::AddDoubleGradKernel, phi::AddDoubleGradKernel,
...@@ -143,7 +143,7 @@ PT_REGISTER_KERNEL(add_double_grad, ...@@ -143,7 +143,7 @@ PT_REGISTER_KERNEL(add_double_grad,
phi::dtype::complex<float>, phi::dtype::complex<float>,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {}
PT_REGISTER_KERNEL(add_triple_grad, PD_REGISTER_KERNEL(add_triple_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::AddTripleGradKernel, phi::AddTripleGradKernel,
...@@ -155,7 +155,7 @@ PT_REGISTER_KERNEL(add_triple_grad, ...@@ -155,7 +155,7 @@ PT_REGISTER_KERNEL(add_triple_grad,
phi::dtype::complex<float>, phi::dtype::complex<float>,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {}
PT_REGISTER_KERNEL(subtract_grad, PD_REGISTER_KERNEL(subtract_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::SubtractGradKernel, phi::SubtractGradKernel,
...@@ -167,7 +167,7 @@ PT_REGISTER_KERNEL(subtract_grad, ...@@ -167,7 +167,7 @@ PT_REGISTER_KERNEL(subtract_grad,
phi::dtype::complex<float>, phi::dtype::complex<float>,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {}
PT_REGISTER_KERNEL(subtract_double_grad, PD_REGISTER_KERNEL(subtract_double_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::SubtractDoubleGradKernel, phi::SubtractDoubleGradKernel,
......
paddle/phi/kernels/gpu/expand_grad_kernel.cu
浏览文件 @ 4a338796
...@@ -18,7 +18,7 @@ ...@@ -18,7 +18,7 @@
#include "paddle/phi/kernels/expand_grad_kernel.h" #include "paddle/phi/kernels/expand_grad_kernel.h"
#include "paddle/phi/kernels/impl/expand_grad_kernel_impl.h" #include "paddle/phi/kernels/impl/expand_grad_kernel_impl.h"
PT_REGISTER_KERNEL(expand_grad, PD_REGISTER_KERNEL(expand_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ExpandGradKernel, phi::ExpandGradKernel,
......
paddle/phi/kernels/gpu/expand_kernel.cu
浏览文件 @ 4a338796
...@@ -19,7 +19,7 @@ ...@@ -19,7 +19,7 @@
#include "paddle/phi/kernels/expand_kernel.h" #include "paddle/phi/kernels/expand_kernel.h"
#include "paddle/phi/kernels/impl/expand_kernel_impl.h" #include "paddle/phi/kernels/impl/expand_kernel_impl.h"
PT_REGISTER_KERNEL(expand, PD_REGISTER_KERNEL(expand,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ExpandKernel, phi::ExpandKernel,
......
paddle/phi/kernels/gpu/full_kernel.cu
浏览文件 @ 4a338796
...@@ -98,7 +98,7 @@ void FullLikeKernel(const Context& dev_ctx, ...@@ -98,7 +98,7 @@ void FullLikeKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(full, PD_REGISTER_KERNEL(full,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::FullKernel, phi::FullKernel,
...@@ -113,7 +113,7 @@ PT_REGISTER_KERNEL(full, ...@@ -113,7 +113,7 @@ PT_REGISTER_KERNEL(full,
phi::dtype::complex<float>, phi::dtype::complex<float>,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {}
PT_REGISTER_KERNEL(full_like, PD_REGISTER_KERNEL(full_like,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::FullLikeKernel, phi::FullLikeKernel,
......
paddle/phi/kernels/gpu/histogram_kernel.cu
浏览文件 @ 4a338796
...@@ -149,7 +149,7 @@ void HistogramKernel(const Context& dev_ctx, ...@@ -149,7 +149,7 @@ void HistogramKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(histogram, PD_REGISTER_KERNEL(histogram,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::HistogramKernel, phi::HistogramKernel,
......
paddle/phi/kernels/gpu/huber_loss_grad_kernel.cu
浏览文件 @ 4a338796
...@@ -17,6 +17,6 @@ ...@@ -17,6 +17,6 @@
#include "paddle/phi/kernels/huber_loss_grad_kernel.h" #include "paddle/phi/kernels/huber_loss_grad_kernel.h"
#include "paddle/phi/kernels/impl/huber_loss_grad_kernel_impl.h" #include "paddle/phi/kernels/impl/huber_loss_grad_kernel_impl.h"
PT_REGISTER_KERNEL( PD_REGISTER_KERNEL(
huber_loss_grad, GPU, ALL_LAYOUT, phi::HuberLossGradKernel, float, double) { huber_loss_grad, GPU, ALL_LAYOUT, phi::HuberLossGradKernel, float, double) {
} }
paddle/phi/kernels/gpu/huber_loss_kernel.cu
浏览文件 @ 4a338796
...@@ -17,5 +17,5 @@ ...@@ -17,5 +17,5 @@
#include "paddle/phi/kernels/huber_loss_kernel.h" #include "paddle/phi/kernels/huber_loss_kernel.h"
#include "paddle/phi/kernels/impl/huber_loss_kernel_impl.h" #include "paddle/phi/kernels/impl/huber_loss_kernel_impl.h"
PT_REGISTER_KERNEL( PD_REGISTER_KERNEL(
huber_loss, GPU, ALL_LAYOUT, phi::HuberLossKernel, float, double) {} huber_loss, GPU, ALL_LAYOUT, phi::HuberLossKernel, float, double) {}
paddle/phi/kernels/gpu/lerp_grad_kernel.cu
浏览文件 @ 4a338796
...@@ -17,5 +17,5 @@ ...@@ -17,5 +17,5 @@
#include "paddle/phi/kernels/impl/lerp_grad_kernel_impl.h" #include "paddle/phi/kernels/impl/lerp_grad_kernel_impl.h"
#include "paddle/phi/kernels/lerp_grad_kernel.h" #include "paddle/phi/kernels/lerp_grad_kernel.h"
PT_REGISTER_KERNEL( PD_REGISTER_KERNEL(
lerp_grad, GPU, ALL_LAYOUT, phi::LerpGradKernel, float, double) {} lerp_grad, GPU, ALL_LAYOUT, phi::LerpGradKernel, float, double) {}
paddle/phi/kernels/gpu/lerp_kernel.cu
浏览文件 @ 4a338796
...@@ -17,4 +17,4 @@ ...@@ -17,4 +17,4 @@
#include "paddle/phi/kernels/impl/lerp_kernel_impl.h" #include "paddle/phi/kernels/impl/lerp_kernel_impl.h"
#include "paddle/phi/kernels/lerp_kernel.h" #include "paddle/phi/kernels/lerp_kernel.h"
PT_REGISTER_KERNEL(lerp, GPU, ALL_LAYOUT, phi::LerpKernel, float, double) {} PD_REGISTER_KERNEL(lerp, GPU, ALL_LAYOUT, phi::LerpKernel, float, double) {}
paddle/phi/kernels/gpu/masked_select_grad_kernel.cu
浏览文件 @ 4a338796
...@@ -96,7 +96,7 @@ void MaskedSelectGradKernel(const Context& dev_ctx, ...@@ -96,7 +96,7 @@ void MaskedSelectGradKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(masked_select_grad, PD_REGISTER_KERNEL(masked_select_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::MaskedSelectGradKernel, phi::MaskedSelectGradKernel,
......
paddle/phi/kernels/gpu/masked_select_kernel.cu
浏览文件 @ 4a338796
...@@ -108,7 +108,7 @@ void MaskedSelectKernel(const Context& dev_ctx, ...@@ -108,7 +108,7 @@ void MaskedSelectKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(masked_select, PD_REGISTER_KERNEL(masked_select,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::MaskedSelectKernel, phi::MaskedSelectKernel,
......
paddle/phi/kernels/gpu/math_kernel.cu
浏览文件 @ 4a338796
...@@ -95,7 +95,7 @@ using float16 = phi::dtype::float16; ...@@ -95,7 +95,7 @@ using float16 = phi::dtype::float16;
using complex64 = ::phi::dtype::complex<float>; using complex64 = ::phi::dtype::complex<float>;
using complex128 = ::phi::dtype::complex<double>; using complex128 = ::phi::dtype::complex<double>;
PT_REGISTER_KERNEL(add_raw, PD_REGISTER_KERNEL(add_raw,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::AddRawKernel, phi::AddRawKernel,
...@@ -107,7 +107,7 @@ PT_REGISTER_KERNEL(add_raw, ...@@ -107,7 +107,7 @@ PT_REGISTER_KERNEL(add_raw,
float16, float16,
complex64, complex64,
complex128) {} complex128) {}
PT_REGISTER_KERNEL(subtract_raw, PD_REGISTER_KERNEL(subtract_raw,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::SubtractRawKernel, phi::SubtractRawKernel,
...@@ -119,7 +119,7 @@ PT_REGISTER_KERNEL(subtract_raw, ...@@ -119,7 +119,7 @@ PT_REGISTER_KERNEL(subtract_raw,
float16, float16,
complex64, complex64,
complex128) {} complex128) {}
PT_REGISTER_KERNEL(divide_raw, PD_REGISTER_KERNEL(divide_raw,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::DivideRawKernel, phi::DivideRawKernel,
...@@ -130,7 +130,7 @@ PT_REGISTER_KERNEL(divide_raw, ...@@ -130,7 +130,7 @@ PT_REGISTER_KERNEL(divide_raw,
float16, float16,
complex64, complex64,
complex128) {} complex128) {}
PT_REGISTER_KERNEL(multiply_raw, PD_REGISTER_KERNEL(multiply_raw,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::MultiplyRawKernel, phi::MultiplyRawKernel,
...@@ -142,7 +142,7 @@ PT_REGISTER_KERNEL(multiply_raw, ...@@ -142,7 +142,7 @@ PT_REGISTER_KERNEL(multiply_raw,
float16, float16,
complex64, complex64,
complex128) {} complex128) {}
PT_REGISTER_KERNEL(sum_raw, PD_REGISTER_KERNEL(sum_raw,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::SumRawKernel, phi::SumRawKernel,
...@@ -158,7 +158,7 @@ PT_REGISTER_KERNEL(sum_raw, ...@@ -158,7 +158,7 @@ PT_REGISTER_KERNEL(sum_raw,
kernel->OutputAt(0).SetDataType(paddle::experimental::DataType::UNDEFINED); kernel->OutputAt(0).SetDataType(paddle::experimental::DataType::UNDEFINED);
} }
PT_REGISTER_KERNEL(mean_raw, PD_REGISTER_KERNEL(mean_raw,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::MeanRawKernel, phi::MeanRawKernel,
......
paddle/phi/kernels/gpu/matmul_grad_kernel.cu
浏览文件 @ 4a338796
...@@ -19,7 +19,7 @@ limitations under the License. */ ...@@ -19,7 +19,7 @@ limitations under the License. */
#include "paddle/phi/kernels/impl/matmul_grad_kernel_impl.h" #include "paddle/phi/kernels/impl/matmul_grad_kernel_impl.h"
PT_REGISTER_KERNEL(matmul_grad, PD_REGISTER_KERNEL(matmul_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::MatmulGradKernel, phi::MatmulGradKernel,
...@@ -30,7 +30,7 @@ PT_REGISTER_KERNEL(matmul_grad, ...@@ -30,7 +30,7 @@ PT_REGISTER_KERNEL(matmul_grad,
phi::dtype::complex<float>, phi::dtype::complex<float>,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {}
PT_REGISTER_KERNEL(matmul_double_grad, PD_REGISTER_KERNEL(matmul_double_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::MatmulDoubleGradKernel, phi::MatmulDoubleGradKernel,
...@@ -40,7 +40,7 @@ PT_REGISTER_KERNEL(matmul_double_grad, ...@@ -40,7 +40,7 @@ PT_REGISTER_KERNEL(matmul_double_grad,
phi::dtype::complex<float>, phi::dtype::complex<float>,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {}
PT_REGISTER_KERNEL(matmul_triple_grad, PD_REGISTER_KERNEL(matmul_triple_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::MatmulTripleGradKernel, phi::MatmulTripleGradKernel,
......
paddle/phi/kernels/gpu/matmul_kernel.cu
浏览文件 @ 4a338796
...@@ -20,7 +20,7 @@ limitations under the License. */ ...@@ -20,7 +20,7 @@ limitations under the License. */
#include "paddle/phi/common/complex.h" #include "paddle/phi/common/complex.h"
#include "paddle/phi/kernels/impl/matmul_kernel_impl.h" #include "paddle/phi/kernels/impl/matmul_kernel_impl.h"
PT_REGISTER_KERNEL(matmul, PD_REGISTER_KERNEL(matmul,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::MatmulKernel, phi::MatmulKernel,
......
paddle/phi/kernels/gpu/norm_grad_kernel.cu
浏览文件 @ 4a338796
...@@ -111,7 +111,7 @@ void NormGradKernel(const Context& ctx, ...@@ -111,7 +111,7 @@ void NormGradKernel(const Context& ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(norm_grad, PD_REGISTER_KERNEL(norm_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::NormGradKernel, phi::NormGradKernel,
......
paddle/phi/kernels/gpu/norm_kernel.cu
浏览文件 @ 4a338796
...@@ -124,7 +124,7 @@ void NormKernel(const Context& ctx, ...@@ -124,7 +124,7 @@ void NormKernel(const Context& ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(norm, PD_REGISTER_KERNEL(norm,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::NormKernel, phi::NormKernel,
......
paddle/phi/kernels/gpu/scale_kernel.cu
浏览文件 @ 4a338796
...@@ -63,7 +63,7 @@ void ScaleKernel(const Context& dev_ctx, ...@@ -63,7 +63,7 @@ void ScaleKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(scale, PD_REGISTER_KERNEL(scale,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ScaleKernel, phi::ScaleKernel,
......
paddle/phi/kernels/gpu/sign_kernel.cu.cc
浏览文件 @ 4a338796
...@@ -23,5 +23,5 @@ limitations under the License. */ ...@@ -23,5 +23,5 @@ limitations under the License. */
using float16 = phi::dtype::float16; using float16 = phi::dtype::float16;
PT_REGISTER_KERNEL( PD_REGISTER_KERNEL(
sign, GPU, ALL_LAYOUT, phi::SignKernel, float, double, float16) {} sign, GPU, ALL_LAYOUT, phi::SignKernel, float, double, float16) {}
paddle/phi/kernels/gpu/split_kernel.cu
浏览文件 @ 4a338796
...@@ -59,7 +59,7 @@ void SplitKernel(const Context& dev_ctx, ...@@ -59,7 +59,7 @@ void SplitKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(split, PD_REGISTER_KERNEL(split,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::SplitKernel, phi::SplitKernel,
......
paddle/phi/kernels/gpu/trace_grad_kernel.cu
浏览文件 @ 4a338796
...@@ -18,7 +18,7 @@ ...@@ -18,7 +18,7 @@
#include "paddle/phi/core/kernel_registry.h" #include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/impl/trace_grad_kernel_impl.h" #include "paddle/phi/kernels/impl/trace_grad_kernel_impl.h"
PT_REGISTER_KERNEL(trace_grad, PD_REGISTER_KERNEL(trace_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::TraceGradKernel, phi::TraceGradKernel,
......
paddle/phi/kernels/gpu/trace_kernel.cu
浏览文件 @ 4a338796
...@@ -44,7 +44,7 @@ void TraceKernel(const Context& ctx, ...@@ -44,7 +44,7 @@ void TraceKernel(const Context& ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(trace, PD_REGISTER_KERNEL(trace,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::TraceKernel, phi::TraceKernel,
......
paddle/phi/kernels/gpu/trunc_grad_kernel.cu
浏览文件 @ 4a338796
...@@ -44,7 +44,7 @@ void TruncGradKernel(const Context& dev_ctx, ...@@ -44,7 +44,7 @@ void TruncGradKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(trunc_grad, PD_REGISTER_KERNEL(trunc_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::TruncGradKernel, phi::TruncGradKernel,
......
paddle/phi/kernels/gpu/trunc_kernel.cu
浏览文件 @ 4a338796
...@@ -77,5 +77,5 @@ void TruncKernel(const Context& dev_ctx, ...@@ -77,5 +77,5 @@ void TruncKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL( PD_REGISTER_KERNEL(
trunc, GPU, ALL_LAYOUT, phi::TruncKernel, float, double, int, int64_t) {} trunc, GPU, ALL_LAYOUT, phi::TruncKernel, float, double, int, int64_t) {}
paddle/phi/kernels/math_kernel.cc
浏览文件 @ 4a338796
...@@ -81,10 +81,10 @@ void MultiplyKernel(const Context& dev_ctx, ...@@ -81,10 +81,10 @@ void MultiplyKernel(const Context& dev_ctx,
using complex64 = ::phi::dtype::complex<float>; using complex64 = ::phi::dtype::complex<float>;
using complex128 = ::phi::dtype::complex<double>; using complex128 = ::phi::dtype::complex<double>;
PT_REGISTER_KERNEL( PD_REGISTER_KERNEL(
mean, CPU, ALL_LAYOUT, phi::MeanKernel, float, double, bool) {} mean, CPU, ALL_LAYOUT, phi::MeanKernel, float, double, bool) {}
PT_REGISTER_KERNEL(sum, PD_REGISTER_KERNEL(sum,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::SumKernel, phi::SumKernel,
...@@ -100,7 +100,7 @@ PT_REGISTER_KERNEL(sum, ...@@ -100,7 +100,7 @@ PT_REGISTER_KERNEL(sum,
kernel->OutputAt(0).SetDataType(paddle::experimental::DataType::UNDEFINED); kernel->OutputAt(0).SetDataType(paddle::experimental::DataType::UNDEFINED);
} }
PT_REGISTER_KERNEL(add, PD_REGISTER_KERNEL(add,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::AddKernel, phi::AddKernel,
...@@ -111,7 +111,7 @@ PT_REGISTER_KERNEL(add, ...@@ -111,7 +111,7 @@ PT_REGISTER_KERNEL(add,
int64_t, int64_t,
complex64, complex64,
complex128) {} complex128) {}
PT_REGISTER_KERNEL(subtract, PD_REGISTER_KERNEL(subtract,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::SubtractKernel, phi::SubtractKernel,
...@@ -122,7 +122,7 @@ PT_REGISTER_KERNEL(subtract, ...@@ -122,7 +122,7 @@ PT_REGISTER_KERNEL(subtract,
int64_t, int64_t,
complex64, complex64,
complex128) {} complex128) {}
PT_REGISTER_KERNEL(divide, PD_REGISTER_KERNEL(divide,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::DivideKernel, phi::DivideKernel,
...@@ -132,7 +132,7 @@ PT_REGISTER_KERNEL(divide, ...@@ -132,7 +132,7 @@ PT_REGISTER_KERNEL(divide,
int64_t, int64_t,
complex64, complex64,
complex128) {} complex128) {}
PT_REGISTER_KERNEL(multiply, PD_REGISTER_KERNEL(multiply,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::MultiplyKernel, phi::MultiplyKernel,
...@@ -145,7 +145,7 @@ PT_REGISTER_KERNEL(multiply, ...@@ -145,7 +145,7 @@ PT_REGISTER_KERNEL(multiply,
complex128) {} complex128) {}
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP) #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
PT_REGISTER_KERNEL(mean, PD_REGISTER_KERNEL(mean,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::MeanKernel, phi::MeanKernel,
...@@ -155,7 +155,7 @@ PT_REGISTER_KERNEL(mean, ...@@ -155,7 +155,7 @@ PT_REGISTER_KERNEL(mean,
int, int,
int64_t, int64_t,
phi::dtype::float16) {} phi::dtype::float16) {}
PT_REGISTER_KERNEL(sum, PD_REGISTER_KERNEL(sum,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::SumKernel, phi::SumKernel,
...@@ -170,7 +170,7 @@ PT_REGISTER_KERNEL(sum, ...@@ -170,7 +170,7 @@ PT_REGISTER_KERNEL(sum,
complex128) { complex128) {
kernel->OutputAt(0).SetDataType(paddle::experimental::DataType::UNDEFINED); kernel->OutputAt(0).SetDataType(paddle::experimental::DataType::UNDEFINED);
} }
PT_REGISTER_KERNEL(add, PD_REGISTER_KERNEL(add,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::AddKernel, phi::AddKernel,
...@@ -182,7 +182,7 @@ PT_REGISTER_KERNEL(add, ...@@ -182,7 +182,7 @@ PT_REGISTER_KERNEL(add,
phi::dtype::float16, phi::dtype::float16,
complex64, complex64,
complex128) {} complex128) {}
PT_REGISTER_KERNEL(subtract, PD_REGISTER_KERNEL(subtract,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::SubtractKernel, phi::SubtractKernel,
...@@ -194,7 +194,7 @@ PT_REGISTER_KERNEL(subtract, ...@@ -194,7 +194,7 @@ PT_REGISTER_KERNEL(subtract,
phi::dtype::float16, phi::dtype::float16,
complex64, complex64,
complex128) {} complex128) {}
PT_REGISTER_KERNEL(divide, PD_REGISTER_KERNEL(divide,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::DivideKernel, phi::DivideKernel,
...@@ -205,7 +205,7 @@ PT_REGISTER_KERNEL(divide, ...@@ -205,7 +205,7 @@ PT_REGISTER_KERNEL(divide,
phi::dtype::float16, phi::dtype::float16,
complex64, complex64,
complex128) {} complex128) {}
PT_REGISTER_KERNEL(multiply, PD_REGISTER_KERNEL(multiply,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::MultiplyKernel, phi::MultiplyKernel,
......
paddle/phi/kernels/reshape_grad_kernel.cc
浏览文件 @ 4a338796
...@@ -37,24 +37,24 @@ void ReshapeDoubleGradKernel(const Context& dev_ctx, ...@@ -37,24 +37,24 @@ void ReshapeDoubleGradKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_GENERAL_KERNEL(reshape_grad, PD_REGISTER_GENERAL_KERNEL(reshape_grad,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ReshapeGradKernel<phi::CPUContext>, phi::ReshapeGradKernel<phi::CPUContext>,
ALL_DTYPE) {} ALL_DTYPE) {}
PT_REGISTER_GENERAL_KERNEL(reshape_double_grad, PD_REGISTER_GENERAL_KERNEL(reshape_double_grad,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ReshapeDoubleGradKernel<phi::CPUContext>, phi::ReshapeDoubleGradKernel<phi::CPUContext>,
ALL_DTYPE) {} ALL_DTYPE) {}
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP) #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
PT_REGISTER_GENERAL_KERNEL(reshape_grad, PD_REGISTER_GENERAL_KERNEL(reshape_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ReshapeGradKernel<phi::GPUContext>, phi::ReshapeGradKernel<phi::GPUContext>,
ALL_DTYPE) {} ALL_DTYPE) {}
PT_REGISTER_GENERAL_KERNEL(reshape_double_grad, PD_REGISTER_GENERAL_KERNEL(reshape_double_grad,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ReshapeDoubleGradKernel<phi::GPUContext>, phi::ReshapeDoubleGradKernel<phi::GPUContext>,
...@@ -62,12 +62,12 @@ PT_REGISTER_GENERAL_KERNEL(reshape_double_grad, ...@@ -62,12 +62,12 @@ PT_REGISTER_GENERAL_KERNEL(reshape_double_grad,
#endif #endif
#ifdef PADDLE_WITH_XPU #ifdef PADDLE_WITH_XPU
PT_REGISTER_GENERAL_KERNEL(reshape_grad, PD_REGISTER_GENERAL_KERNEL(reshape_grad,
XPU, XPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ReshapeGradKernel<phi::XPUContext>, phi::ReshapeGradKernel<phi::XPUContext>,
ALL_DTYPE) {} ALL_DTYPE) {}
PT_REGISTER_GENERAL_KERNEL(reshape_double_grad, PD_REGISTER_GENERAL_KERNEL(reshape_double_grad,
XPU, XPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ReshapeDoubleGradKernel<phi::XPUContext>, phi::ReshapeDoubleGradKernel<phi::XPUContext>,
......
paddle/phi/kernels/reshape_kernel.cc
浏览文件 @ 4a338796
...@@ -52,18 +52,18 @@ void ReshapeWithXShape(const Context& dev_ctx, ...@@ -52,18 +52,18 @@ void ReshapeWithXShape(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_GENERAL_KERNEL( PD_REGISTER_GENERAL_KERNEL(
reshape, CPU, ALL_LAYOUT, phi::ReshapeKernel<phi::CPUContext>, ALL_DTYPE) {} reshape, CPU, ALL_LAYOUT, phi::ReshapeKernel<phi::CPUContext>, ALL_DTYPE) {}
PT_REGISTER_GENERAL_KERNEL(reshape_with_xshape, PD_REGISTER_GENERAL_KERNEL(reshape_with_xshape,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ReshapeWithXShape<phi::CPUContext>, phi::ReshapeWithXShape<phi::CPUContext>,
ALL_DTYPE) {} ALL_DTYPE) {}
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP) #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
PT_REGISTER_GENERAL_KERNEL( PD_REGISTER_GENERAL_KERNEL(
reshape, GPU, ALL_LAYOUT, phi::ReshapeKernel<phi::GPUContext>, ALL_DTYPE) {} reshape, GPU, ALL_LAYOUT, phi::ReshapeKernel<phi::GPUContext>, ALL_DTYPE) {}
PT_REGISTER_GENERAL_KERNEL(reshape_with_xshape, PD_REGISTER_GENERAL_KERNEL(reshape_with_xshape,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ReshapeWithXShape<phi::GPUContext>, phi::ReshapeWithXShape<phi::GPUContext>,
...@@ -71,9 +71,9 @@ PT_REGISTER_GENERAL_KERNEL(reshape_with_xshape, ...@@ -71,9 +71,9 @@ PT_REGISTER_GENERAL_KERNEL(reshape_with_xshape,
#endif #endif
#ifdef PADDLE_WITH_XPU #ifdef PADDLE_WITH_XPU
PT_REGISTER_GENERAL_KERNEL( PD_REGISTER_GENERAL_KERNEL(
reshape, XPU, ALL_LAYOUT, phi::ReshapeKernel<phi::XPUContext>, ALL_DTYPE) {} reshape, XPU, ALL_LAYOUT, phi::ReshapeKernel<phi::XPUContext>, ALL_DTYPE) {}
PT_REGISTER_GENERAL_KERNEL(reshape_with_xshape, PD_REGISTER_GENERAL_KERNEL(reshape_with_xshape,
XPU, XPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ReshapeWithXShape<phi::XPUContext>, phi::ReshapeWithXShape<phi::XPUContext>,
......
paddle/phi/kernels/selected_rows/full_kernel.cc
浏览文件 @ 4a338796
...@@ -36,7 +36,7 @@ void FullSR(const Context& dev_ctx, ...@@ -36,7 +36,7 @@ void FullSR(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(full_sr, PD_REGISTER_KERNEL(full_sr,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::FullSR, phi::FullSR,
...@@ -53,7 +53,7 @@ PT_REGISTER_KERNEL(full_sr, ...@@ -53,7 +53,7 @@ PT_REGISTER_KERNEL(full_sr,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {}
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP) #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
PT_REGISTER_KERNEL(full_sr, PD_REGISTER_KERNEL(full_sr,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::FullSR, phi::FullSR,
......
paddle/phi/kernels/selected_rows/scale_kernel.cc
浏览文件 @ 4a338796
...@@ -38,7 +38,7 @@ void ScaleSR(const Context& dev_ctx, ...@@ -38,7 +38,7 @@ void ScaleSR(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(scale_sr, PD_REGISTER_KERNEL(scale_sr,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ScaleSR, phi::ScaleSR,
...@@ -52,7 +52,7 @@ PT_REGISTER_KERNEL(scale_sr, ...@@ -52,7 +52,7 @@ PT_REGISTER_KERNEL(scale_sr,
int64_t) {} int64_t) {}
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP) #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
PT_REGISTER_KERNEL(scale_sr, PD_REGISTER_KERNEL(scale_sr,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ScaleSR, phi::ScaleSR,
......
paddle/phi/kernels/sparse/cpu/sparse_utils_kernel.cc
浏览文件 @ 4a338796
...@@ -284,7 +284,7 @@ void SparseCooToDenseKernel(const Context& dev_ctx, ...@@ -284,7 +284,7 @@ void SparseCooToDenseKernel(const Context& dev_ctx,
} // namespace sparse } // namespace sparse
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(dense_to_sparse_coo, PD_REGISTER_KERNEL(dense_to_sparse_coo,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::sparse::DenseToSparseCooKernel, phi::sparse::DenseToSparseCooKernel,
...@@ -297,7 +297,7 @@ PT_REGISTER_KERNEL(dense_to_sparse_coo, ...@@ -297,7 +297,7 @@ PT_REGISTER_KERNEL(dense_to_sparse_coo,
int, int,
int64_t) {} int64_t) {}
PT_REGISTER_KERNEL(sparse_csr_to_coo, PD_REGISTER_KERNEL(sparse_csr_to_coo,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::sparse::SparseCsrToCooKernel, phi::sparse::SparseCsrToCooKernel,
...@@ -310,7 +310,7 @@ PT_REGISTER_KERNEL(sparse_csr_to_coo, ...@@ -310,7 +310,7 @@ PT_REGISTER_KERNEL(sparse_csr_to_coo,
int, int,
int64_t) {} int64_t) {}
PT_REGISTER_KERNEL(sparse_coo_to_csr, PD_REGISTER_KERNEL(sparse_coo_to_csr,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::sparse::SparseCooToCsrKernel, phi::sparse::SparseCooToCsrKernel,
...@@ -323,7 +323,7 @@ PT_REGISTER_KERNEL(sparse_coo_to_csr, ...@@ -323,7 +323,7 @@ PT_REGISTER_KERNEL(sparse_coo_to_csr,
int, int,
int64_t) {} int64_t) {}
PT_REGISTER_KERNEL(dense_to_sparse_csr, PD_REGISTER_KERNEL(dense_to_sparse_csr,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::sparse::DenseToSparseCsrKernel, phi::sparse::DenseToSparseCsrKernel,
...@@ -336,7 +336,7 @@ PT_REGISTER_KERNEL(dense_to_sparse_csr, ...@@ -336,7 +336,7 @@ PT_REGISTER_KERNEL(dense_to_sparse_csr,
int, int,
int64_t) {} int64_t) {}
PT_REGISTER_KERNEL(sparse_coo_to_dense, PD_REGISTER_KERNEL(sparse_coo_to_dense,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::sparse::SparseCooToDenseKernel, phi::sparse::SparseCooToDenseKernel,
...@@ -349,7 +349,7 @@ PT_REGISTER_KERNEL(sparse_coo_to_dense, ...@@ -349,7 +349,7 @@ PT_REGISTER_KERNEL(sparse_coo_to_dense,
int, int,
int64_t) {} int64_t) {}
PT_REGISTER_KERNEL(sparse_csr_to_dense, PD_REGISTER_KERNEL(sparse_csr_to_dense,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::sparse::SparseCsrToDenseKernel, phi::sparse::SparseCsrToDenseKernel,
......
paddle/phi/kernels/sparse/gpu/sparse_utils_kernel.cu
浏览文件 @ 4a338796
...@@ -553,7 +553,7 @@ void SparseCooToDenseKernel(const Context& dev_ctx, ...@@ -553,7 +553,7 @@ void SparseCooToDenseKernel(const Context& dev_ctx,
} // namespace sparse } // namespace sparse
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(dense_to_sparse_coo, PD_REGISTER_KERNEL(dense_to_sparse_coo,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::sparse::DenseToSparseCooKernel, phi::sparse::DenseToSparseCooKernel,
...@@ -566,7 +566,7 @@ PT_REGISTER_KERNEL(dense_to_sparse_coo, ...@@ -566,7 +566,7 @@ PT_REGISTER_KERNEL(dense_to_sparse_coo,
int, int,
int64_t) {} int64_t) {}
PT_REGISTER_KERNEL(sparse_csr_to_coo, PD_REGISTER_KERNEL(sparse_csr_to_coo,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::sparse::SparseCsrToCooKernel, phi::sparse::SparseCsrToCooKernel,
...@@ -579,7 +579,7 @@ PT_REGISTER_KERNEL(sparse_csr_to_coo, ...@@ -579,7 +579,7 @@ PT_REGISTER_KERNEL(sparse_csr_to_coo,
int, int,
int64_t) {} int64_t) {}
PT_REGISTER_KERNEL(sparse_coo_to_csr, PD_REGISTER_KERNEL(sparse_coo_to_csr,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::sparse::SparseCooToCsrKernel, phi::sparse::SparseCooToCsrKernel,
...@@ -592,7 +592,7 @@ PT_REGISTER_KERNEL(sparse_coo_to_csr, ...@@ -592,7 +592,7 @@ PT_REGISTER_KERNEL(sparse_coo_to_csr,
int, int,
int64_t) {} int64_t) {}
PT_REGISTER_KERNEL(dense_to_sparse_csr, PD_REGISTER_KERNEL(dense_to_sparse_csr,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::sparse::DenseToSparseCsrKernel, phi::sparse::DenseToSparseCsrKernel,
...@@ -605,7 +605,7 @@ PT_REGISTER_KERNEL(dense_to_sparse_csr, ...@@ -605,7 +605,7 @@ PT_REGISTER_KERNEL(dense_to_sparse_csr,
int, int,
int64_t) {} int64_t) {}
PT_REGISTER_KERNEL(sparse_coo_to_dense, PD_REGISTER_KERNEL(sparse_coo_to_dense,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::sparse::SparseCooToDenseKernel, phi::sparse::SparseCooToDenseKernel,
...@@ -618,7 +618,7 @@ PT_REGISTER_KERNEL(sparse_coo_to_dense, ...@@ -618,7 +618,7 @@ PT_REGISTER_KERNEL(sparse_coo_to_dense,
int, int,
int64_t) {} int64_t) {}
PT_REGISTER_KERNEL(sparse_csr_to_dense, PD_REGISTER_KERNEL(sparse_csr_to_dense,
GPU, GPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::sparse::SparseCsrToDenseKernel, phi::sparse::SparseCsrToDenseKernel,
......
paddle/phi/kernels/transfer_layout_kernel.cc
浏览文件 @ 4a338796
...@@ -69,7 +69,7 @@ void TransferLayoutKernel(const Context& dev_ctx, ...@@ -69,7 +69,7 @@ void TransferLayoutKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_GENERAL_KERNEL(pten_transfer_layout, PD_REGISTER_GENERAL_KERNEL(pten_transfer_layout,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::TransferLayoutKernel<phi::CPUContext>, phi::TransferLayoutKernel<phi::CPUContext>,
......
paddle/phi/kernels/xpu/cast_kernel.cc
浏览文件 @ 4a338796
...@@ -86,7 +86,7 @@ void CastKernel(const Context& dev_ctx, ...@@ -86,7 +86,7 @@ void CastKernel(const Context& dev_ctx,
} }
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(cast, PD_REGISTER_KERNEL(cast,
XPU, XPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::CastKernel, phi::CastKernel,
......
paddle/phi/kernels/xpu/copy_kernel.cc
浏览文件 @ 4a338796
...@@ -69,5 +69,5 @@ void Copy(const Context& dev_ctx, ...@@ -69,5 +69,5 @@ void Copy(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_GENERAL_KERNEL( PD_REGISTER_GENERAL_KERNEL(
copy, XPU, ALL_LAYOUT, phi::Copy<phi::XPUContext>, ALL_DTYPE) {} copy, XPU, ALL_LAYOUT, phi::Copy<phi::XPUContext>, ALL_DTYPE) {}
paddle/phi/kernels/xpu/full_kernel.cc
浏览文件 @ 4a338796
...@@ -116,7 +116,7 @@ void FullLikeKernel(const Context& dev_ctx, ...@@ -116,7 +116,7 @@ void FullLikeKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(full, PD_REGISTER_KERNEL(full,
XPU, XPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::FullKernel, phi::FullKernel,
...@@ -132,7 +132,7 @@ PT_REGISTER_KERNEL(full, ...@@ -132,7 +132,7 @@ PT_REGISTER_KERNEL(full,
phi::dtype::complex<float>, phi::dtype::complex<float>,
phi::dtype::complex<double>) {} phi::dtype::complex<double>) {}
PT_REGISTER_KERNEL(full_like, PD_REGISTER_KERNEL(full_like,
XPU, XPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::FullLikeKernel, phi::FullLikeKernel,
......
paddle/phi/kernels/xpu/scale_kernel.cc
浏览文件 @ 4a338796
...@@ -56,7 +56,7 @@ void ScaleKernel(const Context& dev_ctx, ...@@ -56,7 +56,7 @@ void ScaleKernel(const Context& dev_ctx,
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(scale, PD_REGISTER_KERNEL(scale,
XPU, XPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::ScaleKernel, phi::ScaleKernel,
......
paddle/phi/ops/compat/abs_sig.cc
浏览文件 @ 4a338796
...@@ -32,7 +32,7 @@ KernelSignature AbsDoubleGradOpArgumentMapping( ...@@ -32,7 +32,7 @@ KernelSignature AbsDoubleGradOpArgumentMapping(
} // namespace phi } // namespace phi
PT_REGISTER_ARG_MAPPING_FN(abs, phi::AbsOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(abs, phi::AbsOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(abs_grad, phi::AbsGradOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(abs_grad, phi::AbsGradOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(abs_double_grad, PD_REGISTER_ARG_MAPPING_FN(abs_double_grad,
phi::AbsDoubleGradOpArgumentMapping); phi::AbsDoubleGradOpArgumentMapping);
paddle/phi/ops/compat/cast_sig.cc
浏览文件 @ 4a338796
...@@ -22,4 +22,4 @@ KernelSignature CastOpArgumentMapping(const ArgumentMappingContext& ctx) { ...@@ -22,4 +22,4 @@ KernelSignature CastOpArgumentMapping(const ArgumentMappingContext& ctx) {
} // namespace phi } // namespace phi
PT_REGISTER_ARG_MAPPING_FN(cast, phi::CastOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(cast, phi::CastOpArgumentMapping);
paddle/phi/ops/compat/concat_sig.cc
浏览文件 @ 4a338796
...@@ -25,4 +25,4 @@ KernelSignature ConcatOpArgumentMapping(const ArgumentMappingContext& ctx) { ...@@ -25,4 +25,4 @@ KernelSignature ConcatOpArgumentMapping(const ArgumentMappingContext& ctx) {
} // namespace phi } // namespace phi
PT_REGISTER_ARG_MAPPING_FN(concat, phi::ConcatOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(concat, phi::ConcatOpArgumentMapping);
paddle/phi/ops/compat/diagonal_sig.cc
浏览文件 @ 4a338796
...@@ -25,4 +25,4 @@ KernelSignature DiagonalGradOpArgumentMapping( ...@@ -25,4 +25,4 @@ KernelSignature DiagonalGradOpArgumentMapping(
} }
} // namespace phi } // namespace phi
PT_REGISTER_ARG_MAPPING_FN(diagonal_grad, phi::DiagonalGradOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(diagonal_grad, phi::DiagonalGradOpArgumentMapping);
paddle/phi/ops/compat/digamma_sig.cc
浏览文件 @ 4a338796
...@@ -24,4 +24,4 @@ KernelSignature DigammaGradOpArgumentMapping( ...@@ -24,4 +24,4 @@ KernelSignature DigammaGradOpArgumentMapping(
} // namespace phi } // namespace phi
PT_REGISTER_ARG_MAPPING_FN(digamma_grad, phi::DigammaGradOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(digamma_grad, phi::DigammaGradOpArgumentMapping);
paddle/phi/ops/compat/dot_sig.cc
浏览文件 @ 4a338796
...@@ -25,4 +25,4 @@ KernelSignature DotGradOpArgumentMapping(const ArgumentMappingContext& ctx) { ...@@ -25,4 +25,4 @@ KernelSignature DotGradOpArgumentMapping(const ArgumentMappingContext& ctx) {
} // namespace phi } // namespace phi
PT_REGISTER_ARG_MAPPING_FN(dot_grad, phi::DotGradOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(dot_grad, phi::DotGradOpArgumentMapping);
paddle/phi/ops/compat/elementwise_sig.cc
浏览文件 @ 4a338796
...@@ -102,28 +102,28 @@ KernelSignature ElementwiseSubGradOpArgumentMapping( ...@@ -102,28 +102,28 @@ KernelSignature ElementwiseSubGradOpArgumentMapping(
} // namespace phi } // namespace phi
PT_REGISTER_BASE_KERNEL_NAME(elementwise_add, add); PD_REGISTER_BASE_KERNEL_NAME(elementwise_add, add);
PT_REGISTER_BASE_KERNEL_NAME(elementwise_sub, subtract); PD_REGISTER_BASE_KERNEL_NAME(elementwise_sub, subtract);
PT_REGISTER_BASE_KERNEL_NAME(elementwise_mul, multiply); PD_REGISTER_BASE_KERNEL_NAME(elementwise_mul, multiply);
PT_REGISTER_BASE_KERNEL_NAME(elementwise_div, divide); PD_REGISTER_BASE_KERNEL_NAME(elementwise_div, divide);
PT_REGISTER_BASE_KERNEL_NAME(elementwise_add_grad, add_grad); PD_REGISTER_BASE_KERNEL_NAME(elementwise_add_grad, add_grad);
PT_REGISTER_BASE_KERNEL_NAME(elementwise_add_grad_grad, add_double_grad); PD_REGISTER_BASE_KERNEL_NAME(elementwise_add_grad_grad, add_double_grad);
PT_REGISTER_BASE_KERNEL_NAME(elementwise_add_triple_grad, add_triple_grad); PD_REGISTER_BASE_KERNEL_NAME(elementwise_add_triple_grad, add_triple_grad);
PT_REGISTER_BASE_KERNEL_NAME(elementwise_sub_grad, subtract_grad); PD_REGISTER_BASE_KERNEL_NAME(elementwise_sub_grad, subtract_grad);
PT_REGISTER_ARG_MAPPING_FN(elementwise_add, PD_REGISTER_ARG_MAPPING_FN(elementwise_add,
phi::ElementwiseAddOpArgumentMapping); phi::ElementwiseAddOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(elementwise_sub, PD_REGISTER_ARG_MAPPING_FN(elementwise_sub,
phi::ElementwiseSubOpArgumentMapping); phi::ElementwiseSubOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(elementwise_mul, PD_REGISTER_ARG_MAPPING_FN(elementwise_mul,
phi::ElementwiseMulOpArgumentMapping); phi::ElementwiseMulOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(elementwise_div, PD_REGISTER_ARG_MAPPING_FN(elementwise_div,
phi::ElementwiseDivOpArgumentMapping); phi::ElementwiseDivOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(elementwise_add_grad, PD_REGISTER_ARG_MAPPING_FN(elementwise_add_grad,
phi::ElementwiseAddGradOpArgumentMapping); phi::ElementwiseAddGradOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(elementwise_add_grad_grad, PD_REGISTER_ARG_MAPPING_FN(elementwise_add_grad_grad,
phi::ElementwiseAddDoubleGradOpArgumentMapping); phi::ElementwiseAddDoubleGradOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(elementwise_add_triple_grad, PD_REGISTER_ARG_MAPPING_FN(elementwise_add_triple_grad,
phi::ElementwiseAddTripleGradOpArgumentMapping); phi::ElementwiseAddTripleGradOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(elementwise_sub_grad, PD_REGISTER_ARG_MAPPING_FN(elementwise_sub_grad,
phi::ElementwiseSubGradOpArgumentMapping); phi::ElementwiseSubGradOpArgumentMapping);
paddle/phi/ops/compat/empty_sig.cc
浏览文件 @ 4a338796
...@@ -28,4 +28,4 @@ KernelSignature EmptyOpArgumentMapping(const ArgumentMappingContext& ctx) { ...@@ -28,4 +28,4 @@ KernelSignature EmptyOpArgumentMapping(const ArgumentMappingContext& ctx) {
} // namespace phi } // namespace phi
PT_REGISTER_ARG_MAPPING_FN(empty, phi::EmptyOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(empty, phi::EmptyOpArgumentMapping);
paddle/phi/ops/compat/expand_sig.cc
浏览文件 @ 4a338796
...@@ -47,8 +47,8 @@ KernelSignature ExpandGradOpArgumentMapping(const ArgumentMappingContext& ctx) { ...@@ -47,8 +47,8 @@ KernelSignature ExpandGradOpArgumentMapping(const ArgumentMappingContext& ctx) {
} // namespace phi } // namespace phi
PT_REGISTER_BASE_KERNEL_NAME(expand_v2, expand); PD_REGISTER_BASE_KERNEL_NAME(expand_v2, expand);
PT_REGISTER_BASE_KERNEL_NAME(expand_v2_grad, expand_grad); PD_REGISTER_BASE_KERNEL_NAME(expand_v2_grad, expand_grad);
PT_REGISTER_ARG_MAPPING_FN(expand_v2, phi::ExpandOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(expand_v2, phi::ExpandOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(expand_v2_grad, phi::ExpandGradOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(expand_v2_grad, phi::ExpandGradOpArgumentMapping);
paddle/phi/ops/compat/fill_any_like_sig.cc
浏览文件 @ 4a338796
...@@ -23,6 +23,6 @@ KernelSignature FillAnyLikeOpArgumentMapping( ...@@ -23,6 +23,6 @@ KernelSignature FillAnyLikeOpArgumentMapping(
} // namespace phi } // namespace phi
PT_REGISTER_BASE_KERNEL_NAME(fill_any_like, full_like); PD_REGISTER_BASE_KERNEL_NAME(fill_any_like, full_like);
PT_REGISTER_ARG_MAPPING_FN(fill_any_like, phi::FillAnyLikeOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(fill_any_like, phi::FillAnyLikeOpArgumentMapping);
paddle/phi/ops/compat/fill_constant_sig.cc
浏览文件 @ 4a338796
...@@ -123,6 +123,6 @@ KernelSignature FillConstantOpArgumentMapping( ...@@ -123,6 +123,6 @@ KernelSignature FillConstantOpArgumentMapping(
} // namespace phi } // namespace phi
PT_REGISTER_BASE_KERNEL_NAME(fill_constant, full); PD_REGISTER_BASE_KERNEL_NAME(fill_constant, full);
PT_REGISTER_ARG_MAPPING_FN(fill_constant, phi::FillConstantOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(fill_constant, phi::FillConstantOpArgumentMapping);
paddle/phi/ops/compat/flatten_sig.cc
浏览文件 @ 4a338796
...@@ -36,10 +36,10 @@ KernelSignature FlattenGradOpArgumentMapping( ...@@ -36,10 +36,10 @@ KernelSignature FlattenGradOpArgumentMapping(
} // namespace phi } // namespace phi
PT_REGISTER_BASE_KERNEL_NAME(flatten_contiguous_range, flatten); PD_REGISTER_BASE_KERNEL_NAME(flatten_contiguous_range, flatten);
PT_REGISTER_BASE_KERNEL_NAME(flatten_contiguous_range_grad, flatten_grad); PD_REGISTER_BASE_KERNEL_NAME(flatten_contiguous_range_grad, flatten_grad);
PT_REGISTER_ARG_MAPPING_FN(flatten_contiguous_range, PD_REGISTER_ARG_MAPPING_FN(flatten_contiguous_range,
phi::FlattenOpArgumentMapping); phi::FlattenOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(flatten_contiguous_range_grad, PD_REGISTER_ARG_MAPPING_FN(flatten_contiguous_range_grad,
phi::FlattenGradOpArgumentMapping); phi::FlattenGradOpArgumentMapping);
paddle/phi/ops/compat/histogram_sig.cc
浏览文件 @ 4a338796
...@@ -22,4 +22,4 @@ KernelSignature HistogramOpArgumentMapping(const ArgumentMappingContext& ctx) { ...@@ -22,4 +22,4 @@ KernelSignature HistogramOpArgumentMapping(const ArgumentMappingContext& ctx) {
} // namespace phi } // namespace phi
PT_REGISTER_ARG_MAPPING_FN(histogram, phi::HistogramOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(histogram, phi::HistogramOpArgumentMapping);
paddle/phi/ops/compat/huber_loss_sig.cc
浏览文件 @ 4a338796
...@@ -31,6 +31,6 @@ KernelSignature HuberLossGradOpArgumentMapping( ...@@ -31,6 +31,6 @@ KernelSignature HuberLossGradOpArgumentMapping(
} // namespace phi } // namespace phi
PT_REGISTER_ARG_MAPPING_FN(huber_loss, phi::HuberLossOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(huber_loss, phi::HuberLossOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(huber_loss_grad, PD_REGISTER_ARG_MAPPING_FN(huber_loss_grad,
phi::HuberLossGradOpArgumentMapping); phi::HuberLossGradOpArgumentMapping);
paddle/phi/ops/compat/lerp_sig.cc
浏览文件 @ 4a338796
...@@ -29,5 +29,5 @@ KernelSignature LerpGradOpArgumentMapping(const ArgumentMappingContext& ctx) { ...@@ -29,5 +29,5 @@ KernelSignature LerpGradOpArgumentMapping(const ArgumentMappingContext& ctx) {
} // namespace phi } // namespace phi
PT_REGISTER_ARG_MAPPING_FN(lerp, phi::LerpOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(lerp, phi::LerpOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(lerp_grad, phi::LerpGradOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(lerp_grad, phi::LerpGradOpArgumentMapping);
paddle/phi/ops/compat/masked_select_sig.cc
浏览文件 @ 4a338796
...@@ -31,6 +31,6 @@ KernelSignature MaskedSelectGradOpArgumentMapping( ...@@ -31,6 +31,6 @@ KernelSignature MaskedSelectGradOpArgumentMapping(
} // namespace phi } // namespace phi
PT_REGISTER_ARG_MAPPING_FN(masked_select, phi::MaskedSelectOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(masked_select, phi::MaskedSelectOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(masked_select_grad, PD_REGISTER_ARG_MAPPING_FN(masked_select_grad,
phi::MaskedSelectGradOpArgumentMapping); phi::MaskedSelectGradOpArgumentMapping);
paddle/phi/ops/compat/matmul_sig.cc
浏览文件 @ 4a338796
...@@ -49,13 +49,13 @@ KernelSignature MatmulTripleGradOpArgumentMapping( ...@@ -49,13 +49,13 @@ KernelSignature MatmulTripleGradOpArgumentMapping(
} // namespace phi } // namespace phi
PT_REGISTER_BASE_KERNEL_NAME(matmul_v2, matmul); PD_REGISTER_BASE_KERNEL_NAME(matmul_v2, matmul);
PT_REGISTER_BASE_KERNEL_NAME(matmul_v2_grad, matmul_grad); PD_REGISTER_BASE_KERNEL_NAME(matmul_v2_grad, matmul_grad);
PT_REGISTER_BASE_KERNEL_NAME(matmul_v2_grad_grad, matmul_double_grad); PD_REGISTER_BASE_KERNEL_NAME(matmul_v2_grad_grad, matmul_double_grad);
PT_REGISTER_BASE_KERNEL_NAME(matmul_v2_triple_grad, matmul_triple_grad); PD_REGISTER_BASE_KERNEL_NAME(matmul_v2_triple_grad, matmul_triple_grad);
PT_REGISTER_ARG_MAPPING_FN(matmul_v2_grad, phi::MatmulGradOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(matmul_v2_grad, phi::MatmulGradOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(matmul_v2_grad_grad, PD_REGISTER_ARG_MAPPING_FN(matmul_v2_grad_grad,
phi::MatmulDoubleGradOpArgumentMapping); phi::MatmulDoubleGradOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(matmul_v2_triple_grad, PD_REGISTER_ARG_MAPPING_FN(matmul_v2_triple_grad,
phi::MatmulTripleGradOpArgumentMapping); phi::MatmulTripleGradOpArgumentMapping);
paddle/phi/ops/compat/norm_sig.cc
浏览文件 @ 4a338796
...@@ -30,5 +30,5 @@ KernelSignature NormGradOpArgumentMapping(const ArgumentMappingContext& ctx) { ...@@ -30,5 +30,5 @@ KernelSignature NormGradOpArgumentMapping(const ArgumentMappingContext& ctx) {
} // namespace phi } // namespace phi
PT_REGISTER_ARG_MAPPING_FN(norm, phi::NormOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(norm, phi::NormOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(norm_grad, phi::NormGradOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(norm_grad, phi::NormGradOpArgumentMapping);
paddle/phi/ops/compat/reduce_sig.cc
浏览文件 @ 4a338796
...@@ -45,8 +45,8 @@ KernelSignature ReduceMeanOpArgumentMapping(const ArgumentMappingContext& ctx) { ...@@ -45,8 +45,8 @@ KernelSignature ReduceMeanOpArgumentMapping(const ArgumentMappingContext& ctx) {
} // namespace phi } // namespace phi
PT_REGISTER_BASE_KERNEL_NAME(reduce_sum, sum); PD_REGISTER_BASE_KERNEL_NAME(reduce_sum, sum);
PT_REGISTER_BASE_KERNEL_NAME(reduce_mean, mean); PD_REGISTER_BASE_KERNEL_NAME(reduce_mean, mean);
PT_REGISTER_ARG_MAPPING_FN(reduce_sum, phi::ReduceSumOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(reduce_sum, phi::ReduceSumOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(reduce_mean, phi::ReduceMeanOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(reduce_mean, phi::ReduceMeanOpArgumentMapping);
paddle/phi/ops/compat/reshape_sig.cc
浏览文件 @ 4a338796
...@@ -45,11 +45,11 @@ KernelSignature ReshapeDoubleGradOpArgumentMapping( ...@@ -45,11 +45,11 @@ KernelSignature ReshapeDoubleGradOpArgumentMapping(
} // namespace phi } // namespace phi
PT_REGISTER_BASE_KERNEL_NAME(reshape2, reshape); PD_REGISTER_BASE_KERNEL_NAME(reshape2, reshape);
PT_REGISTER_BASE_KERNEL_NAME(reshape2_grad, reshape_grad); PD_REGISTER_BASE_KERNEL_NAME(reshape2_grad, reshape_grad);
PT_REGISTER_BASE_KERNEL_NAME(reshape2_grad_grad, reshape_double_grad); PD_REGISTER_BASE_KERNEL_NAME(reshape2_grad_grad, reshape_double_grad);
PT_REGISTER_ARG_MAPPING_FN(reshape2, phi::ReshapeOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(reshape2, phi::ReshapeOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(reshape2_grad, phi::ReshapeGradOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(reshape2_grad, phi::ReshapeGradOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(reshape2_grad_grad, PD_REGISTER_ARG_MAPPING_FN(reshape2_grad_grad,
phi::ReshapeDoubleGradOpArgumentMapping); phi::ReshapeDoubleGradOpArgumentMapping);
paddle/phi/ops/compat/scale_sig.cc
浏览文件 @ 4a338796
...@@ -72,4 +72,4 @@ KernelSignature ScaleOpArgumentMapping(const ArgumentMappingContext& ctx) { ...@@ -72,4 +72,4 @@ KernelSignature ScaleOpArgumentMapping(const ArgumentMappingContext& ctx) {
} // namespace phi } // namespace phi
// op_type, api_name, arg_mapping_fn // op_type, api_name, arg_mapping_fn
PT_REGISTER_ARG_MAPPING_FN(scale, phi::ScaleOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(scale, phi::ScaleOpArgumentMapping);
paddle/phi/ops/compat/split_sig.cc
浏览文件 @ 4a338796
...@@ -46,4 +46,4 @@ KernelSignature SplitOpArgumentMapping(const ArgumentMappingContext& ctx) { ...@@ -46,4 +46,4 @@ KernelSignature SplitOpArgumentMapping(const ArgumentMappingContext& ctx) {
} // namespace phi } // namespace phi
PT_REGISTER_ARG_MAPPING_FN(split, phi::SplitOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(split, phi::SplitOpArgumentMapping);
paddle/phi/ops/compat/trace_sig.cc
浏览文件 @ 4a338796
...@@ -30,5 +30,5 @@ KernelSignature TraceGradOpArgumentMapping(const ArgumentMappingContext& ctx) { ...@@ -30,5 +30,5 @@ KernelSignature TraceGradOpArgumentMapping(const ArgumentMappingContext& ctx) {
} // namespace phi } // namespace phi
PT_REGISTER_ARG_MAPPING_FN(trace, phi::TraceOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(trace, phi::TraceOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(trace_grad, phi::TraceGradOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(trace_grad, phi::TraceGradOpArgumentMapping);
paddle/phi/ops/compat/trunc_sig.cc
浏览文件 @ 4a338796
...@@ -27,5 +27,5 @@ KernelSignature TruncGradOpArgumentMapping(const ArgumentMappingContext& ctx) { ...@@ -27,5 +27,5 @@ KernelSignature TruncGradOpArgumentMapping(const ArgumentMappingContext& ctx) {
} // namespace phi } // namespace phi
PT_REGISTER_ARG_MAPPING_FN(trunc, phi::TruncOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(trunc, phi::TruncOpArgumentMapping);
PT_REGISTER_ARG_MAPPING_FN(trunc_grad, phi::TruncGradOpArgumentMapping); PD_REGISTER_ARG_MAPPING_FN(trunc_grad, phi::TruncGradOpArgumentMapping);
paddle/phi/tests/core/test_custom_kernel.cc
浏览文件 @ 4a338796
...@@ -17,6 +17,8 @@ limitations under the License. */ ...@@ -17,6 +17,8 @@ limitations under the License. */
#define _LINUX #define _LINUX
#endif #endif
#include <gtest/gtest.h>
#ifdef _LINUX #ifdef _LINUX
#include "paddle/fluid/framework/lod_tensor.h" #include "paddle/fluid/framework/lod_tensor.h"
#include "paddle/fluid/platform/device_context.h" #include "paddle/fluid/platform/device_context.h"
...@@ -30,8 +32,6 @@ limitations under the License. */ ...@@ -30,8 +32,6 @@ limitations under the License. */
#include "paddle/phi/core/meta_tensor.h" #include "paddle/phi/core/meta_tensor.h"
#include "paddle/phi/infermeta/binary.h" #include "paddle/phi/infermeta/binary.h"
#include <gtest/gtest.h>
// user kernel function // user kernel function
namespace custom_kernel { namespace custom_kernel {
...@@ -98,16 +98,16 @@ void FakeDot(const Context& dev_ctx, ...@@ -98,16 +98,16 @@ void FakeDot(const Context& dev_ctx,
} }
} // namespace custom_kernel } // namespace custom_kernel
PD_REGISTER_KERNEL(fake_dot, PD_REGISTER_BUILTIN_KERNEL(fake_dot,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
custom_kernel::FakeDot, custom_kernel::FakeDot,
float, float,
double, double,
int, int,
int64_t, int64_t,
int8_t, int8_t,
uint8_t) {} uint8_t) {}
namespace phi { namespace phi {
namespace tests { namespace tests {
......
paddle/phi/tests/core/test_kernel_factory.cc
浏览文件 @ 4a338796
...@@ -22,7 +22,7 @@ limitations under the License. */ ...@@ -22,7 +22,7 @@ limitations under the License. */
#include "gtest/gtest.h" #include "gtest/gtest.h"
PT_DECLARE_KERNEL(scale, CPU, ALL_LAYOUT); PD_DECLARE_KERNEL(scale, CPU, ALL_LAYOUT);
namespace phi { namespace phi {
namespace tests { namespace tests {
...@@ -76,7 +76,7 @@ TEST(KernelRegistry, SetFP32Input) { ...@@ -76,7 +76,7 @@ TEST(KernelRegistry, SetFP32Input) {
} // namespace tests } // namespace tests
} // namespace phi } // namespace phi
PT_REGISTER_KERNEL(test, PD_REGISTER_KERNEL(test,
CPU, CPU,
ALL_LAYOUT, ALL_LAYOUT,
phi::tests::TestKernel, phi::tests::TestKernel,
......
paddle/phi/tests/kernels/test_flatten_dev_api.cc
浏览文件 @ 4a338796
...@@ -23,14 +23,14 @@ limitations under the License. */ ...@@ -23,14 +23,14 @@ limitations under the License. */
#include "paddle/phi/core/dense_tensor.h" #include "paddle/phi/core/dense_tensor.h"
#include "paddle/phi/core/kernel_registry.h" #include "paddle/phi/core/kernel_registry.h"
PT_DECLARE_KERNEL(copy, CPU, ALL_LAYOUT); PD_DECLARE_KERNEL(copy, CPU, ALL_LAYOUT);
#if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP) #if defined(PADDLE_WITH_CUDA) || defined(PADDLE_WITH_HIP)
PT_DECLARE_KERNEL(copy, GPU, ALL_LAYOUT); PD_DECLARE_KERNEL(copy, GPU, ALL_LAYOUT);
#endif #endif
#ifdef PADDLE_WITH_XPU #ifdef PADDLE_WITH_XPU
PT_DECLARE_KERNEL(copy, XPU, ALL_LAYOUT); PD_DECLARE_KERNEL(copy, XPU, ALL_LAYOUT);
#endif #endif
namespace phi { namespace phi {
......
python/paddle/fluid/tests/custom_kernel/custom_kernel_dot.cc
浏览文件 @ 4a338796
...@@ -45,7 +45,7 @@ void DotKernel(const Context& dev_ctx, ...@@ -45,7 +45,7 @@ void DotKernel(const Context& dev_ctx,
} // namespace custom_kernel } // namespace custom_kernel
} // namespace paddle } // namespace paddle
PD_REGISTER_KERNEL( PD_REGISTER_BUILTIN_KERNEL(
dot, CPU, ALL_LAYOUT, paddle::custom_kernel::DotKernel, int8_t) { dot, CPU, ALL_LAYOUT, paddle::custom_kernel::DotKernel, int8_t) {
kernel->OutputAt(0).SetDataType(paddle::experimental::DataType::INT8); kernel->OutputAt(0).SetDataType(paddle::experimental::DataType::INT8);
} }
python/paddle/utils/code_gen/api_gen.py
浏览文件 @ 4a338796
...@@ -118,7 +118,7 @@ def source_include(header_file_path): ...@@ -118,7 +118,7 @@ def source_include(header_file_path):
def api_register(): def api_register():
return """ return """
PT_REGISTER_API(Math); PD_REGISTER_API(Math);
""" """
......
python/paddle/utils/code_gen/wrapped_infermeta_gen.py
浏览文件 @ 4a338796
...@@ -26,7 +26,7 @@ def get_wrapped_infermeta_name(api_name): ...@@ -26,7 +26,7 @@ def get_wrapped_infermeta_name(api_name):
def gene_wrapped_infermeta_and_register(api): def gene_wrapped_infermeta_and_register(api):
if api.is_base_api and not api.is_dygraph_api: if api.is_base_api and not api.is_dygraph_api:
register_code = f""" register_code = f"""
PT_REGISTER_INFER_META_FN({api.kernel['func'][0]}, phi::{api.infer_meta['func']});""" PD_REGISTER_INFER_META_FN({api.kernel['func'][0]}, phi::{api.infer_meta['func']});"""
if api.infer_meta['param'] is not None: if api.infer_meta['param'] is not None:
kernel_params = api.kernel['param'] kernel_params = api.kernel['param']
...@@ -73,7 +73,7 @@ void {wrapped_infermeta_name}({", ".join(args)}) {{ ...@@ -73,7 +73,7 @@ void {wrapped_infermeta_name}({", ".join(args)}) {{
""" """
register_code = f""" register_code = f"""
PT_REGISTER_INFER_META_FN({api.kernel['func'][0]}, phi::{get_wrapped_infermeta_name(api.kernel['func'][0])});""" PD_REGISTER_INFER_META_FN({api.kernel['func'][0]}, phi::{get_wrapped_infermeta_name(api.kernel['func'][0])});"""
return declare_code, defind_code, register_code return declare_code, defind_code, register_code
else: else:
......
tools/infrt/get_pten_kernel_function.sh
浏览文件 @ 4a338796
...@@ -24,9 +24,9 @@ set -e ...@@ -24,9 +24,9 @@ set -e
kernel_register_info_file=`mktemp` kernel_register_info_file=`mktemp`
PADDLE_ROOT="$( cd "$( dirname "$0" )/../../" && pwd )" PADDLE_ROOT="$( cd "$( dirname "$0" )/../../" && pwd )"
unset GREP_OPTIONS && find ${PADDLE_ROOT}/paddle/phi/kernels -name "*.c*" \ unset GREP_OPTIONS && find ${PADDLE_ROOT}/paddle/phi/kernels -name "*.c*" \
| xargs sed -e '/PT_REGISTER_\(GENERAL_\)\?KERNEL(/,/)/!d' \ | xargs sed -e '/PD_REGISTER_\(GENERAL_\)\?KERNEL(/,/)/!d' \
| awk 'BEGIN { RS="{" }{ gsub(/\n /,""); print $0 }' \ | awk 'BEGIN { RS="{" }{ gsub(/\n /,""); print $0 }' \
| grep PT_REGISTER \ | grep PD_REGISTER \
| awk -F ",|\(|\)" '{gsub(/ /,"");$1="";print}' \ | awk -F ",|\(|\)" '{gsub(/ /,"");$1="";print}' \
| sort -u | awk '{gsub(/phi::/,"");gsub(/paddle::platform::/,"");gsub(/dtype::/,"");gsub(/paddle::/,"");print $0}' \ | sort -u | awk '{gsub(/phi::/,"");gsub(/paddle::platform::/,"");gsub(/dtype::/,"");gsub(/paddle::/,"");print $0}' \
| grep -v "_grad" > $kernel_register_info_file | grep -v "_grad" > $kernel_register_info_file
...@@ -38,7 +38,7 @@ python3 ${PADDLE_ROOT}/python/paddle/utils/code_gen/wrapped_infermeta_gen.py \ ...@@ -38,7 +38,7 @@ python3 ${PADDLE_ROOT}/python/paddle/utils/code_gen/wrapped_infermeta_gen.py \
--wrapped_infermeta_header_path ${temp_path}/generate.h \ --wrapped_infermeta_header_path ${temp_path}/generate.h \
--wrapped_infermeta_source_path ${temp_path}/generate.cc --wrapped_infermeta_source_path ${temp_path}/generate.cc
grep PT_REGISTER_INFER_META_FN ${temp_path}/generate.cc \ grep PD_REGISTER_INFER_META_FN ${temp_path}/generate.cc \
| awk -F "\(|,|::|\)" '{print $2, $4}' > ${temp_path}/wrap_info.txt | awk -F "\(|,|::|\)" '{print $2, $4}' > ${temp_path}/wrap_info.txt
#step 3: merge all infos #step 3: merge all infos
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或