34 for (
auto& engine_config : generate_engine_config(opGraph)) {
35 #ifndef NV_CUDNN_DISABLE_EXCEPTION 38 plans.push_back(
ExecutionPlanBuilder().setHandle(handle).setEngineConfig(engine_config, opGraph.getTag()).build());
39 getLogger() <<
"[cudnn_frontend] Added plan " << plans.back().getTag() <<
" " <<
to_string(plans.back().get_status()) << std::endl;
40 #ifndef NV_CUDNN_DISABLE_EXCEPTION 54 return filter(pred, plans);
ConditionalStreamer & getLogger()
std::function< bool(cudnn_frontend::ExecutionPlan const &plan)> Predicate
static auto filter(Predicate pred, executionPlans_t &plans) -> executionPlans_t
auto cudnnGetPlan(cudnnHandle_t handle, cudnn_frontend::OperationGraph &opGraph, Predicate pred) -> executionPlans_t
Returns the concatenated plan in the order of heuristic results.
std::vector< cudnn_frontend::ExecutionPlan > executionPlans_t
Variety of renames.
ExecutionPlanBuilder_v8 ExecutionPlanBuilder
static std::string to_string(cudnnDataType_t type)