Skip to content

Commit

Permalink
Update codecache.py
Browse files Browse the repository at this point in the history
  • Loading branch information
malfet committed Jun 19, 2024
1 parent 94d8aa0 commit c5e565d
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions torch/_inductor/codecache.py
Original file line number Diff line number Diff line change
Expand Up @@ -1497,9 +1497,8 @@ def _check_and_append_supported_isa(
def valid_vec_isa_list() -> List[VecISA]:
isa_list: List[VecISA] = []
if sys.platform == "darwin" and platform.processor() == "arm":
return isa_list.append(VecNEON())
isa_list.append(VecNEON())

cur_os = sys.platform
if sys.platform not in ["linux", "win32"]:
return isa_list

Expand Down

0 comments on commit c5e565d

Please sign in to comment.