8000 Fix merge mistakes · pytorch/pytorch@9cb6f3e · GitHub
[go: up one dir, main page]

Skip to content

Commit 9cb6f3e

Browse files
committed
Fix merge mistakes
1 parent 95e470b commit 9cb6f3e

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

torch/_inductor/codecache.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1490,11 +1490,11 @@ def valid_vec_isa_list() -> List[VecISA]:
14901490
if re.search(r"[\^ ]+vxe[\$ ]+", group):
14911491
isa_list.append(VecZVECTOR())
14921492
break
1493-
elif arch == "aarch64":
1493+
elif platform.machine() == "aarch64":
14941494
isa_list.append(VecNEON())
1495-
elif arch in ["x86_64", "AMD64"]:
1495+
elif platform.machine() in ["x86_64", "AMD64"]:
14961496
"""
1497-
arch value is x86_64 on Linux, and the value is AMD64 on Windows.
1497+
platform.machine() value is x86_64 on Linux, and the value is AMD64 on Windows.
14981498
"""
14991499
_cpu_supported_x86_isa = x86_isa_checker()
15001500
for isa in supported_vec_isa_list:

0 commit comments

Comments
 (0)
0