| 60 | } |
| 61 | |
| 62 | bool Convolution3DForwardImpl::AlgoGroupConvGeneral::is_available( |
| 63 | const SizeArgs& args) const { |
| 64 | if (args.filter_meta.group <= 1) |
| 65 | return false; |
| 66 | if (args.filter_meta.format != Param::Format::NCDHW && |
| 67 | args.filter_meta.format != Param::Format::NDHWC) { |
| 68 | return false; |
| 69 | } |
| 70 | |
| 71 | auto config = prepare_sub_opr(args); |
| 72 | |
| 73 | return has_available_algo<Convolution3DForwardImpl>( |
| 74 | static_cast<Convolution3DForwardImpl*>(config.second.get()), |
| 75 | config.first[0], config.first[1], config.first[2]); |
| 76 | } |
| 77 | |
| 78 | WorkspaceBundle Convolution3DForwardImpl::AlgoGroupConvGeneral::get_workspace_bundle( |
| 79 | void* ptr, const SizeArgs& args) const { |
nothing calls this directly
no test coverage detected