Skip to content

P
Paddle

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

通知 2299
Star 20931
Fork 5422
P
Paddle
未验证 提交 380a9bf7 编写于 作者: C Chitsing KUI 提交者： GitHub
浏览文件
操作

fix backend bug (#52526)

上级 8ac5a6b6
隐藏空白更改
内联 并排
Showing with 2 addition and 2 deletion
paddle/phi/kernels/gpu/flash_attn_grad_kernel.cu
浏览文件 @ 380a9bf7
...@@ -229,7 +229,7 @@ PD_REGISTER_KERNEL(flash_attn_unpadded_grad, ...@@ -229,7 +229,7 @@ PD_REGISTER_KERNEL(flash_attn_unpadded_grad,
phi::FlashAttnUnpaddedGradKernel, phi::FlashAttnUnpaddedGradKernel,
phi::dtype::float16, phi::dtype::float16,
phi::dtype::bfloat16) { phi::dtype::bfloat16) {
kernel->InputAt(7).SetBackend(phi::Backend::CPU); // seed_offset kernel->InputAt(7).SetBackend(phi::Backend::ALL_BACKEND); // seed_offset
} }
PD_REGISTER_KERNEL(flash_attn_grad, PD_REGISTER_KERNEL(flash_attn_grad,
...@@ -238,5 +238,5 @@ PD_REGISTER_KERNEL(flash_attn_grad, ...@@ -238,5 +238,5 @@ PD_REGISTER_KERNEL(flash_attn_grad,
phi::FlashAttnGradKernel, phi::FlashAttnGradKernel,
phi::dtype::float16, phi::dtype::float16,
phi::dtype::bfloat16) { phi::dtype::bfloat16) {
kernel->InputAt(5).SetBackend(phi::Backend::CPU); // seed_offset kernel->InputAt(5).SetBackend(phi::Backend::ALL_BACKEND); // seed_offset
} }
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或