Skip to content

Commit 0fc198f

Browse files
LiaoYFBHYqGe585
andauthored
【CUDA Kernel No.51】exponential_kernel算子Kernel修复 (#2222)
* Add elementwise_grad_kernel.cu to CMakeLists.txt for GPU backend * Fix: Replace .cu include with .h in elementwise_grad_kernel_register.cu * fix: Correct .cu include and CMake for embedding_with_scaled_gradient_grad kernel. * FIX(iluvatar_gpu): Correct include path for embedding_grad kernel registration file * FIX(iluvatar_gpu): Include missing headers for embedding_grad kernel registration. * Change included .cu files to .h --------- Co-authored-by: Yuqiang Ge <[email protected]>
1 parent 6940574 commit 0fc198f

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

backends/iluvatar_gpu/kernels/cuda_kernels/exponential_kernel_register.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
// limitations under the License.
1414

1515
#include "paddle/phi/core/kernel_registry.h"
16-
#include "paddle/phi/kernels/gpu/exponential_kernel.cu" // NOLINT
16+
#include "paddle/phi/kernels/exponential_kernel.h" // NOLINT
1717

1818
PD_CUSTOM_KERNEL_REGISTER(exponential,
1919
iluvatar_gpu,

backends/metax_gpu/kernels/cuda_kernels/exponential_kernel_register.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
// limitations under the License.
1414

1515
#include "paddle/phi/core/kernel_registry.h"
16-
#include "paddle/phi/kernels/gpu/exponential_kernel.cu" // NOLINT
16+
#include "paddle/phi/kernels/exponential_kernel.h" // NOLINT
1717

1818
PD_CUSTOM_KERNEL_REGISTER(exponential,
1919
metax_gpu,

0 commit comments

Comments
 (0)