Skip to content

Commit

Permalink
[xla:cpu] NFC: Delete SimpleOrcJit
Browse files Browse the repository at this point in the history
All users migrated to xla::cpu::JitCompiler (see xla/backends/cpu/codegen).

PiperOrigin-RevId: 701397325
  • Loading branch information
ezhulenev authored and Google-ML-Automation committed Nov 30, 2024
1 parent 5b17ab7 commit 5414e32
Show file tree
Hide file tree
Showing 9 changed files with 2 additions and 399 deletions.
1 change: 0 additions & 1 deletion xla/pjrt/cpu/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -195,7 +195,6 @@ cc_library(
"//xla/service/cpu:cpu_executable_run_options",
"//xla/service/cpu:cpu_runtime",
"//xla/service/cpu:cpu_xfeed",
"//xla/service/cpu:simple_orc_jit",
"//xla/stream_executor:device_memory",
"//xla/tsl/concurrency:async_value",
"//xla/tsl/concurrency:ref_count",
Expand Down
1 change: 0 additions & 1 deletion xla/pjrt/cpu/cpu_client.cc
Original file line number Diff line number Diff line change
Expand Up @@ -90,7 +90,6 @@ limitations under the License.
#include "xla/service/cpu/cpu_executable_run_options.h"
#include "xla/service/cpu/cpu_runtime.h"
#include "xla/service/cpu/cpu_xfeed.h"
#include "xla/service/cpu/simple_orc_jit.h"
#include "xla/service/custom_call_status.h"
#include "xla/service/custom_call_status_internal.h"
#include "xla/service/dump.h"
Expand Down
38 changes: 0 additions & 38 deletions xla/service/cpu/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,6 @@ load(
"//xla:xla.bzl",
"xla_cc_binary",
"xla_cc_test",
"xla_internal",
)
load("//xla/tests:build_defs.bzl", "xla_test")
load("//xla/tsl:tsl.bzl", "internal_visibility", "tf_openmp_copts", "tsl_copts")
Expand Down Expand Up @@ -231,7 +230,6 @@ cc_library(
":onednn_ops_rewriter",
":parallel_task_assignment",
":runtime_symbol_generator",
":simple_orc_jit",
":thunk_emitter",
":xla_framework",
"//xla:cpu_function_runtime",
Expand Down Expand Up @@ -540,41 +538,6 @@ cc_library(
],
)

cc_library(
name = "simple_orc_jit",
srcs = ["simple_orc_jit.cc"],
hdrs = ["simple_orc_jit.h"],
copts = if_enable_acl(["-DXLA_CPU_USE_ACL=1"]) + tsl_copts(),
deps = [
":orc_jit_memory_mapper",
":runtime_symbol_generator",
"//xla:types",
"//xla:util",
"//xla/backends/cpu/codegen:contiguous_section_memory_manager",
"//xla/backends/cpu/codegen:cpu_features",
"//xla/backends/cpu/codegen:ir_compiler",
"//xla/backends/cpu/codegen:jit_compiler",
"//xla/service:custom_call_target_registry",
"//xla/service:llvm_compiler",
"@com_google_absl//absl/container:flat_hash_map",
"@com_google_absl//absl/container:flat_hash_set",
"@com_google_absl//absl/functional:any_invocable",
"@com_google_absl//absl/memory",
"@com_google_absl//absl/strings",
"@com_google_absl//absl/types:span",
"@llvm-project//llvm:Core",
"@llvm-project//llvm:ExecutionEngine",
"@llvm-project//llvm:MC", # fixdeps: keep
"@llvm-project//llvm:OrcJIT",
"@llvm-project//llvm:OrcShared",
"@llvm-project//llvm:Support",
"@llvm-project//llvm:Target", # fixdeps: keep
"@llvm-project//llvm:TargetParser",
"@llvm-project//mlir:mlir_c_runner_utils",
"@tsl//tsl/platform:logging",
] + xla_internal(["service/cpu:named_orc_jit_memory_mapper"]),
)

cc_library(
name = "runtime_lightweight_check",
hdrs = ["runtime_lightweight_check.h"],
Expand Down Expand Up @@ -617,7 +580,6 @@ cc_library(
hdrs = ["cpu_executable.h"],
deps = [
":cpu_runtime",
":simple_orc_jit",
"//xla:executable_run_options",
"//xla:literal",
"//xla:shape_tree",
Expand Down
1 change: 0 additions & 1 deletion xla/service/cpu/cpu_executable.cc
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,6 @@ limitations under the License.
#include "xla/literal.h"
#include "xla/service/buffer_assignment.h"
#include "xla/service/cpu/cpu_runtime.h"
#include "xla/service/cpu/simple_orc_jit.h"
#include "xla/service/custom_call_status.h"
#include "xla/service/custom_call_status_internal.h"
#include "xla/service/executable.h"
Expand Down
1 change: 0 additions & 1 deletion xla/service/cpu/cpu_executable.h
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,6 @@ limitations under the License.
#include "xla/hlo/ir/hlo_module.h"
#include "xla/literal.h"
#include "xla/service/buffer_assignment.h"
#include "xla/service/cpu/simple_orc_jit.h"
#include "xla/service/custom_call_status.h"
#include "xla/service/custom_call_status_internal.h"
#include "xla/service/executable.h"
Expand Down
6 changes: 2 additions & 4 deletions xla/service/cpu/cpu_runtime.h
Original file line number Diff line number Diff line change
Expand Up @@ -37,10 +37,8 @@ namespace runtime {

// Names of runtime functions. These get resolved from the generated code to the
// right symbol at link time in one of two ways:
// 1. When using the JIT, the symbol resolver (SimpleResolver in
// third_party/tensorflow/compiler/xla/service/cpu/simple_orc_jit.cc) maps
// this symbol name to
// the actual symbol.
// 1. When using the JIT, the symbol resolver (xla::cpu::RuntimeSymbolGenerator)
// maps this symbol name to the actual symbol.
// 2. When using ahead-of-time compilation, the linker can resolve the name
// because it is a symbol in the cpu_runtime library.
extern const char* const kEigenMatMulF16SymbolName;
Expand Down
191 changes: 0 additions & 191 deletions xla/service/cpu/simple_orc_jit.cc

This file was deleted.

Loading

0 comments on commit 5414e32

Please sign in to comment.