experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix sanity CI build script to use new flag to enable CUDA support.

Browse Source 
PiperOrigin-RevId: 360675922
Change-Id: I5f77e2af4e05f65cae2ff8bbb17934684b5487c1
This commit is contained in:
Christian Sigg 2021-03-03 08:30:10 -08:00 committed by TensorFlower Gardener
parent a6d887114b
commit ef8a0e1220
1 changed files with 2 additions and 2 deletions

4
tensorflow/tools/ci_build/ci_sanity.sh
View File

@ -629,11 +629,11 @@ _do_pip_no_cuda_deps_check() {
}
do_pip_no_cuda_deps_check_ubuntu() {
_do_pip_no_cuda_deps_check "--define using_cuda=true --define using_cuda_nvcc=true"
_do_pip_no_cuda_deps_check "--@local_config_cuda//:enable_cuda"
}
do_pip_no_cuda_deps_check_windows() {
_do_pip_no_cuda_deps_check "--define using_cuda=true --define using_cuda_nvcc=true --define framework_shared_object=false"
_do_pip_no_cuda_deps_check "--@local_config_cuda//:enable_cuda --define framework_shared_object=false"
}
do_configure_test() {