Code

Fp8 model init factory#880

Code

Draft

sudhakarsingh27 wants to merge 2 commits intoNVIDIA:mainNVIDIA/TransformerEngine:mainfrom sudhakarsingh27:fp8_model_init_factorysudhakarsingh27/TransformerEngine:fp8_model_init_factoryCopy head branch name to clipboard

+13

Commits

Commits on May 30, 2024

• bake fp8_model_init into the layer

  sudhakarsingh27

  committed
  5bdfa6e

  Copy full SHA for 5bdfa6e

• bake fp8_model_init into module init

  sudhakarsingh27

  committed
  5530a8a

  Copy full SHA for 5530a8a