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.rst

Browse Source
This commit is contained in:
Melantha-S
2025-11-19 15:54:04 -07:00
committed by GitHub
parent c33b9e3611
commit b566858909
1 changed files with 11 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -421,6 +421,17 @@ Key features and enhancements for PyTorch 2.9 with ROCm 7.1.1
- Added stream graph capture status validation in memory copy synchronization functions
Key features and enhancements for PyTorch 2.8 with ROCm 7.1
================================================================================
- MIOpen deep learning optimizations: Further optimized NHWC BatchNorm feature.
- Added float8 support for the DeepSpeed extension, allowing for decreased
memory footprint and increased throughput in training and inference workloads.
- ``torch.nn.functional.scaled_dot_product_attention`` now calling optimized
flash attention kernel automatically.
Key features and enhancements for PyTorch 2.7/2.8 with ROCm 7.0
================================================================================