File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -24,7 +24,7 @@ def patch_module_pom(path: Path) -> None:
2424def patch_build_sh (path : Path ) -> None :
2525 text = path .read_text (encoding = "utf-8" )
2626 old = """# Allows us to use ccache with Bazel on Mac\n export BAZEL_USE_CPP_ONLY_TOOLCHAIN=1\n \n export BAZEL_VC=\" ${VCINSTALLDIR:-}\" \n if [[ -d $BAZEL_VC ]]; then\n # Work around compiler issues on Windows documented mainly in configure.py but also elsewhere\n export BUILD_FLAGS=\" --copt=//arch:AVX `#--copt=//arch:AVX2` --copt=-DWIN32_LEAN_AND_MEAN --host_copt=-DWIN32_LEAN_AND_MEAN --copt=-DNOGDI --host_copt=-DNOGDI --copt=-D_USE_MATH_DEFINES --host_copt=-D_USE_MATH_DEFINES --define=override_eigen_strong_inline=true\" \n # https://software.intel.com/en-us/articles/intel-optimization-for-tensorflow-installation-guide#wind_B_S\n export PATH=$PATH:$(pwd)/bazel-tensorflow-core-api/external/mkl_windows/lib/\n export PYTHON_BIN_PATH=$(which python.exe)\n else\n export BUILD_FLAGS=\" --copt=-msse4.1 --copt=-msse4.2 --copt=-mavx `#--copt=-mavx2 --copt=-mfma` --cxxopt=-std=c++14 --host_cxxopt=-std=c++14 --linkopt=-lstdc++ --host_linkopt=-lstdc++\" \n export PYTHON_BIN_PATH=$(which python3)\n fi\n """
27- new = """# Allows us to use ccache with Bazel on Mac, but Android needs Bazel's Android crosstool.\n if [[ \" ${PLATFORM:-}\" != \" android-arm64\" ]]; then\n export BAZEL_USE_CPP_ONLY_TOOLCHAIN=1\n fi\n \n export BAZEL_VC=\" ${VCINSTALLDIR:-}\" \n if [[ \" ${PLATFORM:-}\" == \" android-arm64\" ]]; then\n export BUILD_FLAGS=\" --crosstool_top=@androidndk//:default_crosstool --cpu=arm64-v8a --fat_apk_cpu=arm64-v8a --host_crosstool_top=@bazel_tools//tools/cpp:toolchain --copt=-DANDROID --cxxopt=-std=c++14 --host_cxxopt=-std=c++14 --cxxopt=-include --cxxopt=cstdint --host_cxxopt=-include --host_cxxopt=cstdint --copt=-Wno-error=array-parameter --host_copt=-Wno-error=array-parameter --copt=-Wno-error=array-bounds --host_copt=-Wno-error=array-bounds\" \n export PYTHON_BIN_PATH=$(which python3)\n elif [[ -d $BAZEL_VC ]]; then\n # Work around compiler issues on Windows documented mainly in configure.py but also elsewhere\n export BUILD_FLAGS=\" --copt=//arch:AVX `#--copt=//arch:AVX2` --copt=-DWIN32_LEAN_AND_MEAN --host_copt=-DWIN32_LEAN_AND_MEAN --copt=-DNOGDI --host_copt=-DNOGDI --copt=-D_USE_MATH_DEFINES --host_copt=-D_USE_MATH_DEFINES --define=override_eigen_strong_inline=true\" \n # https://software.intel.com/en-us/articles/intel-optimization-for-tensorflow-installation-guide#wind_B_S\n export PATH=$PATH:$(pwd)/bazel-tensorflow-core-api/external/mkl_windows/lib/\n export PYTHON_BIN_PATH=$(which python.exe)\n else\n export BUILD_FLAGS=\" --copt=-msse4.1 --copt=-msse4.2 --copt=-mavx `#--copt=-mavx2 --copt=-mfma` --cxxopt=-std=c++14 --host_cxxopt=-std=c++14 --linkopt=-lstdc++ --host_linkopt=-lstdc++\" \n export PYTHON_BIN_PATH=$(which python3)\n fi\n """
27+ new = """# Allows us to use ccache with Bazel on Mac, but Android needs Bazel's Android crosstool.\n if [[ \" ${PLATFORM:-}\" != \" android-arm64\" ]]; then\n export BAZEL_USE_CPP_ONLY_TOOLCHAIN=1\n fi\n \n export BAZEL_VC=\" ${VCINSTALLDIR:-}\" \n if [[ \" ${PLATFORM:-}\" == \" android-arm64\" ]]; then\n export TF_ANDROID_COMPAT_LIB_DIR=\" $(pwd)/android-compat-libs\" \n mkdir -p \" ${TF_ANDROID_COMPAT_LIB_DIR}\" \n printf 'INPUT(-lc)\\ n' > \" ${TF_ANDROID_COMPAT_LIB_DIR}/libpthread.so\" \n printf 'INPUT(-lc)\\ n' > \" ${TF_ANDROID_COMPAT_LIB_DIR}/librt.so\" \n export BUILD_FLAGS=\" --crosstool_top=@androidndk//:default_crosstool --cpu=arm64-v8a --fat_apk_cpu=arm64-v8a --host_crosstool_top=@bazel_tools//tools/cpp:toolchain --copt=-DANDROID --cxxopt=-std=c++14 --host_cxxopt=-std=c++14 --cxxopt=-include --cxxopt=cstdint --host_cxxopt=-include --host_cxxopt=cstdint --copt=-Wno-error=array-parameter --host_copt=-Wno-error=array-parameter --copt=-Wno-error=array-bounds --host_copt=-Wno-error=array-bounds --linkopt=-L${TF_ANDROID_COMPAT_LIB_DIR}\" \n export PYTHON_BIN_PATH=$(which python3)\n elif [[ -d $BAZEL_VC ]]; then\n # Work around compiler issues on Windows documented mainly in configure.py but also elsewhere\n export BUILD_FLAGS=\" --copt=//arch:AVX `#--copt=//arch:AVX2` --copt=-DWIN32_LEAN_AND_MEAN --host_copt=-DWIN32_LEAN_AND_MEAN --copt=-DNOGDI --host_copt=-DNOGDI --copt=-D_USE_MATH_DEFINES --host_copt=-D_USE_MATH_DEFINES --define=override_eigen_strong_inline=true\" \n # https://software.intel.com/en-us/articles/intel-optimization-for-tensorflow-installation-guide#wind_B_S\n export PATH=$PATH:$(pwd)/bazel-tensorflow-core-api/external/mkl_windows/lib/\n export PYTHON_BIN_PATH=$(which python.exe)\n else\n export BUILD_FLAGS=\" --copt=-msse4.1 --copt=-msse4.2 --copt=-mavx `#--copt=-mavx2 --copt=-mfma` --cxxopt=-std=c++14 --host_cxxopt=-std=c++14 --linkopt=-lstdc++ --host_linkopt=-lstdc++\" \n export PYTHON_BIN_PATH=$(which python3)\n fi\n """
2828 text = replace_once (text , old , new , path )
2929 path .write_text (text , encoding = "utf-8" )
3030
You can’t perform that action at this time.
0 commit comments