Code

Support FP16 for user buffer#690

Code

Merged

ksivaman merged 6 commits intoNVIDIA:mainNVIDIA/TransformerEngine:mainfrom lhb8125:add_dtype_for_userbuflhb8125/TransformerEngine:add_dtype_for_userbufCopy head branch name to clipboardMar 7, 2024

+5-2

Commits

Commits on Feb 29, 2024

• add_dtype_for_userbuf

  Show description for 34c1322
  

  Hongbin Liu

  committed
  34c1322

  Copy full SHA for 34c1322

Commits on Mar 6, 2024

• Update transformer_engine/pytorch/module/base.py

  Show description for a4c8ad9
  timmoon10

  authored
  a4c8ad9

  Copy full SHA for a4c8ad9

• Merge branch 'main' into add_dtype_for_userbuf

  timmoon10

  authored
  835e5f3

  Copy full SHA for 835e5f3

• Fix syntax

  Show description for 20298d8
  ksivaman

  authored
  20298d8

  Copy full SHA for 20298d8

• Merge branch 'main' into add_dtype_for_userbuf

  ksivaman

  authored
  2dec56f

  Copy full SHA for 2dec56f

Commits on Mar 7, 2024

• Fix lint

  Show description for cad2c2a
  ksivaman

  committed
  cad2c2a

  Copy full SHA for cad2c2a