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

Adding no_rocm tag to //tensorflow/python/ops/numpy_ops:np_interop_test_gpu

Browse Source 
This test recently started failing on the ROCm platform, because of OOM errors. Will re-enable it once we root cause and issue and fix it.
This commit is contained in:
Deven Desai 2020-12-22 19:41:11 +00:00
parent c451597c70
commit 358cbc9e53
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/ops/numpy_ops/BUILD
View File

@ -110,6 +110,7 @@ cuda_py_test(
cuda_py_test(
name = "np_interop_test",
srcs = ["np_interop_test.py"],
tags = ["no_rocm"],
deps = [
":numpy",
"//tensorflow:tensorflow_py",