32b7183acd
This commit mostly either enables or disables unittests on the ROCM platform, details listed below * adding/removing no_rocm tag for tests in the //tensorflow/compiler/mlir dir * enabling / disabling subtests within //tensorflow/core/grappler/optimizers:constant_folding_test for the ROCm platform * disabling a subtest within //tensorflow/core/distributed_runtime:collective_param_resolver_distributed_test fir the ROCm platform * adding no_rocm tag to tests that are failing on the ROCm platform * minor bug fix to ensure that the //tensorflow/compiler/mlir/tensorflow:error_util_test passes on a consistent basis
40 lines
962 B
Python
40 lines
962 B
Python
load("//tensorflow/compiler/mlir:glob_lit_test.bzl", "glob_lit_tests")
|
|
|
|
licenses(["notice"])
|
|
|
|
glob_lit_tests(
|
|
data = [
|
|
":quant_stats_files",
|
|
":test_utilities",
|
|
],
|
|
driver = "@llvm-project//mlir:run_lit.sh",
|
|
tags_override = {
|
|
"add.pbtxt": ["no_rocm"],
|
|
"conv_2d.pbtxt": ["no_rocm"],
|
|
"fake_quant_per_channel.pbtxt": ["no_rocm"],
|
|
"ophint_lstm.pbtxt": ["no_rocm"],
|
|
},
|
|
test_file_exts = [
|
|
"pbtxt",
|
|
],
|
|
)
|
|
|
|
# Bundle together all of the test utilities that are used by tests.
|
|
filegroup(
|
|
name = "test_utilities",
|
|
testonly = True,
|
|
data = [
|
|
"//tensorflow/compiler/mlir/lite:flatbuffer_to_string",
|
|
"//tensorflow/compiler/mlir/lite:tf_tfl_translate",
|
|
"@llvm-project//llvm:FileCheck",
|
|
],
|
|
)
|
|
|
|
# Bundle together all the quant stats files that are used by the tests.
|
|
filegroup(
|
|
name = "quant_stats_files",
|
|
srcs = glob(
|
|
["**/*.stats"],
|
|
),
|
|
)
|