Use cudnnGetConvolutionForwardAlgorithm_v7 when cudnn 8 or later
@@ -1380,7 +1380,7 @@ cpdef _Algorithm _get_algorithm_fwd(
return algo
cdef list ret
cdef bint skip
- if use_tensor_core and _cudnn_version >= 7000:
+ if (use_tensor_core and _cudnn_version >= 7000) or _cudnn_version >= 8000:
ret = cudnn.getConvolutionForwardAlgorithm_v7(
handle, x_desc, filter_desc, conv_desc, y_desc, 10)
skip = False
