File tree Expand file tree Collapse file tree 2 files changed +9
-0
lines changed Expand file tree Collapse file tree 2 files changed +9
-0
lines changed Original file line number Diff line number Diff line change @@ -64,6 +64,11 @@ include(GNUInstallDirs)
6464# Check AVX
6565set (CHECK_AVX_COMPILE ON )
6666include (${MODULE_PATH} /FindAVX.cmake)
67+ # For some reason, AVX flag values end up being packed into a single quoted
68+ # string when FBGEMM is built as part of PyTorch build, so we need to separate
69+ # the flags here back into list-string format
70+ separate_arguments (CXX_AVX2_FLAGS)
71+ separate_arguments (CXX_AVX512_FLAGS)
6772
6873# Load Python
6974find_package (Python)
Original file line number Diff line number Diff line change @@ -106,3 +106,7 @@ foreach(lang C;CXX)
106106 CHECK_SSE(${lang} "AVX512" "-mavx512f -mavx512dq -mavx512vl -mavx512bw -mfma;/arch:AVX512" )
107107 endif ()
108108endforeach ()
109+
110+ MESSAGE (STATUS "CXX_AVX_FLAGS: ${CXX_AVX_FLAGS} " )
111+ MESSAGE (STATUS "CXX_AVX2_FLAGS: ${CXX_AVX2_FLAGS} " )
112+ MESSAGE (STATUS "CXX_AVX512_FLAGS: ${CXX_AVX512_FLAGS} " )
You can’t perform that action at this time.
0 commit comments