"docs/source/features/quantization/fp8_e4m3_kvcache.md" did not exist on "94d545a1a18e20ea8763a6760194589b8a3c9065"
Unverified Commit 2ed4722e authored by Zhengxu Chen's avatar Zhengxu Chen Committed by GitHub
Browse files

[compile] Reduce log spam from compile. (#36044)


Signed-off-by: default avatarzhxchen17 <zhxchen17@fb.com>
parent a3299c3d
......@@ -582,7 +582,6 @@ def _support_torch_compile(
self.aot_compiled_fn and self._aot_compilation_path and self._aot_cache_dir
)
logger.info("saving AOT compiled function to %s", self._aot_compilation_path)
try:
os.makedirs(self._aot_cache_dir, exist_ok=True)
# File saving should be atomic, so we will save to a temporary location
......@@ -590,7 +589,11 @@ def _support_torch_compile(
tmp_file = f"{self._aot_compilation_path}.{os.getpid()}.tmp"
self.aot_compiled_fn.save_compiled_function(tmp_file)
os.replace(tmp_file, self._aot_compilation_path)
logger.info("saved AOT compiled function to %s", self._aot_compilation_path)
logger.info_once(
"saved AOT compiled function to %s",
self._aot_compilation_path,
scope="local",
)
except Exception as e:
logger.warning(
"unable to save AOT compiled function to %s: %s",
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment