Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!18441 Fix conv3d cudnn algorithm error 

Merge pull request !18441 from tom_chen/conv3d
This commit is contained in:
i-robot 2021-06-17 22:31:13 +08:00 committed by Gitee
parent ab3259c43a 35f6a1af56
commit 26c7d274c9
2 changed files with 0 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/backend/kernel_compiler/gpu/nn/conv3d_gpu_kernel.h
View File

@ -320,9 +320,6 @@ class Conv3dGpuKernel : public GpuKernel {
output_desc_, requested_algo_count, &returned_algo_count, &perf_results),
"cudnnGetConvolutionForwardAlgorithm_v7 failed");
conv_algorithm_ = perf_results.algo;
if (cudnn_data_type_ == CUDNN_DATA_HALF) {
conv_algorithm_ = CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_PRECOMP_GEMM;
}
}
void SetStrideAndDilation(const CNodePtr &kernel_node) {

3
mindspore/ccsrc/backend/kernel_compiler/gpu/nn/conv3d_grad_input_gpu_kernel.h
View File

@ -297,9 +297,6 @@ class Conv3dGradInputGpuKernel : public GpuKernel {
requested_algo_count, &returned_algo_count, &perf_results),
"cudnnGetConvolutionBackwardDataAlgorithm_v7 failed");
algo_ = perf_results.algo;
if (cudnn_data_type_ == CUDNN_DATA_HALF) {
algo_ = CUDNN_CONVOLUTION_BWD_DATA_ALGO_1;
}
}
void GetInputShape(const CNodePtr &kernel_node, std::vector<size_t> *input_shape) {