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

PiperOrigin-RevId: 625265401
ROCm · Apr 16, 2024 · 38f5ed1 · 38f5ed1
1 parent a734046
commit 38f5ed1
Show file tree

Hide file tree

Showing 6 changed files with 381 additions and 166 deletions.
diff --git a/xla/service/gpu/model/BUILD b/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",