Retry of D58015187 Move AsyncCompile to a different file (#842)

Summary:
X-link: facebookresearch/param#123

Pull Request resolved: #842

X-link: pytorch/ao#302

X-link: pytorch/pytorch#127691

This is a retry of https://github.com/pytorch/pytorch/pull/127545/files
and
D58015187, fixing the internal test that also imported codecache

Reviewed By: oulgen, msaroufim

Differential Revision: D58054611

fbshipit-source-id: 7a4d6602effa51c839ee8f650548e254c82a42a4

Author: jamesjwu

torchtnt/framework/callbacks/torch_compile.py

@@ -9,7 +9,7 @@
 import logging
 
 try:
-    from torch._inductor.codecache import shutdown_compile_workers
+    from torch._inductor.async_compile import shutdown_compile_workers
 except ImportError:
 
     def shutdown_compile_workers() -> None: