turn off dynamic shape for torch.compile

pytorch / torchtitan

A native PyTorch Library for large model training

BSD 3-Clause "New" or "Revised" License

1.29k stars 115 forks source link

Closed wanchaol closed 2 months ago

wanchaol commented 2 months ago

as titled. This could make 1-D and 2-D works with the lastest main build. thanks @bdhirsh for all the fixes!

We should figure out why dynamic shape gets turned on as a follow up