Code

[PyTorch] cuda graph support#575

Code

Merged

ksivaman merged 8 commits intoNVIDIA:mainNVIDIA/TransformerEngine:mainfrom ksivaman:fp8_cuda_graphsksivaman/TransformerEngine-1:fp8_cuda_graphsCopy head branch name to clipboardApr 12, 2024

+2,196-789

Commits

Commits on Apr 10, 2024

• FP8 cuda graphs

  Show description for 31dc133
  

  3 peoplecommitted
  31dc133

  Copy full SHA for 31dc133

• Fix numerics

  Show description for 19d0bd4
  ksivaman

  committed
  19d0bd4

  Copy full SHA for 19d0bd4

• exclude torch compile from numerics tests

  Show description for 86e9162
  ksivaman

  committed
  86e9162

  Copy full SHA for 86e9162

• More numerics fixes

  Show description for 4cbda4a
  ksivaman

  committed
  4cbda4a

  Copy full SHA for 4cbda4a

• Merge branch 'main' into fp8_cuda_graphs

  ksivaman

  authored
  020a870

  Copy full SHA for 020a870

Commits on Apr 11, 2024

• Fix tests

  Show description for 798ea11
  ksivaman

  committed
  798ea11

  Copy full SHA for 798ea11

Commits on Apr 12, 2024

• Fix CI

  Show description for 3c50a17
  ksivaman

  committed
  3c50a17

  Copy full SHA for 3c50a17

• rm fusion from unfused path

  Show description for 3bdadcf
  ksivaman

  committed
  3bdadcf

  Copy full SHA for 3bdadcf