From 8fad723f0487b42c17d7f9e7770cb51dddd30125 Mon Sep 17 00:00:00 2001 From: Adrian Kuegel Date: Tue, 3 Dec 2019 01:30:07 -0800 Subject: [PATCH] Add replay_computation target for mlir_gpu backend. PiperOrigin-RevId: 283500767 Change-Id: I2839e5e698427a9df1997aed2e57d5b865bfb7e6 --- tensorflow/compiler/xla/tools/BUILD | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/tensorflow/compiler/xla/tools/BUILD b/tensorflow/compiler/xla/tools/BUILD index 09d2d76cabf..da20d28ea81 100644 --- 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 = [