Skip to content

GitLab

"...git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "c54f36f0872f0cf131a0880cb80b69cc9ad8f346"
Commit bb99f03d authored by yuguo-Jack's avatar yuguo-Jack
Browse files

fix depthwise conv grad bug

parent 993cf390
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 6 deletions
paddle/phi/kernels/gpudnn/conv_grad_kernel.cu
View file @ bb99f03d
...@@ -1469,12 +1469,12 @@ PD_REGISTER_KERNEL(conv3d_grad, ...@@ -1469,12 +1469,12 @@ PD_REGISTER_KERNEL(conv3d_grad,
float, float,
phi::dtype::float16) {} phi::dtype::float16) {}
PD_REGISTER_KERNEL(depthwise_conv2d_grad, // PD_REGISTER_KERNEL(depthwise_conv2d_grad,
GPUDNN, // GPUDNN,
ALL_LAYOUT, // ALL_LAYOUT,
phi::DepthwiseConvCudnnGradKernel, // phi::DepthwiseConvCudnnGradKernel,
float, // float,
phi::dtype::float16) {} // phi::dtype::float16) {}
PD_REGISTER_KERNEL(conv2d_double_grad, PD_REGISTER_KERNEL(conv2d_double_grad,
GPUDNN, GPUDNN,
ALL_LAYOUT, ALL_LAYOUT,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment