From ab725abcb4ddec8e5239337b9313f7718eb0a4e5 Mon Sep 17 00:00:00 2001
From: Naif Tarafdar <135640067+ntarafdar@users.noreply.github.com>
Date: Wed, 9 Oct 2024 23:05:09 -0700
Subject: [PATCH] Attach Golden Function (already implemented ) For Reshape
 (accidentally unattached in previous PR) (#13614)

#13613: add golden function for reshape

Co-authored-by: Saad Jameel <163029024+sjameelTT@users.noreply.github.com>
---
 ttnn/ttnn/operations/core.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/ttnn/ttnn/operations/core.py b/ttnn/ttnn/operations/core.py
index 8a1e2beb603..45e024ac82f 100644
--- 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)