• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1load("//tensorflow:tensorflow.bzl", "filegroup")
2load("//tensorflow/compiler/mlir:glob_lit_test.bzl", "glob_lit_tests")
3
4licenses(["notice"])
5
6glob_lit_tests(
7    data = [
8        ":debug_info_files",
9        ":test_utilities",
10    ],
11    driver = "@llvm-project//mlir:run_lit.sh",
12    test_file_exts = [
13        "pbtxt",
14        #        "py", TODO(b/150304798)
15    ],
16)
17
18# Bundle together all the debug info files that are used by the tests.
19filegroup(
20    name = "debug_info_files",
21    srcs = glob(
22        ["**/*.debug"],
23    ),
24)
25
26# Bundle together all of the test utilities that are used by tests.
27filegroup(
28    name = "test_utilities",
29    testonly = True,
30    data = [
31        ":concrete_function_error",
32        ":saved_model_error",
33        "//tensorflow/compiler/mlir/lite:flatbuffer_to_string",
34        "//tensorflow/compiler/mlir/lite:tf_tfl_translate",
35        "@llvm-project//llvm:FileCheck",
36        "@llvm-project//llvm:not",
37    ],
38)
39
40py_binary(
41    name = "saved_model_error",
42    srcs = ["saved_model_error.py"],
43    main = "saved_model_error.py",
44    python_version = "PY3",
45    deps = [
46        "//tensorflow:tensorflow_py",
47        "@absl_py//absl:app",
48    ],
49)
50
51py_binary(
52    name = "concrete_function_error",
53    srcs = ["concrete_function_error.py"],
54    main = "concrete_function_error.py",
55    python_version = "PY3",
56    deps = [
57        "//tensorflow:tensorflow_py",
58        "@absl_py//absl:app",
59    ],
60)
61