Update setup.py

This commit is contained in:
Luke Southam 2022-12-14 16:42:05 +00:00 committed by GitHub
parent bf8d7bb3cc
commit a021ae79da
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -3,12 +3,12 @@ from Cython.Build import cythonize
import numpy, os, sys
if sys.platform == 'darwin':
os.environ['CFLAGS'] = '-DGGML_USE_ACCELERATE -O3 -I./whisper.cpp/'
os.environ['CXXFLAGS'] = '-DGGML_USE_ACCELERATE -O3 -I./whisper.cpp/'
os.environ['LDFLAGS'] = '-framework Accelerate -I./whisper.cpp/'
os.environ['CFLAGS'] = '-DGGML_USE_ACCELERATE -O3'
os.environ['CXXFLAGS'] = '-DGGML_USE_ACCELERATE -O3'
os.environ['LDFLAGS'] = '-framework Accelerate'
else:
os.environ['CFLAGS'] = '-mavx -mavx2 -mfma -mf16c -O3 -I./whisper.cpp/'
os.environ['CXXFLAGS'] = '-mavx -mavx2 -mfma -mf16c -O3 -I./whisper.cpp/'
os.environ['CFLAGS'] = '-mavx -mavx2 -mfma -mf16c -O3'
os.environ['CXXFLAGS'] = '-mavx -mavx2 -mfma -mf16c -O3'
setup(