Enable arch on windows

1 files changed, 15 insertions, 5 deletions

diff --git a/build.bat b/build.bat
index 91cc4db..b082e03 100644
--- a/build.bat
+++ b/build.bat
@@ -9,15 +9,25 @@ if [%1]==[/d] (
 

 SET CC=clang

 SET CXX=clang++

-SET ARCH=PORTABLE

 SET THREADS=16

 SET SANITIZE=

 

-:: TODO depends on ARCH variable

-SET ARCHOPTS=

-:: TODO depends on SANITIZE variable

-SET DFLAGS=-g3 -DDEBUG

+SET ARCH=PORTABLE

+clang -march=native -dM -E - < NUL | findstr /C:"__AVX2__" >NUL 2>&1

+if %ERRORLEVEL% EQU 0 (

+    SET ARCH=AVX2

+    SET ARCHOPTS=-mavx2

+    goto :ArchDone

+)

+clang -march=native -dM -E - < NUL | findstr /C:"__ARM_NEON" >NUL 2>&1

+if %ERRORLEVEL% EQU 0 (

+    SET ARCH=NEON

+    goto :ArchDone

+)

+:ArchDone

+

 

+SET DFLAGS=-g3 -DDEBUG

 SET WARNINGS=-Wno-deprecated-declarations

 SET VARIABLES=-DTHREADS=%THREADS% -D%ARCH%

 SET OFLAGS=-O3