message: Rollback: Not able to reproduce the failure.
diff: @@ -132,7 +132,6 @@ cuda_py_test(
  # XLA is not enabled by default on Mac or Windows.
  "no_mac",
  "no_windows",
  - "notap", # TODO(b/155761551): Flaky
  ],
  xla_enabled = True,
  deps = [
  