From 648cf1747e71529973dcd3659cb80525b7d66a9a Mon Sep 17 00:00:00 2001
From: Ionel Gog <icgog@google.com>
Date: Wed, 12 Feb 2025 12:41:09 -0800
Subject: [PATCH] [IFRT] Add log message to CHECK in compile_atom_program_pass
 to aid with debugging

PiperOrigin-RevId: 726157929
---
 .../ifrt/ir/transforms/ifrt_compile_atom_program_pass.cc      | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/xla/python/ifrt/ir/transforms/ifrt_compile_atom_program_pass.cc b/xla/python/ifrt/ir/transforms/ifrt_compile_atom_program_pass.cc
index 8f6a7e2d1192a..910297e26644c 100644
--- a/xla/python/ifrt/ir/transforms/ifrt_compile_atom_program_pass.cc
+++ b/xla/python/ifrt/ir/transforms/ifrt_compile_atom_program_pass.cc
@@ -210,7 +210,9 @@ void IfrtCompileAtomProgramPass::runOnOperation() {
             CHECK(atom_executable_map_
                       ->try_emplace(compile_result->name,
                                     std::move(compile_result->executable))
-                      .second);
+                      .second)
+                << "Failed to insert atom executable to map. Executable `"
+                << compile_result->name << "` already exists";
           }
 
           // Generate CallLoadedExecutableOp.