Add options for SIMD extensions

This commit is contained in:
Chris Simons
2023-06-16 09:09:08 -07:00
parent 9dad68426c
commit 833e893a94

View File

@@ -45,7 +45,7 @@ BINARY_ALIAS= python3=${PYTHON_CMD} python=${PYTHON_CMD}
SOVERSION= ${DISTVERSION}
PLIST_SUB= SOVERSION=${SOVERSION}
OPTIONS_DEFINE= CUDA OPENCL_SYCL ROCM XLA
OPTIONS_DEFINE= CUDA OPENCL_SYCL ROCM XLA SSE3 SSE41 SSE42
OPTIONS_DEFAULT= AVX
OPTIONS_SINGLE= CPUFEATURE
@@ -80,6 +80,15 @@ XLA_DESC= Enable Accelerated Linear Algebra (XLA)
XLA_VARS= TF_ENABLE_XLA=1
XLA_VARS_OFF= TF_ENABLE_XLA=0
SSE3_DESC= Enable Streaming SIMD Extensions 3 (Prescott)
SSE3_VARS= BAZEL_ARGS+="--copt=-msse3"
SSE41_DESC= Enable Streaming SIMD Extensions 4.1 (Penryn)
SSE41_VARS= BAZEL_ARGS+="--copt=-msse4.1"
SSE42_DESC= Enable Streaming SIMD Extensions 4.2 (Nehalem)
SSE41_VARS= BAZEL_ARGS+="--copt=-msse4.2"
BAZEL_ARGS+= --action_env=PATH=${PATH} \
--color=no \
--discard_analysis_cache \