commit bab643cbcb155b37d52e89e73f19c9c4fda69e2c
author George Karpenkov <cheshire@google.com> Mon Aug 12 15:24:29 2019 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Mon Aug 12 16:23:43 2019 -0700
tree 7dd968b298bddaf15118224546ed966538032acc
parent 9e7be87b437e69b875942ac7507c8cac1e050c08

[MLIR/GPU] Fix the build.

PiperOrigin-RevId: 263017663

tensorflow/compiler/xla/service/mlir_gpu/mlir_compiler.cc

diff --git a/tensorflow/compiler/xla/service/mlir_gpu/mlir_compiler.cc b/tensorflow/compiler/xla/service/mlir_gpu/mlir_compiler.cc
index 5421a3a..44c143d 100644
--- a/tensorflow/compiler/xla/service/mlir_gpu/mlir_compiler.cc
+++ b/tensorflow/compiler/xla/service/mlir_gpu/mlir_compiler.cc
@@ -51,11 +51,10 @@
     se::DeviceMemoryAllocator* device_allocator) {
   // Until we find a reason to do something different, run the same passes
   // that the normal GPU backend runs.
-  TF_RETURN_IF_ERROR(xla::gpu::impl::OptimizeHloModule(
-      module.get(), stream_exec, device_allocator));
-
-  TF_RETURN_IF_ERROR(
-      xla::gpu::impl::PrepareHloModuleForIrEmitting(module.get()));
+  gpu::NVPTXCompiler xla_compiler;
+  TF_RETURN_IF_ERROR(xla_compiler.OptimizeHloModule(module.get(), stream_exec,
+                                                    device_allocator));
+  TF_RETURN_IF_ERROR(xla_compiler.PrepareHloModuleForIrEmitting(module.get()));
 
   return std::move(module);
 }