diff --git a/lighthouse/schedule/builders.py b/lighthouse/schedule/builders.py index 70d8351..787e327 100644 --- a/lighthouse/schedule/builders.py +++ b/lighthouse/schedule/builders.py @@ -31,7 +31,7 @@ def create_named_sequence( Args: sym_name: Sequence name input_types: Input types (default: a single arg) - result_types: Result types (default: a single return) + result_types: Result types (default: no returns) is_readonly: Mark inputs as readonly Returns: @@ -40,8 +40,6 @@ def create_named_sequence( if not input_types: input_types = [transform.any_op_t()] - if not result_types: - result_types = [transform.any_op_t()] arg_attrs = [{"transform.consumed": ir.UnitAttr.get()}] if is_readonly: diff --git a/lighthouse/workload/runner.py b/lighthouse/workload/runner.py index c3d3017..673abc6 100644 --- a/lighthouse/workload/runner.py +++ b/lighthouse/workload/runner.py @@ -70,7 +70,10 @@ def execute( def get_bench_wrapper_schedule(workload: Workload): - with schedule_boilerplate() as (schedule, named_seq): + with schedule_boilerplate(result_types=[transform.any_op_t()]) as ( + schedule, + named_seq, + ): named_func = structured.structured_match( transform.AnyOpType.get(), target=named_seq.bodyTarget,