1 files changed, 15 insertions, 3 deletions
diff --git a/src/graph/PassManager.cpp b/src/graph/PassManager.cpp
index 92860e2987..99d979842a 100644
--- a/src/graph/PassManager.cpp
+++ b/src/graph/PassManager.cpp
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2018 ARM Limited.
+ * Copyright (c) 2018-2019 ARM Limited.
  *
  * SPDX-License-Identifier: MIT
  *
@@ -70,7 +70,19 @@ void PassManager::run_all(Graph &g)
     }
 }
 
-void PassManager::run(Graph &g, size_t index)
+void PassManager::run_type(Graph &g, IGraphMutator::MutationType type)
+{
+    for(auto &pass : _passes)
+    {
+        if(pass && (pass->type() == type))
+        {
+            ARM_COMPUTE_LOG_GRAPH_INFO("Running mutating pass : " << pass->name() << std::endl);
+            pass->mutate(g);
+        }
+    }
+}
+
+void PassManager::run_index(Graph &g, size_t index)
 {
     if(index >= _passes.size())
     {
@@ -78,9 +90,9 @@ void PassManager::run(Graph &g, size_t index)
     }
 
     auto &pass = _passes.at(index);
-
     if(pass != nullptr)
     {
+        ARM_COMPUTE_LOG_GRAPH_INFO("Running mutating pass : " << pass->name() << std::endl);
         pass->mutate(g);
     }
 }