未验证 提交 137dc3e3 编写于 作者: W wuyefeilin 提交者: GitHub

add fp16 kernel to clip_grad (#41661)

上级 cade0018
...@@ -15,6 +15,7 @@ ...@@ -15,6 +15,7 @@
#include "paddle/phi/kernels/clip_grad_kernel.h" #include "paddle/phi/kernels/clip_grad_kernel.h"
#include "paddle/phi/backends/gpu/gpu_context.h" #include "paddle/phi/backends/gpu/gpu_context.h"
#include "paddle/phi/common/float16.h"
#include "paddle/phi/core/kernel_registry.h" #include "paddle/phi/core/kernel_registry.h"
#include "paddle/phi/kernels/impl/clip_grad_kernel_impl.h" #include "paddle/phi/kernels/impl/clip_grad_kernel_impl.h"
...@@ -25,4 +26,5 @@ PD_REGISTER_KERNEL(clip_grad, ...@@ -25,4 +26,5 @@ PD_REGISTER_KERNEL(clip_grad,
float, float,
double, double,
int, int,
int64_t) {} int64_t,
phi::dtype::float16) {}
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册