Make CompileEventLogger more defensive w.r.t to AOTAutogradCache and FXGraphCache (#150423)

Summary:
This PR makes it so that we don't crash due to logging if we invoke AOTAutogradCache/FXGraphCache without using dynamo. This is preparation for supporting certain VLLM use cases where they store graph modules and have special handling in conjunection with the caches.

X-link: pytorch/pytorch#150423
Approved by: https://github.com/oulgen

Reviewed By: atalman

Differential Revision: D72453905

fbshipit-source-id: 77da32a50b8020e03e97d3098b9c6b7d2546f13f

Author: jamesjwu

userbenchmark/dynamo/dynamobench/_dynamo/utils.py

@@ -579,12 +579,27 @@ def instant(
     @staticmethod
     def try_add_pt2_compile(event_name: str, **metadata: object):
         """
-        Adds to an existing pt2_compile event, but silently returns if the event doesn't exist.
+        Adds to an existing pt2_compile event, but silently returns if the event doesn't exist
+        or ChromiumEventLogger is not initialized.
         This function is syntactic sugar for chromium_event_logger().try_add_event_data.
         """
+        if CHROMIUM_EVENT_LOG is None:
+            return
         chromium_log = get_chromium_event_logger()
         chromium_log.try_add_event_data(event_name, **metadata)
 
+    @staticmethod
+    def try_(method_fn, *args, **kwargs):
+        """
+        Special function that quietly runs a given method, returning if CHROMIUM_EVENT_LOG is None or metrics context is not set
+        """
+        if CHROMIUM_EVENT_LOG is None:
+            return
+        metrics_context = get_metrics_context()
+        if not metrics_context.in_progress():
+            return
+        method_fn(*args, **kwargs)
+
 
 @contextmanager
 def dynamo_timed(