From 1c04e4ffb6294514b3502e13f2c9c741a2626594 Mon Sep 17 00:00:00 2001 From: vkovinicTT Date: Mon, 6 Oct 2025 20:12:05 +0000 Subject: [PATCH 1/2] enabled renaming op name without adding stack frame locations --- torch_xla/csrc/lowering_context.cpp | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/torch_xla/csrc/lowering_context.cpp b/torch_xla/csrc/lowering_context.cpp index 65f438643f17..b07a9b4dad05 100644 --- a/torch_xla/csrc/lowering_context.cpp +++ b/torch_xla/csrc/lowering_context.cpp @@ -84,9 +84,12 @@ class HloMetadataSetter { } metadata.set_op_name(absl::StrCat(op_name_prefix, op_type)); - // Sets file, line and stack_frame_id in metadata - lowering_context.stack_frame_index_builder()->AddStackFrameLocations( - nmeta.frame_info, static_cast(max_stack_depth), metadata); + // NOTE: if max_stack_depth is 0, we are just renaming the op, so we don't need to add stack frame locations + if (max_stack_depth > 0) { + // Sets file, line and stack_frame_id in metadata + lowering_context.stack_frame_index_builder()->AddStackFrameLocations( + nmeta.frame_info, static_cast(max_stack_depth), metadata); + } lowering_context.builder()->SetOpMetadata(std::move(metadata)); } From 098d0b9a0737b501be6d5581366cfdcf266fa4da Mon Sep 17 00:00:00 2001 From: vkovinicTT Date: Thu, 9 Oct 2025 08:22:55 +0000 Subject: [PATCH 2/2] prevent scope from overwriting custom metadata --- torch_xla/csrc/lowering_context.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch_xla/csrc/lowering_context.cpp b/torch_xla/csrc/lowering_context.cpp index b07a9b4dad05..0fbd702d9b92 100644 --- a/torch_xla/csrc/lowering_context.cpp +++ b/torch_xla/csrc/lowering_context.cpp @@ -78,7 +78,7 @@ class HloMetadataSetter { max_stack_depth = custom_opname_meta->max_stack_depth; } - if (!nmeta.scope.empty()) { + else if (!nmeta.scope.empty()) { op_name_prefix = absl::StrCat(absl::StrReplaceAll(nmeta.scope, {{":", "_"}}), "/"); }