From d915d82ac4876a487c7eadca0cc4c028cb207bec Mon Sep 17 00:00:00 2001 From: Corey adams <6619961+coreyjadams@users.noreply.github.com> Date: Fri, 22 May 2026 16:51:23 -0500 Subject: [PATCH] Update README.md --- examples/minimal/ShardTensorExamples/6_ring_attention/README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/examples/minimal/ShardTensorExamples/6_ring_attention/README.md b/examples/minimal/ShardTensorExamples/6_ring_attention/README.md index 4877f4a144..37c8738368 100644 --- a/examples/minimal/ShardTensorExamples/6_ring_attention/README.md +++ b/examples/minimal/ShardTensorExamples/6_ring_attention/README.md @@ -37,6 +37,8 @@ torchrun --nproc-per-node 4 benchmark_sharded_attention.py \ After collecting JSON results in `results/`, generate scaling plots: +> Note! To run the scaling plot, you will need matplotlib installed. `pip install matplotlib`. + ```bash python plot_scaling_results.py ```