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