Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/torchaudio.git" did not exist on "980528e9d74d733e2cbafad61e2f77b0ac8b6409"
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,
float,
phi::dtype::float16) {}
PD_REGISTER_KERNEL(depthwise_conv2d_grad,
GPUDNN,
ALL_LAYOUT,
phi::DepthwiseConvCudnnGradKernel,
float,
phi::dtype::float16) {}
// PD_REGISTER_KERNEL(depthwise_conv2d_grad,
// GPUDNN,
// ALL_LAYOUT,
// phi::DepthwiseConvCudnnGradKernel,
// float,
// phi::dtype::float16) {}
PD_REGISTER_KERNEL(conv2d_double_grad,
GPUDNN,
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