From b29c05c7a62f26c62c10bcfd133e903ec7555437 Mon Sep 17 00:00:00 2001
From: Chen Weihang <chenweihang@baidu.com>
Date: Sat, 19 Feb 2022 13:09:24 +0800
Subject: [PATCH] [PTen] Support parse cc file in gpu (#39691)

* support parse cc in gpu

* change file name
---
 cmake/pten.cmake                                         | 9 ++++++---
 .../kernels/gpu/{sign_kernel.cu => sign_kernel.cu.cc}    | 2 +-
 2 files changed, 7 insertions(+), 4 deletions(-)
 rename paddle/pten/kernels/gpu/{sign_kernel.cu => sign_kernel.cu.cc} (100%)

diff --git a/cmake/pten.cmake b/cmake/pten.cmake
index 2a040c73b98..8475c33e1e8 100644
--- a/cmake/pten.cmake
+++ b/cmake/pten.cmake
@@ -114,6 +114,9 @@ function(kernel_library TARGET)
             if (EXISTS ${CMAKE_CURRENT_SOURCE_DIR}/gpu/${TARGET}.cu)
                 list(APPEND gpu_srcs ${CMAKE_CURRENT_SOURCE_DIR}/gpu/${TARGET}.cu)
             endif()
+            if (EXISTS ${CMAKE_CURRENT_SOURCE_DIR}/gpu/${TARGET}.cu.cc)
+                list(APPEND gpu_srcs ${CMAKE_CURRENT_SOURCE_DIR}/gpu/${TARGET}.cu.cc)
+            endif()
         endif()
         if (WITH_XPU)
             if (EXISTS ${CMAKE_CURRENT_SOURCE_DIR}/xpu/${TARGET}.cc)
@@ -161,7 +164,7 @@ function(kernel_library TARGET)
 
     # Build Target according different src organization
     if((${cpu_srcs_len} GREATER 0 OR ${gpu_srcs_len} GREATER 0 OR
-        ${xpu_srcs_len} GREATER 0) AND (${common_srcs_len} GREATER 0 OR 
+        ${xpu_srcs_len} GREATER 0) AND (${common_srcs_len} GREATER 0 OR
         ${selected_rows_srcs_len} GREATER 0))
         # If the common_srcs/selected_rows_srcs depends on specific device srcs, build target using this rule.
         if (WITH_GPU)
@@ -225,11 +228,11 @@ function(kernel_library TARGET)
             cc_library(${TARGET} SRCS ${selected_rows_srcs} DEPS ${kernel_library_DEPS} ${kernel_deps})
         endif()
     else()
-         message(FATAL_ERROR "Cannot find any implementation for ${TARGET}")
+        message(FATAL_ERROR "Cannot find any implementation for ${TARGET}")
     endif()
 
     if (${common_srcs_len} GREATER 0 OR ${cpu_srcs_len} GREATER 0 OR
-        ${gpu_srcs_len} GREATER 0 OR ${xpu_srcs_len} GREATER 0 OR 
+        ${gpu_srcs_len} GREATER 0 OR ${xpu_srcs_len} GREATER 0 OR
         ${selected_rows_srcs_len} GREATER 0)
         # append target into PTEN_KERNELS property
         get_property(pten_kernels GLOBAL PROPERTY PTEN_KERNELS)
diff --git a/paddle/pten/kernels/gpu/sign_kernel.cu b/paddle/pten/kernels/gpu/sign_kernel.cu.cc
similarity index 100%
rename from paddle/pten/kernels/gpu/sign_kernel.cu
rename to paddle/pten/kernels/gpu/sign_kernel.cu.cc
index d479d6a2b2d..61354af5dd5 100644
--- a/paddle/pten/kernels/gpu/sign_kernel.cu
+++ b/paddle/pten/kernels/gpu/sign_kernel.cu.cc
@@ -12,8 +12,8 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License. */
 
-#include "paddle/pten/kernels/impl/sign_kernel_impl.h"
 #include "paddle/pten/kernels/sign_kernel.h"
+#include "paddle/pten/kernels/impl/sign_kernel_impl.h"
 
 #include "paddle/pten/backends/gpu/gpu_context.h"
 #include "paddle/pten/core/kernel_registry.h"
-- 
GitLab