make torch compile callback OSS compatible

Summary: Touches few torch.compile calls in TorchTNT which rely on nightlies to be OSS friendly

Reviewed By: galrotem

Differential Revision: D56483268

fbshipit-source-id: d6d277e2e7eda895ca93047558b10e554effc96a

Author: JKSenthil

torchtnt/framework/callbacks/torch_compile.py

@@ -8,7 +8,17 @@
 
 import logging
 
-from torch._inductor.codecache import shutdown_compile_workers
+try:
+    from torch._inductor.codecache import shutdown_compile_workers
+except ImportError:
+
+    def shutdown_compile_workers() -> None:
+        logging.warning(
+            "shutdown_compile_workers is not available in your version of PyTorch. \
+            Please use nightly version to enable this feature."
+        )
+
+
 from torchtnt.framework.callback import Callback
 from torchtnt.framework.state import State
 from torchtnt.framework.unit import TTrainUnit