8000
We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 95e470b commit 9cb6f3eCopy full SHA for 9cb6f3e
torch/_inductor/codecache.py
@@ -1490,11 +1490,11 @@ def valid_vec_isa_list() -> List[VecISA]:
1490
if re.search(r"[\^ ]+vxe[\$ ]+", group):
1491
isa_list.append(VecZVECTOR())
1492
break
1493
- elif arch == "aarch64":
+ elif platform.machine() == "aarch64":
1494
isa_list.append(VecNEON())
1495
- elif arch in ["x86_64", "AMD64"]:
+ elif platform.machine() in ["x86_64", "AMD64"]:
1496
"""
1497
- arch value is x86_64 on Linux, and the value is AMD64 on Windows.
+ platform.machine() value is x86_64 on Linux, and the value is AMD64 on Windows.
1498
1499
_cpu_supported_x86_isa = x86_isa_checker()
1500
for isa in supported_vec_isa_list:
0 commit comments