test(flops_counter): Revert the introduction of incorrect comments

Author: mikequan0425

tests/utils/test_flops_counter.py

@@ -198,6 +198,13 @@ def __init__(self, config_dict):
         # seqlen_square_sum: 180355072 (calculated with sliding window logic)
         # attn flops: 12 * 180355072 * 256 * 16 = 8864812498944
         # total: 283517065887744 / 1e12 = 283.517065887744
+        #
+        # For batch [4096, 4096, 4096], tokens_sum = 12288:
+        # dense flops: 6 * 12772147200 * 12288 = 941664868761600
+        # seqlen_square_sum: 905969664 (calculated with sliding window logic)
+        # attn flops: 12 * 905969664 * 256 * 16 = 44530220924928
+        # total: 986195089686528 / 1e12 = 986.195089686528
+        "expected_flops_tuple": (283517065887744 / 1e12, 986195089686528 / 1e12),
     },
     "gpt_oss": {
         "config": {