Skip to content

Commit b67f3c8

Browse files
authored
use available CPU instructions
1 parent 63bc266 commit b67f3c8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

CMakeLists.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,8 +41,8 @@ ExternalProject_Add(${PROJECT_NAME}-ext
4141
TF_CUDA_COMPUTE_CAPABILITIES=3.5,5.2
4242
./configure
4343

44-
BUILD_COMMAND bazel build -c opt -j 200 ${TF_BUILD_CUDA} //tensorflow:libtensorflow.so
45-
//tensorflow/tools/pip_package:build_pip_package
44+
BUILD_COMMAND bazel build -c opt -j 200 --config=opt ${TF_BUILD_CUDA} //tensorflow:libtensorflow.so
45+
//tensorflow/tools/pip_package:build_pip_package
4646
INSTALL_COMMAND cp <SOURCE_DIR>/bazel-bin/tensorflow/libtensorflow.so ${CMAKE_SOURCE_DIR}/lib
4747
COMMAND <SOURCE_DIR>/bazel-bin/tensorflow/tools/pip_package/build_pip_package ${CMAKE_SOURCE_DIR}/lib
4848
COMMAND /bin/bash -c "wheel unpack ${CMAKE_SOURCE_DIR}/lib/tensorflow-*.whl"

0 commit comments

Comments
 (0)