more meticulous bsym check to tell if the trace is bwd

Signed-off-by: Masaki Kozuki <mkozuki@nvidia.com>
Lightning-AI · Nov 30, 2024 · 8475ff7 · 8475ff7
1 parent d60b4dc
commit 8475ff7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/thunder/transforms/tensor_subclasses.py b/thunder/transforms/tensor_subclasses.py
@@ -257,8 +257,8 @@ def __post_init__(self) -> None:
             is_backward_trace = maybe_unpack_C0_bsym.args and maybe_unpack_C1_bsym.args and (
                 maybe_unpack_C0_bsym.sym.id,
                 maybe_unpack_C1_bsym.sym.id,
-                maybe_unpack_C0_bsym.args[0].name,
-                maybe_unpack_C1_bsym.args[0].name,
+                getattr(maybe_unpack_C0_bsym.args[0], "name", ""),
+                getattr(maybe_unpack_C1_bsym.args[0], "name", ""),
             ) == (
                 prims.PrimIDs.UNPACK_SEQUENCE,
                 prims.PrimIDs.UNPACK_SEQUENCE,