From 71b425c1de7b3e1991650a94bdbd478b0b7e52ac Mon Sep 17 00:00:00 2001
From: JanLJL <jan.laukemann@fau.de>
Date: Tue, 14 Mar 2023 17:00:49 +0100
Subject: [PATCH] added support for optional condition flag dependency analysis

---
 osaca/osaca.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/osaca/osaca.py b/osaca/osaca.py
index 8ef44d3..f5f9bcf 100755
--- a/osaca/osaca.py
+++ b/osaca/osaca.py
@@ -170,6 +170,14 @@ def create_parser(parser=None):
         " its analysis with the dependency paths found up to this point. Defaults to 10."
         " Set to -1 for no timeout.",
     )
+    parser.add_argument(
+        "--consider-flag-deps",
+        "-f",
+        dest="consider_flag_deps",
+        action="store_true",
+        default=False,
+        help="Consider flag dependencies (carry, zero, ...)",
+    )
     parser.add_argument(
         "--verbose", "-v", action="count", default=0, help="Increases verbosity level."
     )
@@ -333,7 +341,9 @@ def inspect(args, output_file=sys.stdout):
         semantics.assign_optimal_throughput(kernel)
 
     # Create DiGrahps
-    kernel_graph = KernelDG(kernel, parser, machine_model, semantics, args.lcd_timeout)
+    kernel_graph = KernelDG(
+        kernel, parser, machine_model, semantics, args.lcd_timeout, args.consider_flag_deps
+    )
     if args.dotpath is not None:
         kernel_graph.export_graph(args.dotpath if args.dotpath != "." else None)
     # Print analysis