#0: remove debug code

tenstorrent · Nov 21, 2024 · e9191c8 · e9191c8
1 parent 1223510
commit e9191c8
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/models/demos/llama3/tt/distributed_norm.py b/models/demos/llama3/tt/distributed_norm.py
@@ -11,10 +11,9 @@ def __init__(self, norm, args):
         self.norm = norm
         self.args = args
 
-    def forward(self, orig_x, mode):
+    def forward(self, x, mode):
         """Apply a norm, possibly gathering inputs if required."""
         input_mem_cfg = self.norm.sharded_output_config if mode == "decode" else ttnn.DRAM_MEMORY_CONFIG
-        x = orig_x
 
         # Distributed norm already performs a gather
         if self.args.is_multichip and not self.args.is_distributed_norm(mode):