Revert "Add resharding to post all gather layernorm/ rms norm op (#17156

) (#17304)

.github/workflows/tg-unit-tests-impl.yaml

@@ -45,7 +45,6 @@ jobs:
           { name: "TG Llama3-small unit tests", arch: wormhole_b0, model: llama3-small, timeout: 45, owner_id: U06F3ER8X9A}, # Stuti Raizada
           { name: "TG Llama3-70b unit tests", arch: wormhole_b0, model: llama3-70b, timeout: 45, owner_id: U06F3ER8X9A}, # Stuti Raizada
           { name: "TG DRAM Prefetcher unit tests", arch: wormhole_b0, model: prefetcher, timeout: 30, owner_id: U071CKL4AFK}, # Ammar Vora, Yu Gao
-          { name: "TG distributed ops tests", arch: wormhole_b0, model: distributed-ops, timeout: 15, owner_id: U044T8U8DEF},  # Johanna Rock
         ]
     name: ${{ matrix.test-group.name }}
     env:

tests/scripts/tg/run_tg_unit_tests.sh

@@ -66,24 +66,6 @@ run_tg_llama3.1-70b_tests() {
   fi
 }
 
-run_tg_distributed_op_tests() {
-  # Record the start time
-  fail=0
-  start_time=$(date +%s)
-
-  echo "LOG_METAL: Running run_tg_distributed_op_tests"
-
-  pytest tests/ttnn/distributed/test_distributed_layernorm_TG.py ; fail+=$?
-
-  # Record the end time
-  end_time=$(date +%s)
-  duration=$((end_time - start_time))
-  echo "LOG_METAL: run_tg_distributed_op_tests $duration seconds to complete"
-  if [[ $fail -ne 0 ]]; then
-    exit 1
-  fi
-}
-
 run_tg_prefetcher_tests() {
   # Record the start time
   fail=0
@@ -127,9 +109,6 @@ run_tg_tests() {
   elif [[ "$1" == "prefetcher" ]]; then
     run_tg_prefetcher_tests
 
-  elif [[ "$1" == "distributed-ops" ]]; then
-    run_tg_distributed_op_tests
-
   else
     echo "LOG_METAL: Unknown model type: $1"
     return 1