frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[BE] Standardize CPU capabilities name (#157809)

Browse Source 
It's weird to call default x86 CPU capability `NO AVX`, when in reality it's something different. Also it's a bit strange to have it assigned different names on different platforms

Fixes https://github.com/pytorch/pytorch/issues/157538

Pull Request resolved: https://github.com/pytorch/pytorch/pull/157809
Approved by: https://github.com/Skylion007
This commit is contained in:
Nikita Shulga
2025-07-08 23:06:04 +00:00
committed by PyTorch MergeBot
parent 179dcc10e4
commit c5b46b5408
1 changed files with 1 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
aten/src/ATen/Version.cpp
View File

@ -95,24 +95,18 @@ std::string get_cpu_capability() {
// environment variable
auto capability = native::get_cpu_capability();
switch (capability) {
#if defined(HAVE_VSX_CPU_DEFINITION)
case native::CPUCapability::DEFAULT:
return "DEFAULT";
#if defined(HAVE_VSX_CPU_DEFINITION)
case native::CPUCapability::VSX:
return "VSX";
#elif defined(HAVE_ZVECTOR_CPU_DEFINITION)
case native::CPUCapability::DEFAULT:
return "DEFAULT";
case native::CPUCapability::ZVECTOR:
return "Z VECTOR";
#elif defined(HAVE_SVE256_CPU_DEFINITION) && defined(HAVE_ARM_BF16_CPU_DEFINITION)
case native::CPUCapability::DEFAULT:
return "DEFAULT";
case native::CPUCapability::SVE256:
return "SVE256";
#else
case native::CPUCapability::DEFAULT:
return "NO AVX";
case native::CPUCapability::AVX2:
return "AVX2";
case native::CPUCapability::AVX512: