Skip to content

Commit c1c07bb

Browse files
authored
Update CudnnConvLayer.cpp
1 parent 95da095 commit c1c07bb

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

paddle/gserver/layers/CudnnConvLayer.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -133,6 +133,8 @@ void CudnnConvLayer::reshape(int batchSize) {
133133
getOutput().setFrameHeight(outputH_);
134134
getOutput().setFrameWidth(outputW_);
135135

136+
// if the batchSize remains the same, set isSelectAlgo_ true.
137+
// Otherwise, set isSelectAlgo_ false and select algo again.
136138
isSelectAlgo_ = (batchSize == batchNum_);
137139
batchNum_ = batchSize;
138140

0 commit comments

Comments
 (0)