[AOTI] config target platform (#126306)
Test Plan: AOTI compile stories15M for Android
Differential Revision: D57392830
Pull Request resolved: https://github.com/pytorch/pytorch/pull/126306
Approved by: https://github.com/desertfire
diff --git a/torch/_inductor/codecache.py b/torch/_inductor/codecache.py
index 0850b4a..70b4671 100644
--- a/torch/_inductor/codecache.py
+++ b/torch/_inductor/codecache.py
@@ -1961,6 +1961,14 @@
return consts_o
def _compile_consts_darwin(consts: bytes) -> str:
+ if config.aot_inductor.debug_dump_consts_bin:
+ _, _binary_constants_path = write(
+ consts,
+ "bin",
+ specified_dir=specified_output_path,
+ )
+ log.debug("binary constants path: %s", _binary_constants_path)
+
is_large_consts = len(consts) > 1024
consts_asm = "\t.section\t__DATA,__data\n"
consts_asm += "\t.globl\t__binary_constants_bin_start\n"
diff --git a/torch/_inductor/config.py b/torch/_inductor/config.py
index 7f14dc6..db8a6d9 100644
--- a/torch/_inductor/config.py
+++ b/torch/_inductor/config.py
@@ -727,6 +727,10 @@
debug_compile = os.environ.get("AOT_INDUCTOR_DEBUG_COMPILE", "0") == "1"
+ debug_dump_consts_bin: bool = (
+ os.environ.get("AOT_INDUCTOR_DEBUG_DUMP_CONSTS_BIN", "0") == "1"
+ )
+
# Serialized tree spec for flattening inputs
serialized_in_spec = ""
