1. 21 Jun, 2023 1 commit
  2. 04 Jan, 2023 1 commit
    • Yifan Xiong's avatar
      Benchmarks - Support FP8 in BERT models (#446) · 5197cdf5
      Yifan Xiong authored
      Support FP8 in PyTorch BERT models:
      
      * add fp8 hybrid/e4m3/e5m2 in precision arguments
      * build BERT encoders with `te.TransformerLayer` to repalce
      `transformers.BertModel`
      * wrap forward steps with fp8 autocast
      5197cdf5
  3. 21 Oct, 2021 1 commit
  4. 21 Jun, 2021 1 commit
  5. 29 Mar, 2021 1 commit
  6. 24 Feb, 2021 1 commit