mlcommons / chakra

Repository for MLCommons Chakra schema and tools
https://mlcommons.org/working-groups/research/chakra/
Apache License 2.0
45 stars 17 forks source link

some questions about generating trace #107

Open tyn513 opened 2 weeks ago

tyn513 commented 2 weeks ago

1. how to generate et trace from astra-sim?

the trace from pytorch is too complex. As mentioned in the issue https://github.com/mlcommons/chakra/issues/76,it seems possible to generate trace from Astra-sim. Could you please provide tutorial or suggestion?

2. how to get et trace with memory nodes?

the trace i got only have COMP_NODE and COMM_COLL_NODE, as mentioned in your paper, it is possible to get the memory access node. May i ask how to get the traces with nodes like that?

3. trace examples required

Could you please provide several generated chakra et, which includes the comp,mem,comm, as mentioned in some other issues ? e.g. https://github.com/mlcommons/chakra/issues/66

dageita commented 1 week ago

I also noticed that the Chakra paper mentioned using a generative ML model to generate ET. How is this achieved?