Disable some tests on internal CI

PiperOrigin-RevId: 303784690
Change-Id: I696c3f367fa57816f770552aed8075e26cbf559a
This commit is contained in:
A. Unique TensorFlower 2020-03-30 11:31:09 -07:00 committed by TensorFlower Gardener
parent 503179f662
commit be8d751324
4 changed files with 23 additions and 13 deletions

View File

@ -8,6 +8,10 @@ glob_lit_tests(
":test_utilities", ":test_utilities",
], ],
driver = "@llvm-project//mlir:run_lit.sh", driver = "@llvm-project//mlir:run_lit.sh",
exclude = [
"**/v1_1.0_224_frozen.wrong_attr.line.part.pbtxt",
"**/v1_1.0_224_frozen.wrong_attr.stack.part.pbtxt",
],
test_file_exts = [ test_file_exts = [
"pbtxt", "pbtxt",
# "py", TODO(b/150304798) # "py", TODO(b/150304798)

View File

@ -1116,6 +1116,10 @@ tf_cc_test(
name = "compile_mlir_util_test", name = "compile_mlir_util_test",
size = "small", size = "small",
srcs = ["utils/compile_mlir_util_test.cc"], srcs = ["utils/compile_mlir_util_test.cc"],
tags = [
"manual",
"notap",
],
deps = [ deps = [
":compile_mlir_util", ":compile_mlir_util",
"//tensorflow/compiler/tf2xla:common", "//tensorflow/compiler/tf2xla:common",

View File

@ -542,6 +542,7 @@ gentbl(
td_file = "include/mlir/Dialect/Shape/IR/ShapeOps.td", td_file = "include/mlir/Dialect/Shape/IR/ShapeOps.td",
td_srcs = [ td_srcs = [
":StdOpsTdFiles", ":StdOpsTdFiles",
"include/mlir/Interfaces/InferTypeOpInterface.td",
], ],
) )
@ -560,6 +561,7 @@ cc_library(
":CallOpInterfaces", ":CallOpInterfaces",
":CommonFolders", ":CommonFolders",
":IR", ":IR",
":InferTypeOpInterface",
":ShapeOpsIncGen", ":ShapeOpsIncGen",
":SideEffects", ":SideEffects",
":Support", ":Support",

View File

@ -48,35 +48,35 @@ gentbl(
gentbl( gentbl(
name = "TestOpsIncGen", name = "TestOpsIncGen",
strip_include_prefix = "lib/TestDialect", strip_include_prefix = "lib/Dialect/Test",
tbl_outs = [ tbl_outs = [
( (
"-gen-op-decls", "-gen-op-decls",
"lib/TestDialect/TestOps.h.inc", "lib/Dialect/Test/TestOps.h.inc",
), ),
( (
"-gen-op-defs", "-gen-op-defs",
"lib/TestDialect/TestOps.cpp.inc", "lib/Dialect/Test/TestOps.cpp.inc",
), ),
( (
"-gen-dialect-decls", "-gen-dialect-decls",
"lib/TestDialect/TestOpsDialect.h.inc", "lib/Dialect/Test/TestOpsDialect.h.inc",
), ),
( (
"-gen-enum-decls", "-gen-enum-decls",
"lib/TestDialect/TestOpEnums.h.inc", "lib/Dialect/Test/TestOpEnums.h.inc",
), ),
( (
"-gen-enum-defs", "-gen-enum-defs",
"lib/TestDialect/TestOpEnums.cpp.inc", "lib/Dialect/Test/TestOpEnums.cpp.inc",
), ),
( (
"-gen-rewriters", "-gen-rewriters",
"lib/TestDialect/TestPatterns.inc", "lib/Dialect/Test/TestPatterns.inc",
), ),
], ],
tblgen = "@llvm-project//mlir:mlir-tblgen", tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "lib/TestDialect/TestOps.td", td_file = "lib/Dialect/Test/TestOps.td",
td_srcs = [ td_srcs = [
"@llvm-project//mlir:OpBaseTdFiles", "@llvm-project//mlir:OpBaseTdFiles",
"@llvm-project//mlir:include/mlir/IR/OpAsmInterface.td", "@llvm-project//mlir:include/mlir/IR/OpAsmInterface.td",
@ -91,15 +91,15 @@ gentbl(
cc_library( cc_library(
name = "TestDialect", name = "TestDialect",
srcs = [ srcs = [
"lib/TestDialect/TestDialect.cpp", "lib/Dialect/Test/TestDialect.cpp",
"lib/TestDialect/TestPatterns.cpp", "lib/Dialect/Test/TestPatterns.cpp",
], ],
hdrs = [ hdrs = [
"lib/TestDialect/TestDialect.h", "lib/Dialect/Test/TestDialect.h",
], ],
includes = [ includes = [
"lib/DeclarativeTransforms", "lib/DeclarativeTransforms",
"lib/TestDialect", "lib/Dialect/Test",
], ],
deps = [ deps = [
":TestOpsIncGen", ":TestOpsIncGen",
@ -154,7 +154,7 @@ cc_library(
"lib/Transforms/*.cpp", "lib/Transforms/*.cpp",
]), ]),
defines = ["MLIR_CUDA_CONVERSIONS_ENABLED"], defines = ["MLIR_CUDA_CONVERSIONS_ENABLED"],
includes = ["lib/TestDialect"], includes = ["lib/Dialect/Test"],
deps = [ deps = [
":TestDialect", ":TestDialect",
":TestLinalgTransformPatternsIncGen", ":TestLinalgTransformPatternsIncGen",