Add coloring to fx graph print out (#128476)

Summary:
Note: Won't land immediately, at least I'll need to add a color option to the field. But curious if any tests fail.

Old:
<img width="1294" alt="image" src="https://github.com/pytorch/pytorch/assets/6355099/c3a750ed-5e54-4621-b2e4-be5481be15b6">

New:
<img width="1303" alt="image" src="https://github.com/pytorch/pytorch/assets/6355099/3a1f1adc-6f3a-413e-8b87-ee53da9bf4ed">

X-link: pytorch/pytorch#128476
Approved by: https://github.com/ezyang

Reviewed By: clee2000

Differential Revision: D58628122

Pulled By: Chillee

fbshipit-source-id: f506d0aeeb92b00f5a96b58bb87b1d1bce2c123a

userbenchmark/dynamo/dynamobench/_dynamo/utils.py

@@ -2767,3 +2767,9 @@ def warn_once(msg, stacklevel=1):
  return
  warn_once_cache.add(msg)
  warnings.warn(msg, stacklevel=stacklevel + 1)
+
+
+def strip_color_from_string(text):
+ # This regular expression matches ANSI escape codes
+ ansi_escape = re.compile(r"\x1B[@-_][0-?]*[ -/]*[@-~]")
+ return ansi_escape.sub("", text)