Register data movement for ttnn.sum

tenstorrent · Nov 8, 2024 · 79ffb55 · 79ffb55
1 parent 0313f00
commit 79ffb55
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch_ttnn/passes/lowering/add_data_move_pass.py b/torch_ttnn/passes/lowering/add_data_move_pass.py
@@ -171,6 +171,7 @@ def is_tt_compute(node) -> bool:
             ttnn.tril,
             ttnn.arange,
             ttnn.zeros_like,
+            ttnn.sum,
             ttnn.mean,
             ttnn.global_avg_pool2d,
             ttnn.clip,