benchmark
e11cbaa1 - Log graph_node_shape to dynamo_compile (#160556)

Commit
127 days ago
Log graph_node_shape to dynamo_compile (#160556) Summary: This PR adds the dynamo graph node shape logging to dynamo compile. Also added unit tests to check if correct graph node shape is being logged. X-link: https://github.com/pytorch/pytorch/pull/160556 Approved by: https://github.com/masnesral, https://github.com/jingsh Reviewed By: clee2000 Differential Revision: D80292167 fbshipit-source-id: 479a9d19fc2676064f6b58b12836efe7f7f99822
Author
Parents
Loading