Code

Enable torch.compile#322

Code

Draft

tohtana wants to merge 3 commits intomainmicrosoft/Megatron-DeepSpeed:mainfrom tohtana/enable_compilemicrosoft/Megatron-DeepSpeed:tohtana/enable_compileCopy head branch name to clipboard

+442-3

Commits

Commits on Dec 23, 2023

• add graph breaks for compile

  tohtana

  committed
  fa9623b

Commits on Dec 25, 2023

• add example using torch.compile

  tohtana

  committed
  4158f0e

Commits on Dec 28, 2023

• update example to use torch.compile

  tohtana

  committed
  62242ee