Merge branch 'main' into llu/ps_unroll_inner_outer

NVIDIA · Nov 1, 2024 · 63a38f1 · 63a38f1
2 parents 416bc31 + 4511aa3
commit 63a38f1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nvfuser/__init__.py b/nvfuser/__init__.py
@@ -177,15 +177,15 @@ def execute(
 
         results = None
         try:
+            if print_repro:
+                print(self.repro_script_for(inputs))
             results = self._execute(
                 inputs,
                 device=device,
                 override_user_schedule=override_user_schedule,
                 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))