Skip to content

Commit

Permalink
Fix the AOTI compile failure with ARM CPU for Meta internal (pytorch#…
Browse files Browse the repository at this point in the history
…147204)

Summary: Fix the AOTI compile failure with ARM CPU for Meta internal

Differential Revision: D69642211

Pull Request resolved: pytorch#147204
Approved by: https://github.com/houseroad
  • Loading branch information
hl475 authored and pytorchmergebot committed Feb 18, 2025
1 parent 5d675de commit dd2a943
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions torch/_inductor/cpu_vec_isa.py
Original file line number Diff line number Diff line change
Expand Up @@ -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__
Expand All @@ -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__
Expand Down

0 comments on commit dd2a943

Please sign in to comment.