[xla:runtime] NFC: Move execution_engine library from jitrt to xla/ru…

…ntime PiperOrigin-RevId: 467292686
tensorflow · Aug 12, 2022 · 9bb23f7 · 9bb23f7
1 parent 4101937
commit 9bb23f7
Show file tree

Hide file tree

Showing 7 changed files with 5 additions and 566 deletions.
diff --git a/backends/jitrt/BUILD b/backends/jitrt/BUILD
@@ -77,32 +77,12 @@ licenses(["notice"])
 # )
 #
 # tfrt_cc_library(
-#     name = "execution_engine",
-#     srcs = ["lib/execution_engine.cc"],
-#     hdrs = ["include/tfrt/jitrt/execution_engine.h"],
-#     compatible_with = ["//buildenv/target:non_prod"],
-#     visibility = ["@tf_runtime//:friends"],
-#     deps = [
-#         "@llvm-project//llvm:ARMAsmParser",
-#         "@llvm-project//llvm:ARMCodeGen",
-#         "@llvm-project//llvm:Core",
-#         "@llvm-project//llvm:ExecutionEngine",
-#         "@llvm-project//llvm:OrcJIT",
-#         "@llvm-project//llvm:Support",
-#         "@llvm-project//llvm:X86AsmParser",
-#         "@llvm-project//llvm:X86CodeGen",
-#         "@tf_runtime//:support",
-#     ],
-# )
-#
-# tfrt_cc_library(
 #     name = "jitrt",
 #     srcs = ["lib/jitrt.cc"],
 #     hdrs = ["include/tfrt/jitrt/jitrt.h"],
 #     compatible_with = ["//buildenv/target:non_prod"],
 #     visibility = ["@tf_runtime//:friends"],
 #     deps = [
-#         ":execution_engine",
 #         ":results",
 #         ":xla",
 #         "@llvm-project//llvm:Core",
@@ -131,6 +111,7 @@ licenses(["notice"])
 #         "//third_party/tensorflow/compiler/xla/runtime:custom_call",
 #         "//third_party/tensorflow/compiler/xla/runtime:custom_call_registry",
 #         "//third_party/tensorflow/compiler/xla/runtime:diagnostics",
+#         "//third_party/tensorflow/compiler/xla/runtime:execution_engine",
 #         "//third_party/tensorflow/compiler/xla/runtime:memory_mapper",
 #         "//third_party/tensorflow/compiler/xla/runtime:symbolic_shape",
 #         "//third_party/tensorflow/compiler/xla/runtime:types",

diff --git a/backends/jitrt/cpp_tests/BUILD b/backends/jitrt/cpp_tests/BUILD
@@ -45,11 +45,11 @@ licenses(["notice"])
 #         "@llvm-project//mlir:Support",
 #         "//third_party/tensorflow/compiler/xla/runtime:custom_call",
 #         "//third_party/tensorflow/compiler/xla/runtime:diagnostics",
+#         "//third_party/tensorflow/compiler/xla/runtime:execution_engine",
 #         "@tf_runtime//:dtype",
 #         "@tf_runtime//:support",
 #         "@tf_runtime//backends/jitrt",
 #         "@tf_runtime//backends/jitrt:custom_call_testlib",
-#         "@tf_runtime//backends/jitrt:execution_engine",
 #         "@tf_runtime//backends/jitrt:jitrt_compiler",
 #     ],
 # )

diff --git a/backends/jitrt/cpp_tests/custom_call_test.cc b/backends/jitrt/cpp_tests/custom_call_test.cc
@@ -23,10 +23,10 @@
 #include "mlir/Support/LogicalResult.h"
 #include "tfrt/dtype/dtype.h"
 #include "tfrt/jitrt/custom_calls/custom_call_testlib.h"
-#include "tfrt/jitrt/execution_engine.h"
 #include "tfrt/jitrt/jitrt.h"
 #include "tfrt/jitrt/jitrt_compiler.h"
 #include "third_party/tensorflow/compiler/xla/runtime/diagnostics.h"
+#include "third_party/tensorflow/compiler/xla/runtime/execution_engine.h"
 
 // Forward declare types enabling compiled kernel <-> runtime integration.
 namespace xla {

diff --git a/backends/jitrt/include/tfrt/jitrt/execution_engine.h b/backends/jitrt/include/tfrt/jitrt/execution_engine.h
diff --git a/backends/jitrt/include/tfrt/jitrt/jitrt.h b/backends/jitrt/include/tfrt/jitrt/jitrt.h
@@ -37,7 +37,6 @@
 #include "mlir/Support/LogicalResult.h"
 #include "mlir/Transforms/DialectConversion.h"
 #include "tfrt/host_context/task_function.h"
-#include "tfrt/jitrt/execution_engine.h"
 #include "tfrt/jitrt/results.h"
 #include "tfrt/support/forward_decls.h"
 #include "third_party/tensorflow/compiler/xla/mlir/transforms/runtime/specialization.h"
@@ -48,6 +47,7 @@
 #include "third_party/tensorflow/compiler/xla/runtime/constraints.h"
 #include "third_party/tensorflow/compiler/xla/runtime/custom_call.h"
 #include "third_party/tensorflow/compiler/xla/runtime/diagnostics.h"
+#include "third_party/tensorflow/compiler/xla/runtime/execution_engine.h"
 #include "third_party/tensorflow/compiler/xla/runtime/memory_mapper.h"
 #include "third_party/tensorflow/compiler/xla/runtime/symbolic_shape.h"
 #include "third_party/tensorflow/compiler/xla/runtime/types.h"