Skip to content

Commit 8e9f4c9

Browse files
author
Copilot
committed
Move compiler info print
1 parent 36ba6b2 commit 8e9f4c9

1 file changed

Lines changed: 12 additions & 10 deletions

File tree

setup.py

Lines changed: 12 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -481,16 +481,6 @@ def _probe_isa() -> None:
481481
else:
482482
X86KernelBuild._isa_level = None
483483

484-
print("[X86 Build] compiler check")
485-
print("- Found compiler:", cxx)
486-
print(
487-
"- AVX512 support:",
488-
"Yes" if X86KernelBuild._isa_at_least("avx512") else "No",
489-
)
490-
print(
491-
"- AVX10.2 support:",
492-
"Yes" if X86KernelBuild._isa_at_least("avx10_2") else "No",
493-
)
494484
X86KernelBuild._isa_probed = True
495485

496486
@staticmethod
@@ -579,6 +569,18 @@ def precompile_isa_objects(build_temp: str, extensions: list) -> None:
579569
return
580570

581571
cxx = X86KernelBuild._cxx
572+
573+
print("[X86 Build] compiler check")
574+
print("- Found compiler:", cxx)
575+
print(
576+
"- AVX512 support:",
577+
"Yes" if X86KernelBuild._isa_at_least("avx512") else "No",
578+
)
579+
print(
580+
"- AVX10.2 support:",
581+
"Yes" if X86KernelBuild._isa_at_least("avx10_2") else "No",
582+
)
583+
582584
aten_kernels_dir = os.path.join("torchao", "csrc", "cpu", "aten_kernels")
583585
all_kernel_sources = glob.glob(os.path.join(aten_kernels_dir, "*_krnl.cpp"))
584586
include_flags = X86KernelBuild.get_include_flags()

0 commit comments

Comments
 (0)