Skip to content

Missing the corelation between forward and backward processes #263

@yongpan0709

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

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions