Request for Easy-to-Use Commands for Generating Execution Traces for Popular Models

[yue2388253]

Hello,

I’ve been exploring the symbolic_tensor_graph tool and found it to be a very useful resource for generating execution traces of AI model training. However, it would be even more helpful if there were pre-configured, easy-to-use commands or scripts that allow users to quickly generate traces for some of the popular models like GPT, ResNet, BERT, and others. This would greatly improve the tool's accessibility and ease of use, enabling a broader audience to seamlessly generate relevant execution traces.

Would it be possible to implement or document commands that can facilitate this?

Thank you for your work on this project!

Best,
Lin

[312hzeng]

hi Lin @yue2388253 , I wonder if you found a way to extend the symbolic_tensor_graph tool to generate traces for other models? thank you