Skip to content

Missing the corelation between forward and backward processes #263

Description

@yongpan0709

🐛 Describe the bug

when
Traceback (most recent call last):
File "/.conda/envs/hta/lib/python3.10/site-packages/pandas/core/indexes/base.py", line 3812, in get_loc
return self._engine.get_loc(casted_key)
File "pandas/_libs/index.pyx", line 167, in pandas._libs.index.IndexEngine.get_loc
File "pandas/_libs/index.pyx", line 196, in pandas._libs.index.IndexEngine.get_loc
File "pandas/_libs/hashtable_class_helper.pxi", line 7088, in pandas._libs.hashtable.PyObjectHashTable.get_item
File "pandas/_libs/hashtable_class_helper.pxi", line 7096, in pandas._libs.hashtable.PyObjectHashTable.get_item
KeyError: 'fwdbwd_index'

Steps to reproduce

class TraceFWDBWDLinkTestCase(unittest.TestCase):
def setUp(self) -> None:
super().setUp()
test_data_path = Path(file).parent.parent.joinpath(
"tests/data/h100/h100_trace.json"
)
print(f'test_data_path: {test_data_path}')
self.trace = Trace(
trace_files={0: str(test_data_path)},
trace_dir="",
)
self.trace.parse_traces()
self.trace.decode_symbol_ids(use_shorten_name=False)

def test_fwdbwd_index_column(self):
    self.assertIn(
        "fwdbwd_index",
        self.trace.get_trace(0).columns,
        "fwdbwd_index column not found in trace DataFrame",
    )

def test_fwdbwd_symbol_and_id(self):
    self.assertIn(
        "fwdbwd",
        self.trace.symbol_table.sym_index,
        "fwdbwd symbol not found in trace symbol table",
    )

Expected behavior

As the testcases in "Steps to reproduce", they should be passed

Environment

Linux and Mac, both.
Python version: 3.10
Install HTA from source.

Image

Additional Info

No response

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Fields

    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions