Skip to content

Commit

Permalink
Reverts 1c27e02
Browse files Browse the repository at this point in the history 
PiperOrigin-RevId: 702843057
  • Loading branch information
@beckerhe @tensorflower-gardener
beckerhe authored and tensorflower-gardener committed Dec 4, 2024
1 parent a2b8904 commit a27025f
Show file tree
Hide file tree
Showing 8 changed files with 522 additions and 276 deletions.
46 changes: 17 additions & 29 deletions third_party/xla/xla/service/gpu/BUILD
View file
Original file line number Diff line number Diff line change
Expand Up @@ -1766,7 +1766,6 @@ cc_library(
],
deps = [
":nvptx_compiler_impl",
"//xla:debug_options_flags",
"//xla/service:compiler",
"//xla/stream_executor/cuda:cuda_platform_id",
"@local_tsl//tsl/platform:path",
Expand All @@ -1790,13 +1789,12 @@ cc_library(
deps = [
":buffer_sharing",
":cublas_padding_requirements",
":gpu_asm_opts_util",
":gpu_compiler",
":ir_emission_utils",
":metrics",
":ptx_compile_options_from_debug_options",
":target_constants",
"//xla:autotune_results_proto_cc",
"//xla:debug_options_flags",
"//xla:util",
"//xla:xla_data_proto_cc",
"//xla:xla_proto_cc",
Expand Down Expand Up @@ -1846,28 +1844,40 @@ cc_library(
"//xla/stream_executor:dnn",
"//xla/stream_executor:semantic_version",
"//xla/stream_executor:stream_executor_h",
"//xla/stream_executor/cuda:assemble_compilation_provider",
"//xla/stream_executor/cuda:caching_compilation_provider",
"//xla/stream_executor/cuda:compilation_options",
"//xla/stream_executor/cuda:compilation_provider",
"//xla/stream_executor/cuda:cuda_diagnostics",
"//xla/stream_executor/cuda:cuda_platform_id",
"//xla/stream_executor/cuda:driver_compilation",
"//xla/stream_executor/cuda:nvjitlink",
"//xla/stream_executor/cuda:nvjitlink_known_issues",
"//xla/stream_executor/cuda:nvjitlink_support",
"//xla/stream_executor/cuda:ptx_compilation_method",
"//xla/stream_executor/cuda:ptx_compiler",
"//xla/stream_executor/cuda:ptx_compiler_support",
"//xla/stream_executor/cuda:ptx_linking_method",
"//xla/stream_executor/cuda:subprocess_compilation",
"//xla/stream_executor/gpu:gpu_asm_opts",
"//xla/tsl/util:env_var",
"@com_google_absl//absl/algorithm:container",
"@com_google_absl//absl/base",
"@com_google_absl//absl/base:core_headers",
"@com_google_absl//absl/cleanup",
"@com_google_absl//absl/container:node_hash_map",
"@com_google_absl//absl/log",
"@com_google_absl//absl/log:check",
"@com_google_absl//absl/status",
"@com_google_absl//absl/status:statusor",
"@com_google_absl//absl/strings",
"@com_google_absl//absl/strings:str_format",
"@com_google_absl//absl/strings:string_view",
"@com_google_absl//absl/synchronization",
"@com_google_absl//absl/types:span",
"@llvm-project//llvm:IRReader",
"@llvm-project//llvm:Support",
"@llvm-project//llvm:ir_headers",
"@local_tsl//tsl/platform:env",
"@local_tsl//tsl/platform:errors",
"@local_tsl//tsl/platform:path",
"@local_tsl//tsl/platform:status",
"@local_tsl//tsl/platform:statusor",
"@local_tsl//tsl/profiler/lib:scoped_annotation",
"@local_tsl//tsl/profiler/lib:traceme",
Expand Down Expand Up @@ -3238,28 +3248,6 @@ xla_cc_test(
],
)

cc_library(
name = "ptx_compile_options_from_debug_options",
srcs = ["ptx_compile_options_from_debug_options.cc"],
hdrs = ["ptx_compile_options_from_debug_options.h"],
deps = [
"//xla:xla_proto_cc_impl",
"//xla/stream_executor/cuda:compilation_options",
],
)

xla_cc_test(
name = "ptx_compile_options_from_debug_options_test",
srcs = ["ptx_compile_options_from_debug_options_test.cc"],
deps = [
":ptx_compile_options_from_debug_options",
"//xla/stream_executor/cuda:compilation_options",
"@com_google_googletest//:gtest_main",
"@local_tsl//tsl/platform:status_matchers",
"@local_tsl//tsl/platform:test",
],
)

cc_library(
name = "flag_utils",
srcs = ["flag_utils.cc"],
Expand Down
Loading

0 comments on commit a27025f

Please sign in to comment.