Skip to content

Commit

Permalink
Reverts 1ce2bbc
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 692972293
  • Loading branch information
changm authored and tensorflower-gardener committed Nov 4, 2024
1 parent e49484f commit 9bde08c
Show file tree
Hide file tree
Showing 6 changed files with 0 additions and 291 deletions.
51 changes: 0 additions & 51 deletions third_party/xla/xla/service/cpu/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,6 @@ load(
"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")
load("//xla/tsl:tsl.default.bzl", "filegroup", "get_compatible_with_portable")
load(
Expand Down Expand Up @@ -227,7 +226,6 @@ cc_library(
":ir_emission_utils",
":ir_emitter",
":ir_emitter2",
":metrics",
":onednn_contraction_rewriter",
":onednn_ops_rewriter",
":parallel_task_assignment",
Expand Down Expand Up @@ -412,31 +410,6 @@ cc_library(
]),
)

xla_test(
name = "cpu_compiler_test",
srcs = ["cpu_compiler_test.cc"],
backends = [
"cpu",
],
deps = [
"//xla:shape_util",
"//xla/pjrt:pjrt_client",
"//xla/service:hlo_runner_interface",
"//xla/service:hlo_runner_pjrt",
"//xla/service:platform_util",
"//xla/tests:hlo_test_base",
"//xla/tests:new_hlo_test_base",
"//xla/tests:pjrt_client_registry",
"//xla/tests:pjrt_cpu_client_registry",
"//xla/tests:xla_internal_test_main",
"//xla/tsl/lib/core:status_test_util",
"//xla/tsl/lib/monitoring:collected_metrics",
"//xla/tsl/lib/monitoring:collection_registry",
"@local_tsl//tsl/platform:statusor",
"@local_tsl//tsl/platform:test",
],
)

cc_library(
# The old target name will still be used so that dependencies won't break.
# In the future, dependencies should be cleaned up and relinked to the above
Expand Down Expand Up @@ -2092,27 +2065,3 @@ cc_library(
hdrs = ["cpu_executable_run_options.h"],
deps = [":collectives_interface"],
)

cc_library(
name = "metrics",
srcs = ["metrics.cc"],
hdrs = ["metrics.h"],
deps = [
"//xla/tsl/lib/monitoring:counter",
"@com_google_absl//absl/strings",
"@com_google_absl//absl/strings:string_view",
"@local_tsl//tsl/platform:stacktrace",
],
)

xla_cc_test(
name = "metrics_test",
srcs = ["metrics_test.cc"],
deps = [
":metrics",
"//xla/tests:xla_internal_test_main",
"//xla/tsl/lib/monitoring:collected_metrics",
"//xla/tsl/lib/monitoring:collection_registry",
"@local_tsl//tsl/platform:test",
],
)
2 changes: 0 additions & 2 deletions third_party/xla/xla/service/cpu/cpu_compiler.cc
Original file line number Diff line number Diff line change
Expand Up @@ -155,7 +155,6 @@ limitations under the License.
#include "xla/service/cpu/executable.pb.h"
#include "xla/service/cpu/ir_emitter.h"
#include "xla/service/cpu/ir_emitter2.h"
#include "xla/service/cpu/metrics.h"
#include "xla/service/cpu/parallel_task_assignment.h"
#include "xla/service/cpu/simple_orc_jit.h"
#include "xla/service/cpu/target_machine_features.h"
Expand Down Expand Up @@ -1717,7 +1716,6 @@ absl::StatusOr<std::unique_ptr<Executable>> CpuCompiler::RunBackend(
[[maybe_unused]] se::StreamExecutor* stream_exec,
const CompileOptions& options) {
VLOG(1) << "Compiling: " << module->name();
RecordCpuCompilerStacktrace();
XLA_SCOPED_LOGGING_TIMER(
absl::StrFormat("Compiling [%s] for CPU using JIT", module->name()));
std::string slow_compilation_msg =
Expand Down
87 changes: 0 additions & 87 deletions third_party/xla/xla/service/cpu/cpu_compiler_test.cc

This file was deleted.

68 changes: 0 additions & 68 deletions third_party/xla/xla/service/cpu/metrics.cc

This file was deleted.

34 changes: 0 additions & 34 deletions third_party/xla/xla/service/cpu/metrics.h

This file was deleted.

49 changes: 0 additions & 49 deletions third_party/xla/xla/service/cpu/metrics_test.cc

This file was deleted.

0 comments on commit 9bde08c

Please sign in to comment.