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

Update presubmit to cuda 10.1. 

PiperOrigin-RevId: 302909511
Change-Id: Ifb0867efd5bddfa1ac9697b3b772d51a9de722d7
This commit is contained in:
A. Unique TensorFlower 2020-03-25 09:46:23 -07:00 committed by TensorFlower Gardener
parent 2faa541d42
commit a3493f7e2b
3 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/core/profiler/internal/gpu/BUILD
View File

@ -53,7 +53,10 @@ tf_cc_test_gpu(
srcs = ["device_tracer_test.cc"], srcs = ["device_tracer_test.cc"],
args = ["--heap_check=local"], args = ["--heap_check=local"],
linkstatic = tf_kernel_tests_linkstatic(), linkstatic = tf_kernel_tests_linkstatic(),
tags = tf_cuda_tests_tags() + ["nomac"], tags = tf_cuda_tests_tags() + [
"nomac",
"gpu_cupti",
],
deps = [ deps = [
":device_tracer", ":device_tracer",
"//tensorflow/cc:cc_ops", "//tensorflow/cc:cc_ops",

3
tensorflow/python/BUILD
View File

@ -6409,6 +6409,9 @@ cuda_py_test(
size = "small", size = "small",
srcs = ["client/timeline_test.py"], srcs = ["client/timeline_test.py"],
python_version = "PY3", python_version = "PY3",
tags = [
"gpu_cupti",
],
xla_enable_strict_auto_jit = False, # Graph structure is different with autojit xla_enable_strict_auto_jit = False, # Graph structure is different with autojit
deps = [ deps = [
":client", ":client",

6
tensorflow/tools/ci_build/presubmit/ubuntu_16/gpu_py36_full/build.sh
View File

@ -45,12 +45,14 @@ function run_build () {
export ACTION_PATH="/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin" export ACTION_PATH="/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin"
export PYTHON_BIN_PATH="/usr/bin/python3" export PYTHON_BIN_PATH="/usr/bin/python3"
export TF2_BEHAVIOR=1 export TF2_BEHAVIOR=1
tag_filters="gpu,-no_gpu,-nogpu,-benchmark-test,-no_oss,-oss_serial,-no_gpu_presubmit""$(maybe_skip_v1)" # TODO(b/152356894):
# Remove -gpu_cupti once RBE supports cupti tests.
tag_filters="gpu,-no_gpu,-nogpu,-benchmark-test,-no_oss,-oss_serial,-no_gpu_presubmit,-gpu_cupti""$(maybe_skip_v1)"
# Get the default test targets for bazel. # Get the default test targets for bazel.
source tensorflow/tools/ci_build/build_scripts/PRESUBMIT_BUILD_TARGETS.sh source tensorflow/tools/ci_build/build_scripts/PRESUBMIT_BUILD_TARGETS.sh
RBE_CONFIG="@ubuntu16.04-py3-gcc7_manylinux2010-cuda10.0-cudnn7-tensorrt5.1" RBE_CONFIG="@ubuntu16.04-py3-gcc7_manylinux2010-cuda10.1-cudnn7-tensorrt6.0"
TF_CUDA_CONFIG_REPO="${RBE_CONFIG}_config_cuda" TF_CUDA_CONFIG_REPO="${RBE_CONFIG}_config_cuda"
TF_TENSORRT_CONFIG_REPO="${RBE_CONFIG}_config_tensorrt" TF_TENSORRT_CONFIG_REPO="${RBE_CONFIG}_config_tensorrt"
TF_PYTHON_CONFIG_REPO="${RBE_CONFIG}_config_python" TF_PYTHON_CONFIG_REPO="${RBE_CONFIG}_config_python"