| 119 | } |
| 120 | |
| 121 | cudnnStatus_t cudnnSetFilter4dDescriptor(cudnnFilterDescriptor_t filterDesc, |
| 122 | cudnnDataType_t dataType, |
| 123 | cudnnTensorFormat_t format, int k, |
| 124 | int c, int h, int w) { |
| 125 | #if CUDNN_VERSION >= 6000 |
| 126 | int version = getCudnnPlugin().cudnnGetVersion(); |
| 127 | if (version >= 6000) { |
| 128 | return getCudnnPlugin().cudnnSetFilter4dDescriptor(filterDesc, dataType, |
| 129 | format, k, c, h, w); |
| 130 | } else if (version == 4000) { |
| 131 | return getCudnnPlugin().cudnnSetFilter4dDescriptor_v4( |
| 132 | filterDesc, dataType, format, k, c, h, w); |
| 133 | } |
| 134 | CUDA_NOT_SUPPORTED( |
| 135 | "cudnnSetFilter4dDescriptor not supported for the current version of " |
| 136 | "cuDNN"); |
| 137 | #elif CUDNN_VERSION == 4000 |
| 138 | return getCudnnPlugin().cudnnSetFilter4dDescriptor_v4(filterDesc, dataType, |
| 139 | format, k, c, h, w); |
| 140 | #else |
| 141 | static_assert(1 != 1, "cuDNN version not supported"); |
| 142 | #endif |
| 143 | } |
| 144 | |
| 145 | cudnnStatus_t cudnnSetTensor4dDescriptor(cudnnTensorDescriptor_t tensorDesc, |
| 146 | cudnnTensorFormat_t format, |
no outgoing calls
no test coverage detected