Fix the AOTI compile failure with ARM CPU for Meta internal (#147204)

hl475 · pytorchmergebot · commit dd2a943e145a · 2025-02-18T17:54:34.000Z
Summary: Fix the AOTI compile failure with ARM CPU for Meta internal Differential Revision: D69642211 Pull Request resolved: #147204 Approved by: https://github.com/houseroad
diff --git a/torch/_inductor/cpu_vec_isa.py b/torch/_inductor/cpu_vec_isa.py
@@ -160,6 +160,8 @@ class VecNEON(VecISA):
     _dtype_nelements = {torch.float: 4, torch.bfloat16: 8, torch.float16: 8}
 
     def __str__(self) -> str:
+        if config.is_fbcode():
+            return "neon"
         return "asimd"  # detects the presence of advanced SIMD on armv8-a kernels
 
     __hash__: Callable[[VecISA], Any] = VecISA.__hash__
@@ -178,6 +180,8 @@ class VecSVE256(VecISA):
     _dtype_nelements = {torch.float: 8, torch.bfloat16: 16, torch.float16: 16}
 
     def __str__(self) -> str:
+        if config.is_fbcode():
+            return "neon"
         return "asimd"
 
     __hash__: Callable[[VecISA], Any] = VecISA.__hash__
