diff --git a/thunder/transforms/extraction_only_prologue_transform.py b/thunder/transforms/extraction_only_prologue_transform.py index 80a069fb7d..7f335ef33c 100644 --- a/thunder/transforms/extraction_only_prologue_transform.py +++ b/thunder/transforms/extraction_only_prologue_transform.py @@ -22,5 +22,5 @@ def transform_traces_pre_prologue(self, prologue_trace, computation_trace, epilo new_prologue_trace.bound_symbols = new_bsyms - new_prologue_trace.set_provenance("Extraction only pass") + new_prologue_trace.set_provenance("Extraction only prologue pass") return new_prologue_trace, computation_trace, epilogue_trace