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

Add mlir based xla gpu jit target.

Browse Source 
PiperOrigin-RevId: 285166520
Change-Id: Ia57dd65fbe19f9819dfeb46ef53b5f808693861e
This commit is contained in:
Stephan Herhut 2019-12-12 04:29:41 -08:00 committed by TensorFlower Gardener
parent a61b8965fe
commit 02a94bf820
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tensorflow/compiler/jit/BUILD
View File

@ -71,6 +71,19 @@ cc_library(
alwayslink = 1,
)
cc_library(
name = "xla_mlir_gpu_jit",
visibility = ["//visibility:public"],
deps = if_cuda_or_rocm([
":jit_compilation_passes",
"//tensorflow/compiler/jit/kernels:xla_ops",
"//tensorflow/compiler/tf2xla/kernels:xla_ops",
"//tensorflow/compiler/tf2xla/kernels:xla_dummy_ops",
"//tensorflow/compiler/xla/service:mlir_gpu_plugin",
]),
alwayslink = 1,
)
cc_library(
name = "xla_cpu_device",
srcs = ["xla_cpu_device.cc"],