Skip to content

Commit

Permalink
Improve structure of GPU IR emission logic. #cleanup
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 726704548
  • Loading branch information
toli-y authored and Google-ML-Automation committed Feb 14, 2025
1 parent d88085e commit 7cd8616
Show file tree
Hide file tree
Showing 5 changed files with 197 additions and 147 deletions.
13 changes: 9 additions & 4 deletions xla/service/gpu/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -1264,6 +1264,7 @@ cc_library(
"//xla:status_macros",
"//xla:util",
"//xla:xla_data_proto_cc",
"//xla:xla_proto_cc",
"//xla/backends/gpu/runtime:sequential_thunk",
"//xla/hlo/analysis:hlo_dataflow_analysis",
"//xla/hlo/analysis:hlo_ordering",
Expand All @@ -1273,10 +1274,18 @@ cc_library(
"//xla/service:dump",
"//xla/service:hlo_proto_cc",
"//xla/service:logical_buffer",
"//xla/service/llvm_ir:llvm_command_line_options",
"//xla/stream_executor:device_description",
"//xla/stream_executor:platform",
"//xla/stream_executor/cuda:cuda_compute_capability",
"//xla/stream_executor/rocm:rocm_platform_id",
"//xla/tsl/platform:env",
"//xla/tsl/platform:errors",
"//xla/tsl/platform:statusor",
"@com_google_absl//absl/container:flat_hash_map",
"@com_google_absl//absl/log",
"@com_google_absl//absl/log:check",
"@com_google_absl//absl/log:globals",
"@com_google_absl//absl/status",
"@com_google_absl//absl/status:statusor",
"@com_google_absl//absl/strings",
Expand All @@ -1287,11 +1296,7 @@ cc_library(
"@llvm-project//mlir:IR",
"@llvm-project//mlir:Pass",
"@llvm-project//mlir:Support",
"@tsl//tsl/platform:env",
"@tsl//tsl/platform:errors",
"@tsl//tsl/platform:logging",
"@tsl//tsl/platform:path",
"@tsl//tsl/platform:statusor",
"@tsl//tsl/profiler/lib:scoped_annotation",
],
)
Expand Down
Loading

0 comments on commit 7cd8616

Please sign in to comment.