Surround aoti_torch_abi_version with not C10_MOBILE too

2025-11-15 06:48:48 +08:00 · 2025-11-03 11:24:30 -08:00
1 changed files with 3 additions and 0 deletions
--- a/torch/csrc/inductor/aoti_torch/c/shim.h
+++ b/torch/csrc/inductor/aoti_torch/c/shim.h
@ -97,7 +97,10 @@ AOTI_TORCH_EXPORT int32_t aoti_torch_memory_format_channels_last_3d();
 AOTI_TORCH_EXPORT int32_t aoti_torch_memory_format_preserve_format();

 // Get TORCH_ABI_VERSION of the built libtorch.so
+// version.h is not included for C10_MOBILE builds
+#ifndef C10_MOBILE
 AOTI_TORCH_EXPORT uint64_t aoti_torch_abi_version();
+#endif // C10_MOBILE

 // Functions for converting a single-element tensor to a scalar value
 AOTI_TORCH_EXPORT AOTITorchError