[NVIDIA] Set proper math type for convloution

openvinotoolkit · Aug 4, 2023 · 91c3561 · 91c3561
1 parent 1f0c276
commit 91c3561
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/modules/nvidia_plugin/src/ops/convolution_components/convolution_cudnn_components.cpp b/modules/nvidia_plugin/src/ops/convolution_components/convolution_cudnn_components.cpp
@@ -74,7 +74,7 @@ CUDA::DnnConvolutionDescriptor ConvolutionParamsCuDnn::MakeConvolutionDescriptor
 
     // Enable computations on Tensor Core hardware which requires at least Volta GPU
     // (compute capability 7.0).
-    const cudnnMathType_t math_type = CUDNN_TENSOR_OP_MATH_ALLOW_CONVERSION;
+    const cudnnMathType_t math_type = CUDNN_TENSOR_OP_MATH;
     throwIfError(::cudnnSetConvolutionMathType(conv_desc.get(), math_type));
     throwIfError(::cudnnSetConvolutionGroupCount(conv_desc.get(), groups_));
 
@@ -96,6 +96,7 @@ ConvolutionDescriptorsCuDnn::ConvolutionDescriptorsCuDnn(const CreationContext&
     } else {
         GetAlgo(dnnHandle);
     }
+    throwIfError(::cudnnSetConvolutionMathType(conv_.get(), algo_perf_.mathType));
 }
 
 void ConvolutionDescriptorsCuDnn::BenchmarkOptimalAlgo(const CUDA::DnnHandle& dnnHandle,