Skip to content

Commit cdb4aec

Browse files
authored
Update images to support dark mode (#7195)
1 parent 73346d7 commit cdb4aec

6 files changed

+9
-3
lines changed

docs/assets/IRgraph_markstep.png

6.78 KB
Loading

docs/assets/IRgraph_no_markstep.png

5.43 KB
Loading

docs/assets/pytorchXLA_flow.svg

Lines changed: 1 addition & 1 deletion
Loading

docs/assets/spmd_debug_1_light.png

98.3 KB
Loading

docs/assets/spmd_debug_2_light.png

100 KB
Loading

docs/spmd.md

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -501,7 +501,10 @@ xs.mark_sharding(t, mesh, ('x', 'y'))
501501
from torch_xla.distributed.spmd.debugging import visualize_tensor_sharding
502502
generated_table = visualize_tensor_sharding(t, use_color=False)
503503
```
504-
![alt_text](assets/spmd_debug_1.png "visualize_tensor_sharding example on TPU v4-8(single-host)")
504+
<picture>
505+
<source media="(prefers-color-scheme: dark)" srcset="assets/spmd_debug_1.png">
506+
<img alt="visualize_tensor_sharding example on TPU v4-8(single-host)" src="assets/spmd_debug_1_light.png">
507+
</picture>
505508

506509
- Code snippet used `visualize_sharding` and visualization result:
507510

@@ -510,7 +513,10 @@ from torch_xla.distributed.spmd.debugging import visualize_sharding
510513
sharding = '{devices=[2,2]0,1,2,3}'
511514
generated_table = visualize_sharding(sharding, use_color=False)
512515
```
513-
![alt_text](assets/spmd_debug_2.png "visualize_sharding example on TPU v4-8(single-host")
516+
<picture>
517+
<source media="(prefers-color-scheme: dark)" srcset="assets/spmd_debug_2.png">
518+
<img alt="visualize_sharding example on TPU v4-8(single-host)" src="assets/spmd_debug_2_light.png">
519+
</picture>
514520

515521
You could use these examples on TPU/GPU/CPU single-host and modify it to run on multi-host. And you could modify it to sharding-style `tiled`, `partial_replication` and `replicated`.
516522

0 commit comments

Comments
 (0)