experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #32738 from ROCmSoftwarePlatform:google-upstream-pr-xla-no-rocm-tag

Browse Source 
PiperOrigin-RevId: 270790979
This commit is contained in:
TensorFlower Gardener 2019-09-23 17:07:04 -07:00
commit e9731534a2
4 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/compiler/mlir/glob_lit_test.bzl
View File

@ -34,6 +34,7 @@ def glob_lit_tests(
native.py_test( native.py_test(
name = "glob_lit_tests", name = "glob_lit_tests",
srcs = ["@llvm//:lit"], srcs = ["@llvm//:lit"],
tags = ["no_rocm"],
args = [ args = [
"tensorflow/compiler/mlir --config-prefix=runlit", "tensorflow/compiler/mlir --config-prefix=runlit",
], ],

1
tensorflow/compiler/mlir/tensorflow/BUILD
View File

@ -632,6 +632,7 @@ cc_library(
tf_cc_test( tf_cc_test(
name = "error_util_test", name = "error_util_test",
srcs = ["utils/error_util_test.cc"], srcs = ["utils/error_util_test.cc"],
tags = ["no_rocm"],
deps = [ deps = [
":error_util", ":error_util",
"//tensorflow/compiler/xla:test", "//tensorflow/compiler/xla:test",

1
tensorflow/compiler/tests/BUILD
View File

@ -395,6 +395,7 @@ tf_xla_py_test(
srcs = ["depthwise_conv_op_test.py"], srcs = ["depthwise_conv_op_test.py"],
shard_count = 5, shard_count = 5,
tags = [ tags = [
"no_rocm",
"noasan", "noasan",
"nomsan", "nomsan",
"notsan", "notsan",

2
tensorflow/compiler/xla/service/gpu/BUILD
View File

@ -1455,7 +1455,7 @@ cc_library(
tf_cc_test( tf_cc_test(
name = "buffer_comparator_test", name = "buffer_comparator_test",
srcs = ["buffer_comparator_test.cc"], srcs = ["buffer_comparator_test.cc"],
tags = tf_cuda_tests_tags(), tags = ["no_rocm"] + tf_cuda_tests_tags(),
deps = [ deps = [
":buffer_comparator", ":buffer_comparator",
"//tensorflow/compiler/xla:shape_util", "//tensorflow/compiler/xla:shape_util",