More accurate AVX detection (#4941)

Summary:
X-link: facebookresearch/FBGEMM#1968

This PR separates MSVC AVX options from GNU ones. The purpose is to avoid some versions of MSVC succeeding with a `unknown option` warning. I observed this invalid behaviour on PyTorch CI.

Pull Request resolved: #4941

Reviewed By: cthi

Differential Revision: D83512683

Pulled By: q10

fbshipit-source-id: 92114b13b0582e278a6533e19d84e9586f253867

Author: cyyever

cmake/modules/FindAVX.cmake

@@ -101,9 +101,15 @@ ENDMACRO()
 get_property(languages GLOBAL PROPERTY ENABLED_LANGUAGES)
 foreach(lang C;CXX)
   if(lang IN_LIST languages)
-    CHECK_SSE(${lang} "AVX" "-mavx;/arch:AVX")
-    CHECK_SSE(${lang} "AVX2" "-mavx2 -mfma -mf16c;/arch:AVX2")
-    CHECK_SSE(${lang} "AVX512" "-mavx512f -mavx512dq -mavx512vl -mavx512bw -mfma;/arch:AVX512")
+    if(MSVC)
+      CHECK_SSE(${lang} "AVX" "/arch:AVX;")
+      CHECK_SSE(${lang} "AVX2" "/arch:AVX2;")
+      CHECK_SSE(${lang} "AVX512" "/arch:AVX512;")
+    else()
+      CHECK_SSE(${lang} "AVX" "-mavx;")
+      CHECK_SSE(${lang} "AVX2" "-mavx2 -mfma -mf16c;")
+      CHECK_SSE(${lang} "AVX512" "-mavx512f -mavx512dq -mavx512vl -mavx512bw -mfma;")
+    endif()
   endif()
 endforeach()