[XLA:CPU] Emit nested computation name rather than caller's

PiperOrigin-RevId: 713993703
openxla · Jan 10, 2025 · 1fe04d9 · 1fe04d9
1 parent 62e57c4
commit 1fe04d9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/xla/backends/cpu/codegen/elemental_kernel_emitter.cc b/xla/backends/cpu/codegen/elemental_kernel_emitter.cc
@@ -363,7 +363,8 @@ ElementalKernelEmitter::ThreadLocalCallbackFactory(llvm::IRBuilderBase& builder,
     bool is_reducer = instr_->opcode() == HloOpcode::kReduce ||
                       instr_->opcode() == HloOpcode::kReduceWindow;
     TF_RETURN_IF_ERROR(ir_emitter->EmitNestedComputation(
-        *nested_computation, llvm_ir::IrName(instr_), is_reducer));
+        *nested_computation, llvm_ir::IrName(nested_computation->name()),
+        is_reducer));
   }
 
   return [ir_emitter = std::move(ir_emitter), &builder](