fix the number of perf algo for conv cudnn in exhaustive mode (#28694)

musl/fix_failed_unittests_in_musl
wangchaochaohu 5 years ago committed by GitHub
parent 8c0ea4bffe
commit 8b853b3030
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -429,7 +429,7 @@ struct SearchAlgorithm<cudnnConvolutionBwdDataAlgoPerf_t> {
x_dims, w_dims, args.s, args.p, args.d, 0,
static_cast<int64_t>(args.cudnn_dtype), [&]() {
int returned_algo_count;
std::array<perf_t, kNUM_CUDNN_FWD_ALGS> perf_stat;
std::array<perf_t, kNUM_CUDNN_BWD_DATA_ALGS> perf_stat;
auto cudnn_find_func = [&](void* cudnn_workspace_ptr) {
PADDLE_ENFORCE_CUDA_SUCCESS(
@ -561,7 +561,7 @@ struct SearchAlgorithm<cudnnConvolutionBwdFilterAlgoPerf_t> {
x_dims, w_dims, args.s, args.p, args.d, 0,
static_cast<int64_t>(args.cudnn_dtype), [&]() {
int returned_algo_count;
std::array<perf_t, kNUM_CUDNN_FWD_ALGS> perf_stat;
std::array<perf_t, kNUM_CUDNN_BWD_FILTER_ALGS> perf_stat;
auto cudnn_find_func = [&](void* cudnn_workspace_ptr) {
PADDLE_ENFORCE_CUDA_SUCCESS(
platform::dynload::

Loading…
Cancel
Save