Code

[pull] main from NVIDIA:main#78

Code

Merged

pull[bot] merged 3 commits intophu0ngng:mainphu0ngng/TransformerEngine:mainfrom NVIDIA:mainNVIDIA/TransformerEngine:mainCopy head branch name to clipboardFeb 25, 2025

+21-6

Commits

Commits on Feb 25, 2025

• Minor fixes for attention (#1504)

  Show description for 8744188
  

  cyanguwa

  and

  pre-commit-ci[bot]

  authored
  8744188

  Copy full SHA for 8744188

• Fix a crash in NeMo 2.0 during module._apply(lambda t: t.cpu()) (#1502)

  Show description for 9351a17
  

  guyueh1

  and

  timmoon10

  authored
  9351a17

  Copy full SHA for 9351a17

• Adding remove_caches API to Float8Tensor class (#1425)

  Show description for 94c9291
  

  3 peopleauthored
  94c9291

  Copy full SHA for 94c9291