diff options
Diffstat (limited to 'src/graph/Utils.cpp')
-rw-r--r-- | src/graph/Utils.cpp | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/src/graph/Utils.cpp b/src/graph/Utils.cpp index 4c34dd85a5..3bf0caca7e 100644 --- a/src/graph/Utils.cpp +++ b/src/graph/Utils.cpp @@ -74,13 +74,17 @@ void force_target_to_graph(Graph &g, Target target) } } -PassManager create_default_pass_manager(Target target) +PassManager create_default_pass_manager(Target target, const GraphConfig &cfg) { PassManager pm; const bool is_target_gc = target == Target::GC; // Passes that mutate graph IR + if(cfg.convert_to_uint8) + { + pm.append(support::cpp14::make_unique<SyntheticDataTypeMutator>(), !is_target_gc); + } pm.append(support::cpp14::make_unique<NodeFusionMutator>(), !is_target_gc); pm.append(support::cpp14::make_unique<GroupedConvolutionMutator>()); pm.append(support::cpp14::make_unique<InPlaceOperationMutator>(), !is_target_gc); |