[XLA:GPU][CoalescingAnalysis] Speed-up computation of coalescing.

PiperOrigin-RevId: 625265401
tensorflow · Apr 16, 2024 · 4405eb1 · 4405eb1
1 parent 0d8c972
commit 4405eb1
Show file tree

Hide file tree

Showing 6 changed files with 381 additions and 166 deletions.
diff --git a/third_party/xla/xla/service/gpu/model/BUILD b/third_party/xla/xla/service/gpu/model/BUILD
@@ -686,6 +686,7 @@ cc_library(
         "//xla/service/gpu/fusions:fusion_emitter",
         "@com_google_absl//absl/container:flat_hash_map",
         "@com_google_absl//absl/container:inlined_vector",
+        "@com_google_absl//absl/log",
         "@com_google_absl//absl/log:check",
         "@com_google_absl//absl/types:span",
         "@llvm-project//llvm:Support",