#1680: Added explicit check for checking golden before getting output… (

#1693) … tensor from device
tenstorrent · Jan 7, 2025 · 028f711 · 028f711
1 parent dca29f4
commit 028f711
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/runtime/tools/python/ttrt/common/callback.py b/runtime/tools/python/ttrt/common/callback.py
@@ -162,12 +162,13 @@ def golden(callback_runtime_config, binary, program_context, op_context):
     loc = ttrt.runtime.get_op_loc_info(op_context)
 
     op_golden_tensor = binary.get_debug_info_golden(loc)
-    op_output_tensor = ttrt.runtime.get_op_output_tensor(op_context, program_context)
 
     if op_golden_tensor is None:
         logging.debug("Golden tensor is None - skipping golden comparison")
         return
 
+    op_output_tensor = ttrt.runtime.get_op_output_tensor(op_context, program_context)
+
     if len(op_output_tensor) == 0:
         logging.debug("Output tensor is empty - skipping golden comparison")
         return