Update images to support dark mode (#7195)

pytorch · Jun 5, 2024 · cdb4aec · cdb4aec
1 parent 73346d7
commit cdb4aec
Show file tree

Hide file tree

Showing 6 changed files with 9 additions and 3 deletions.
diff --git a/docs/assets/IRgraph_markstep.png b/docs/assets/IRgraph_markstep.png
diff --git a/docs/assets/IRgraph_no_markstep.png b/docs/assets/IRgraph_no_markstep.png
diff --git a/docs/assets/pytorchXLA_flow.svg b/docs/assets/pytorchXLA_flow.svg
diff --git a/docs/assets/spmd_debug_1_light.png b/docs/assets/spmd_debug_1_light.png
diff --git a/docs/assets/spmd_debug_2_light.png b/docs/assets/spmd_debug_2_light.png
diff --git a/docs/spmd.md b/docs/spmd.md
@@ -501,7 +501,10 @@ xs.mark_sharding(t, mesh, ('x', 'y'))
 from torch_xla.distributed.spmd.debugging import visualize_tensor_sharding
 generated_table = visualize_tensor_sharding(t, use_color=False)
 ```
-![alt_text](assets/spmd_debug_1.png "visualize_tensor_sharding example on TPU v4-8(single-host)")
+<picture>
+  <source media="(prefers-color-scheme: dark)" srcset="assets/spmd_debug_1.png">
+  <img alt="visualize_tensor_sharding example on TPU v4-8(single-host)" src="assets/spmd_debug_1_light.png">
+</picture>
 
 - Code snippet used `visualize_sharding` and visualization result:
 
@@ -510,7 +513,10 @@ from torch_xla.distributed.spmd.debugging import visualize_sharding
 sharding = '{devices=[2,2]0,1,2,3}'
 generated_table = visualize_sharding(sharding, use_color=False)
 ```
-![alt_text](assets/spmd_debug_2.png "visualize_sharding example on TPU v4-8(single-host")
+<picture>
+  <source media="(prefers-color-scheme: dark)" srcset="assets/spmd_debug_2.png">
+  <img alt="visualize_sharding example on TPU v4-8(single-host)" src="assets/spmd_debug_2_light.png">
+</picture>
 
 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`.