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

Merge pull request #45265 from Flamefire:add-missing-llvm-deps

Browse Source 
PiperOrigin-RevId: 347501080
Change-Id: I57df15ad1f334023a2351da09e6a1094001d8b9d
This commit is contained in:
TensorFlower Gardener 2020-12-14 17:12:02 -08:00
commit 73fbda7026
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/compiler/mlir/tools/kernel_gen/BUILD
View File

@ -12,6 +12,7 @@ load(
"@local_config_rocm//rocm:build_defs.bzl",
"if_rocm_is_configured",
)
load("//tensorflow/core/platform:build_config.bzl", "if_llvm_aarch64_available", "if_llvm_system_z_available")
package(
default_visibility = [":friends"],
@ -102,8 +103,10 @@ tf_cc_binary(
"//tensorflow/stream_executor/lib",
"@com_google_absl//absl/strings",
"@llvm-project//llvm:Analysis",
"@llvm-project//llvm:ARMCodeGen", # fixdeps: keep
"@llvm-project//llvm:CodeGen",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:PowerPCCodeGen", # fixdeps: keep
"@llvm-project//llvm:Support",
"@llvm-project//llvm:Target",
"@llvm-project//llvm:X86CodeGen", # fixdeps: keep
@ -111,7 +114,11 @@ tf_cc_binary(
"@llvm-project//mlir:ExecutionEngineUtils",
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:TargetLLVMIR",
],
] + if_llvm_system_z_available([
"@llvm-project//llvm:SystemZCodeGen", # fixdeps: keep
]) + if_llvm_aarch64_available([
"@llvm-project//llvm:AArch64CodeGen", # fixdeps: keep
]),
)
tf_cc_binary(