Skip to content

Releases: ROCm/apex

ROCm-Apex v0.3.2

04 May 15:17
Compare
Choose a tag to compare

Critical bug fixes for ROCm4.0

ROCm-Apex v0.3.1

18 Dec 17:42
7eed38a
Compare
Choose a tag to compare

This is a bugfix release that addresses the following issue:

  • Fix LayerNorm op on ROCm (#36)

ROCm-Apex v0.3

25 Sep 00:22
e9c43d6
Compare
Choose a tag to compare

The following are the new extensions that are supported in this release.

  • SyncBatchnorm extension.
  • MLP CUDA extension
  • Apex Contrib library support - Deprecated FusedLAMB and Deprecated FusedAdam.
  • Apex Contrib library support - Xentropy extension.

ROCm-Apex v0.2

29 May 21:39
aea81c0
Compare
Choose a tag to compare

ROCm Apex v0.2 Release Notes:

  • BFloat16 Mixed Precision Training support with single GPU.
    • Introduced new Optimization levels - O4 and O5 for BFloat16 training.
    • BFloat16 support for FusedOptimizers
    • BFloat16 support for FusedLayerNorm.
  • Performance Improvements.
  • Test Infrastructure and CI setup on ROCm.

Initial support of Apex on ROCm.

15 May 01:03
02a5274
Compare
Choose a tag to compare

The following features are currently supported for Apex on ROCm:

  • Apex AMP - FP16
  • Extensions - Multi Tensor Apply, FusedLayerNorm, FusedOptimizers,
  • apex.DDP
  • Sparse Tensor support for Loss Scaling (Dynamic) [ New Feature].