Code

[pull] main from NVIDIA:main#81

Code

Merged

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

+68-37

Commits

Commits on Feb 28, 2025

• Fix shape of new quantized tensor in make_like (#1515)

  Show description for 9588109
  ksivaman

  authored
  9588109

  Copy full SHA for 9588109

• Enforce PyTorch version 2.1 and run attention tests with torch.compile (#1516)

  Show description for 303c6d1
  ksivaman

  authored
  303c6d1

  Copy full SHA for 303c6d1