[ROCm] Updating ROCm CI scripts to skip //tensorflow/lite/...

2020-04-21 14:49:55 +00:00 · 2020-04-21 14:49:55 +00:00 · 463e8ad569
commit 463e8ad569
parent 7550e571ac
3 changed files with 3 additions and 5 deletions
--- a/tensorflow/lite/python/BUILD
+++ b/tensorflow/lite/python/BUILD
@ -143,7 +143,6 @@ py_test(
    shard_count = 4,
    srcs_version = "PY2AND3",
    tags = [
-        "no_rocm",
        "no_windows",
    ],
    deps = [
@ -160,7 +159,6 @@ py_test(
    python_version = "PY3",
    srcs_version = "PY2AND3",
    tags = [
-        "no_rocm",
        "no_windows",
    ],
    deps = [
--- a/tensorflow/tools/ci_build/linux/rocm/run_cc_core.sh
+++ b/tensorflow/tools/ci_build/linux/rocm/run_cc_core.sh
@ -52,8 +52,7 @@ bazel test \
      -- \
      //tensorflow/... \
      -//tensorflow/compiler/... \
-      -//tensorflow/lite/delegates/gpu/gl/... \
-      -//tensorflow/lite/delegates/gpu/cl/... \
+      -//tensorflow/lite/... \
 && bazel test \
      --config=rocm \
      -k \
--- a/tensorflow/tools/ci_build/linux/rocm/run_py3_core.sh
+++ b/tensorflow/tools/ci_build/linux/rocm/run_py3_core.sh
@ -51,4 +51,5 @@ bazel test \
      --run_under=//tensorflow/tools/ci_build/gpu_build:parallel_gpu_execute \
      -- \
      //tensorflow/... \
-      -//tensorflow/compiler/...
+      -//tensorflow/compiler/... \
+      -//tensorflow/lite/...