| 351 | } |
| 352 | |
| 353 | ConvolutionImpl::Algorithm* ConvolutionImpl::get_algorithm_from_desc( |
| 354 | const AlgorithmDesc& desc) { |
| 355 | if (!desc.valid()) { |
| 356 | return nullptr; |
| 357 | } else { |
| 358 | switch (desc.handle_type) { |
| 359 | case Handle::HandleType::FALLBACK: { |
| 360 | const auto& map = algo_pack().all_algos_map(); |
| 361 | megdnn_assert(map.find(desc) != map.end()); |
| 362 | return map.at(desc); |
| 363 | } |
| 364 | case Handle::HandleType::NAIVE: { |
| 365 | auto algo = static_cast<naive::HandleImpl*>(handle()) |
| 366 | ->default_conv_fwd_algo(); |
| 367 | megdnn_assert(algo->info().desc == desc); |
| 368 | return algo; |
| 369 | } |
| 370 | default: |
| 371 | megdnn_throw("Unknown handle type"); |
| 372 | return nullptr; |
| 373 | } |
| 374 | } |
| 375 | } |
| 376 | |
| 377 | ConvolutionImpl::Algorithm* ConvolutionImpl::get_algorithm( |
| 378 | const NCBKernSizeParam& param, size_t workspace_size) { |