releaase 2.11 (#703)
This commit is contained in:
@@ -195,7 +195,12 @@ bool get_cudnn_mathtype(cudnnMathType_t &cudnn_math_type, library::ConvDescripti
|
||||
return true;
|
||||
}
|
||||
case library::OpcodeClassID::kSimt:
|
||||
return false;
|
||||
#if (defined(CUDNN_VERSION) && CUDNN_VERSION <= 8000)
|
||||
cudnn_math_type = CUDNN_DEFAULT_MATH;
|
||||
#else
|
||||
cudnn_math_type = CUDNN_FMA_MATH;
|
||||
#endif
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
|
||||
Reference in New Issue
Block a user