Attach Golden Function (already implemented ) For Reshape (accidental…

…ly unattached in previous PR) (#13614) #13613: add golden function for reshape Co-authored-by: Saad Jameel <[email protected]>
tenstorrent · Oct 10, 2024 · ab725ab · ab725ab
1 parent 04f71b4
commit ab725ab
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ttnn/ttnn/operations/core.py b/ttnn/ttnn/operations/core.py
@@ -143,6 +143,7 @@ def _postprocess_golden_function_outputs(output, args, kwargs):
     ttnn.Shape([32, 64])
 
 """
+ttnn.attach_golden_function(ttnn.reshape, golden_function=_golden_function)
 
 # TODO(arakhmati): remove this once underlying C++ code can handle non-4D shapes
 ttnn.register_python_operation(name="ttnn.unsqueeze_to_4D")(ttnn._ttnn.operations.core.unsqueeze_to_4D)