diff --git a/tensorflow/python/distribute/BUILD b/tensorflow/python/distribute/BUILD
index 5805e0e365b..50762bdc881 100644
--- a/tensorflow/python/distribute/BUILD
+++ b/tensorflow/python/distribute/BUILD
@@ -472,6 +472,7 @@ cuda_py_test(
     ],
     python_version = "PY3",
     tags = [
+        "notap",  # TODO(b/171355671)
         "notsan",  # TODO(b/151841995)
     ],
     deps = [