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 #ifndef NV_CUDNN_DISABLE_EXCEPTION 45 return filter(pred, plans);
std::function< bool(cudnn_frontend::ExecutionPlan const &plan)> Predicate
std::vector< cudnn_frontend::ExecutionPlan > 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.
auto filter(Predicate pred, executionPlans_t &plans) -> executionPlans_t
ExecutionPlanBuilder_v8 ExecutionPlanBuilder