Skip to content

Commit cc7a755

Browse files
committed
set CudnnPooling don't include padded values in default
1 parent 73da972 commit cc7a755

File tree

3 files changed

+8
-8
lines changed

3 files changed

+8
-8
lines changed

paddle/cuda/include/hl_cuda_cudnn.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -22,10 +22,10 @@ limitations under the License. */
2222
*/
2323
typedef enum {
2424
HL_POOLING_MAX = 0,
25-
// average includes padded values
26-
HL_POOLING_AVERAGE = 1,
2725
// average does not include padded values
28-
HL_POOLING_AVERAGE_EXCLUDE_PADDING = 2,
26+
HL_POOLING_AVERAGE = 1,
27+
// average includes padded values
28+
HL_POOLING_AVERAGE_INCLUDE_PADDING = 2,
2929
HL_POOLING_END
3030
} hl_pooling_mode_t;
3131

paddle/cuda/src/hl_cuda_cudnn.cc

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -432,11 +432,11 @@ void hl_create_pooling_descriptor(hl_pooling_descriptor* pooling_desc,
432432
cudnn_mode = CUDNN_POOLING_MAX;
433433
break;
434434
case HL_POOLING_AVERAGE:
435-
cudnn_mode = CUDNN_POOLING_AVERAGE_COUNT_INCLUDE_PADDING;
436-
break;
437-
case HL_POOLING_AVERAGE_EXCLUDE_PADDING:
438435
cudnn_mode = CUDNN_POOLING_AVERAGE_COUNT_EXCLUDE_PADDING;
439436
break;
437+
case HL_POOLING_AVERAGE_INCLUDE_PADDING:
438+
cudnn_mode = CUDNN_POOLING_AVERAGE_COUNT_INCLUDE_PADDING;
439+
break;
440440
default:
441441
LOG(FATAL) << "parameter mode error";
442442
}

paddle/gserver/layers/CudnnPoolLayer.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,9 @@ bool CudnnPoolLayer::typeCheck(const std::string &poolType,
2929
if (mode) {
3030
*mode = HL_POOLING_AVERAGE;
3131
}
32-
} else if (poolType == "cudnn-avg-excl-pad-pool") {
32+
} else if (poolType == "cudnn-avg-incl-pad-pool") {
3333
if (mode) {
34-
*mode = HL_POOLING_AVERAGE_EXCLUDE_PADDING;
34+
*mode = HL_POOLING_AVERAGE_INCLUDE_PADDING;
3535
}
3636
} else {
3737
return false;

0 commit comments

Comments
 (0)