Add replay_computation target for mlir_gpu backend.

PiperOrigin-RevId: 283500767 Change-Id: I2839e5e698427a9df1997aed2e57d5b865bfb7e6
2019-12-03 01:30:07 -08:00 · 2019-12-03 01:30:07 -08:00 · 8fad723f04
commit 8fad723f04
parent 12bacab7bb
1 changed files with 8 additions and 0 deletions
--- a/tensorflow/compiler/xla/tools/BUILD
+++ b/tensorflow/compiler/xla/tools/BUILD
@ -95,6 +95,14 @@ tf_cc_binary(
    ],
 )

+tf_cc_binary(
+    name = "replay_computation_mlir_gpu",
+    deps = [
+        ":replay_computation_library",
+        "//tensorflow/compiler/xla/service:mlir_gpu_plugin",
+    ],
+)
+
 tf_cc_binary(
    name = "replay_computation_interpreter",
    deps = [