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

[TF:TRT] Re-do the workaround for b/165611343.

Browse Source 
We actually want to disable the test in OSS due to issues with CUDA 11.

PiperOrigin-RevId: 328737158
Change-Id: I6699a8ff30e19e5e52376b76c2d387834a6e59a6
This commit is contained in:
Bixia Zheng 2020-08-27 07:59:17 -07:00 committed by TensorFlower Gardener
parent 631b90b3ac
commit a9f448d4df
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/python/compiler/tensorrt/BUILD
View File

@ -145,9 +145,8 @@ cuda_py_tests(
],
python_version = "PY3",
tags = [
"no_cuda11", # TODO(b/165611343): Need to address the failures.
"no_cuda_on_cpu_tap",
"no_oss",
"no_oss", # TODO(b/165611343): Need to address the failures for CUDA 11 in OSS build.
"no_rocm",
"no_windows",
"nomac",