From cb630e8b713e75ba17ed9d9163163abc1becb29a Mon Sep 17 00:00:00 2001 From: Kevin Stephano Date: Tue, 29 Oct 2024 12:02:20 -0700 Subject: [PATCH] Change code point to print FusionDefinition repro. --- nvfuser/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nvfuser/__init__.py b/nvfuser/__init__.py index 4b4f25b9d66..0201ab51c86 100644 --- a/nvfuser/__init__.py +++ b/nvfuser/__init__.py @@ -178,6 +178,8 @@ def execute( results = None try: + if print_repro: + print(self.repro_script_for(inputs)) results = self._execute( inputs, device=device, @@ -185,8 +187,6 @@ def execute( capture_debug_output=capture_debug_output, profile=profile, ) - if print_repro: - print(self.repro_script_for(inputs)) return results except Exception as err: logger.exception(self._repro_error_str("executing", inputs))