From b566858909ec80d90b1ad671c310380461906458 Mon Sep 17 00:00:00 2001 From: Melantha-S <95049630+Melantha-S@users.noreply.github.com> Date: Wed, 19 Nov 2025 15:54:04 -0700 Subject: [PATCH] Update pytorch-compatibility.rst --- .../ml-compatibility/pytorch-compatibility.rst | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/docs/compatibility/ml-compatibility/pytorch-compatibility.rst b/docs/compatibility/ml-compatibility/pytorch-compatibility.rst index 08c524067..0489817db 100644 --- a/docs/compatibility/ml-compatibility/pytorch-compatibility.rst +++ b/docs/compatibility/ml-compatibility/pytorch-compatibility.rst @@ -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 ================================================================================