File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff 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 ()
You can’t perform that action at this time.
0 commit comments