Add mlir based xla gpu jit target.

PiperOrigin-RevId: 285166520 Change-Id: Ia57dd65fbe19f9819dfeb46ef53b5f808693861e
2019-12-12 04:29:41 -08:00 · 2019-12-12 04:29:41 -08:00 · 02a94bf820
commit 02a94bf820
parent a61b8965fe
1 changed files with 13 additions and 0 deletions
--- a/tensorflow/compiler/jit/BUILD
+++ b/tensorflow/compiler/jit/BUILD
@ -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"],