Skip to content

GitLab

Commit bb99f03d authored by yuguo-Jack's avatar yuguo-Jack
Browse files

fix depthwise conv grad bug

parent 993cf390
Show 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