A helper function to query if a CudnnConvolutionDescriptor has tensor_op_math set
| 4024 | // A helper function to query if a CudnnConvolutionDescriptor has tensor_op_math |
| 4025 | // set |
| 4026 | static bool IsTensorMathOpSet(const CudnnConvolutionDescriptor& conv) { |
| 4027 | cudnnMathType_t math_type; |
| 4028 | CHECK_CUDNN_OK(cudnnGetConvolutionMathType(conv.handle(), &math_type)); |
| 4029 | #if CUDNN_VERSION >= 8000 |
| 4030 | return math_type != CUDNN_FMA_MATH; |
| 4031 | #else |
| 4032 | return math_type == CUDNN_TENSOR_OP_MATH; |
| 4033 | #endif |
| 4034 | } |
| 4035 | |
| 4036 | port::Status CudnnSupport::DoConvolve( |
| 4037 | dnn::ConvolutionKind kind, dnn::DataType element_type, |
no test coverage detected