| 175 | } |
| 176 | |
| 177 | size_t ConvolutionImpl::AlgoFallback::get_workspace( |
| 178 | const NCBKernSizeParam& param) const { |
| 179 | MIDOUT_BEGIN(megdnn_fallback_conv, midout_iv("AlgoFallback::get_workspace"_hash)) { |
| 180 | auto FH = param.filter_meta.spatial[0], FW = param.filter_meta.spatial[1]; |
| 181 | size_t nr_threads = param.nr_threads; |
| 182 | if (param.filter_meta.should_flip) { |
| 183 | // need transpose filter |
| 184 | return WorkspaceBundle{nullptr, {FH * FW * sizeof(float)}} |
| 185 | .total_size_in_bytes() * |
| 186 | nr_threads; |
| 187 | } else { |
| 188 | return 0; |
| 189 | } |
| 190 | } |
| 191 | MIDOUT_END(); |
| 192 | return 0; |
| 193 | } |
| 194 | |
| 195 | SmallVector<ConvolutionImpl::NCBKern> ConvolutionImpl::AlgoFallback::dispatch_kern( |
| 196 | const NCBKernSizeParam& param) const { |
no test coverage detected