Skip to content

Commit 2b9ef7e

Browse files
committed
"fix"
1 parent 75d8e8c commit 2b9ef7e

File tree

1 file changed

+13
-13
lines changed

1 file changed

+13
-13
lines changed

paddle/fluid/platform/cudnn_helper.h

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,19 @@ inline const char* cudnnGetErrorString(cudnnStatus_t status) {
6868
} \
6969
} while (false)
7070

71+
enum class DataLayout { // Not use
72+
kNHWC,
73+
kNCHW,
74+
kNCDHW,
75+
kNCHW_VECT_C,
76+
};
77+
78+
enum class PoolingMode {
79+
kMaximum,
80+
kAverage,
81+
kMaximumDeterministic,
82+
};
83+
7184
#if CUDNN_VERSION < 6000
7285
#pragma message "CUDNN version under 6.0 is supported at best effort."
7386
#pragma message "We strongly encourage you to move to 6.0 and above."
@@ -103,19 +116,6 @@ inline cudnnPoolingMode_t GetPoolingMode(const PoolingMode& mode) {
103116
}
104117
#endif // CUDNN_VERSION < 6000
105118

106-
enum class DataLayout { // Not use
107-
kNHWC,
108-
kNCHW,
109-
kNCDHW,
110-
kNCHW_VECT_C,
111-
};
112-
113-
enum class PoolingMode {
114-
kMaximum,
115-
kAverage,
116-
kMaximumDeterministic,
117-
};
118-
119119
template <typename T>
120120
class CudnnDataType;
121121

0 commit comments

Comments
 (0)