Compatibility of Symbolic Tensor Graph (STG) with ASTRA-sim

[yue2388253]

Hi,

I am currently utilizing the Symbolic Tensor Graph (STG) tool to generate synthetic workload traces in the Chakra Execution Trace (ET) format. I have observed that the generated .et files include a comm_group field, which appears to be new to Chakra schema and cannot be correctly parsed by the astra-sim simulator.

Best,
Lin

[jonassvedas]

Maybe this will help:
https://astra-sim.github.io/astra-sim-docs/getting-started/argument-workload-config.html#enable-communicator-groups

Although I dont know if the generated comm_group files are correct.