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

Add missing arch-specific dependencies to tf_to_kernel

Browse Source 
The tf_to_kernel target is missing some dependencies which shows on e.g. POWER by undefined reference errors
Fixes #45104
This commit is contained in:
Alexander Grund 2020-11-30 11:47:03 +01:00
parent f30e7d3a7e
commit 62d5bff6dd
No known key found for this signature in database
GPG Key ID: E92C451FC21EF13F
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
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"],
@ -125,8 +126,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
@ -135,6 +138,11 @@ tf_cc_binary(
"@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(