| 237 | } |
| 238 | |
| 239 | cudnnStatus_t cudnnGetConvolutionForwardAlgorithm( |
| 240 | cudnnHandle_t handle, const cudnnTensorDescriptor_t xDesc, |
| 241 | const cudnnFilterDescriptor_t wDesc, |
| 242 | const cudnnConvolutionDescriptor_t convDesc, |
| 243 | const cudnnTensorDescriptor_t yDesc, |
| 244 | cudnnConvolutionFwdPreference_t preference, size_t memoryLimitInBytes, |
| 245 | cudnnConvolutionFwdAlgo_t *algo) { |
| 246 | auto version = getCudnnPlugin().getVersion(); |
| 247 | if (version.major() < 8) { |
| 248 | return getCudnnPlugin().cudnnGetConvolutionForwardAlgorithm( |
| 249 | handle, xDesc, wDesc, convDesc, yDesc, preference, |
| 250 | memoryLimitInBytes, algo); |
| 251 | } else { |
| 252 | AF_ERROR( |
| 253 | "cudnnGetConvolutionForwardAlgorithm has been removed since cuDNN " |
| 254 | "8", |
| 255 | AF_ERR_NOT_SUPPORTED); |
| 256 | return CUDNN_STATUS_SUCCESS; |
| 257 | } |
| 258 | } |
| 259 | |
| 260 | cudnnStatus_t cudnnGetConvolutionBackwardFilterAlgorithm( |
| 261 | cudnnHandle_t handle, const cudnnTensorDescriptor_t xDesc, |
no test coverage detected