From cc4eb820189c4c11fcca181ac6b3991141b98fa4 Mon Sep 17 00:00:00 2001
From: Peter Gadfort <gadfort@zeroasic.com>
Date: Fri, 26 Jul 2024 15:20:28 -0400
Subject: [PATCH] ensure file io is set on task drivers

---
 switchboard/sc/morty/uniquify.py | 14 ++++++++++++--
 switchboard/sc/sed/remove.py     | 14 ++++++++++++--
 2 files changed, 24 insertions(+), 4 deletions(-)

diff --git a/switchboard/sc/morty/uniquify.py b/switchboard/sc/morty/uniquify.py
index 08b2dbbb..a752fd62 100644
--- a/switchboard/sc/morty/uniquify.py
+++ b/switchboard/sc/morty/uniquify.py
@@ -2,7 +2,7 @@
 # This code is licensed under Apache License 2.0 (see LICENSE for details)
 
 from .morty import setup as setup_tool
-from siliconcompiler.tools._common import get_tool_task
+from siliconcompiler.tools._common import get_tool_task, input_provides
 
 
 def setup(chip):
@@ -24,6 +24,13 @@ def setup(chip):
              'prefix to be added to the beginning of module names',
              field='help')
 
+    design = chip.top()
+    if f'{design}.v' in input_provides(chip, step, index):
+        chip.set('tool', tool, 'task', task, 'input', f'{design}.v', step=step, index=index)
+    else:
+        chip.set('tool', tool, 'task', task, 'input', f'{design}.sv', step=step, index=index)
+    chip.set('tool', tool, 'task', task, 'output', f'{design}.v', step=step, index=index)
+
 
 def runtime_options(chip):
     tool = 'morty'
@@ -48,10 +55,13 @@ def runtime_options(chip):
         else:
             raise ValueError('"suffix" does not have the expected format')
 
+    if f'{design}.v' in input_provides(chip, step, index):
+        infile = f'inputs/{design}.v'
+    else:
+        infile = f'inputs/{design}.sv'
     outfile = f'outputs/{design}.v'
     cmdlist += ['-o', outfile]
 
-    infile = f'inputs/{design}.v'
     cmdlist += [infile]
 
     return cmdlist
diff --git a/switchboard/sc/sed/remove.py b/switchboard/sc/sed/remove.py
index 8a18a45f..dfafa320 100644
--- a/switchboard/sc/sed/remove.py
+++ b/switchboard/sc/sed/remove.py
@@ -2,7 +2,7 @@
 # This code is licensed under Apache License 2.0 (see LICENSE for details)
 
 from .sed import setup as setup_tool
-from siliconcompiler.tools._common import get_tool_task
+from siliconcompiler.tools._common import get_tool_task, input_provides
 
 
 def setup(chip):
@@ -19,6 +19,13 @@ def setup(chip):
              'strings to remove from the Verilog source file',
              field='help')
 
+    design = chip.top()
+    if f'{design}.v' in input_provides(chip, step, index):
+        chip.set('tool', tool, 'task', task, 'input', f'{design}.v', step=step, index=index)
+    else:
+        chip.set('tool', tool, 'task', task, 'input', f'{design}.sv', step=step, index=index)
+    chip.set('tool', tool, 'task', task, 'output', f'{design}.v', step=step, index=index)
+
 
 def runtime_options(chip):
     tool = 'sed'
@@ -27,7 +34,10 @@ def runtime_options(chip):
     _, task = get_tool_task(chip, step, index)
     design = chip.top()
 
-    infile = f'inputs/{design}.v'
+    if f'{design}.v' in input_provides(chip, step, index):
+        infile = f'inputs/{design}.v'
+    else:
+        infile = f'inputs/{design}.sv'
     outfile = f'outputs/{design}.v'
 
     to_remove = chip.get('tool', tool, 'task', task, 'var', 'to_remove', step=step, index=index)