gracefully skip plot generation

Author: stas00

network/benchmarks/all_reduce_bench.py

@@ -75,7 +75,6 @@
 from pathlib import Path
 import datetime
 import gc
-import matplotlib.pyplot as plt
 import os
 import signal
 import socket
@@ -112,6 +111,12 @@ def get_device_info():
 
 def plot(path, x, y, ranks):
 
+    try:
+        import matplotlib.pyplot as plt
+    except:
+        print("!!! Can't generate plot. Please run `pip install matplotlib` to enable plotting. !!!\n")
+        return
+
     plt.figure(dpi=500)
     plt.plot(x, y)
     plt.xlabel(f"Message size")

todo.md

@@ -0,0 +1,6 @@
+# TODO
+
+Also see [stabs](./stabs)
+
+- re-run all-reduce bench and update plot+table as the bench switched to KiB/MiB/etc.
+https://github.com/stas00/ml-engineering/tree/master/network/benchmarks#all_reduce-benchmark