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

[TF:TRT] Disable quantization_mnist_test for CUDA 11.

Browse Source 
This is to workaround b/166308253.

PiperOrigin-RevId: 328740058
Change-Id: I5dcc4ee390e7f5971cb5861067b8785ade5dc40d
This commit is contained in:
Bixia Zheng 2020-08-27 08:20:02 -07:00 committed by TensorFlower Gardener
parent a9f448d4df
commit 99372b6016
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -169,6 +169,7 @@ cuda_py_test(
], ],
python_version = "PY3", python_version = "PY3",
tags = [ tags = [
"no_cuda11", # TODO(b/166308253): enable the test for CUDA 11.
"no_cuda_on_cpu_tap", "no_cuda_on_cpu_tap",
"no_oss", # TODO(b/125290478): allow running in at least some OSS configurations. "no_oss", # TODO(b/125290478): allow running in at least some OSS configurations.
"no_pip", "no_pip",