Skip to content

Commit 30fb07e

Browse files
authored
fix search sort bug (#41664) (#41684)
* fix search sort bug (#41664) * fix depthwise dnn bug (#41666)
1 parent 58c9903 commit 30fb07e

File tree

3 files changed

+3
-0
lines changed

3 files changed

+3
-0
lines changed

paddle/phi/kernels/gpudnn/conv_grad_kernel.cu

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -640,6 +640,7 @@ void DepthwiseConvCudnnGradKernel(const Context& dev_ctx,
640640
bool use_addto,
641641
int workspace_size_MB,
642642
bool exhaustive_search,
643+
bool fuse_relu,
643644
DenseTensor* input_grad,
644645
DenseTensor* filter_grad) {
645646
ConvCudnnGradKernel<T>(dev_ctx,

paddle/phi/kernels/gpudnn/conv_kernel.cu

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -431,6 +431,7 @@ void DepthwiseConvCudnnKernel(const Context& dev_ctx,
431431
bool use_addto,
432432
int workspace_size_MB,
433433
bool exhaustive_search,
434+
bool fuse_relu,
434435
DenseTensor* out) {
435436
ConvCudnnKernel<T>(dev_ctx,
436437
input,

python/paddle/utils/code_gen/api.yaml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1614,6 +1614,7 @@
16141614
func : SearchsortedInferMeta
16151615
kernel :
16161616
func : searchsorted
1617+
data_type : sorted_sequence
16171618

16181619
# segment_pool
16191620
- api : segment_pool

0 commit comments

Comments
 (0)