github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-01-08 22:28:06 -05:00
Code Issues Packages Projects Releases Wiki Activity

Update PyTorch compatibility documentation

Browse Source
This commit is contained in:
Shao
2025-11-19 09:52:47 -07:00
parent bb692dfd84
commit 407a9d4cb0
1 changed files with 8 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
docs/compatibility/ml-compatibility/pytorch-compatibility.rst
View File

@@ -399,18 +399,19 @@ with ROCm.
**Note:** Only official release exists.
Key features and enhancements for PyTorch 2.8 with ROCm 7.1
Key features and enhancements for PyTorch 2.9 with ROCm 7.1.1
================================================================================
- Added OCP Micro-scaling Format (mx-fp8/mx-fp4) support for advanced precision training.
- MIOpen deep learning optimizations: Further optimized NHWC BatchNorm feature.
- `torch.backends.miopen.immediate` flag to toggle MIOpen Immediate Mode independently of
deterministic and benchmark settings, providing finer control over convolution execution.
- Added float8 support for the DeepSpeed extension, allowing for decreased
memory footprint and increased throughput in training and inference workloads.
- rocSOLVER now used for Cholesky inversion operations, providing improved numerical stability
and performance for linear algebra workloads.
- ``torch.nn.functional.scaled_dot_product_attention`` now calling optimized
flash attention kernel automatically.
- MI355X GPU testing enabled in CI.
Key features and enhancements for PyTorch 2.7/2.8 with ROCm 7.0
Key features and enhancements for PyTorch 2.7/2.8 with ROCm 7.1.1
================================================================================
- Enhanced TunableOp framework: Introduces ``tensorfloat32`` support for