From e97c745ba57234a3ced9f3a851e3304569f0580b Mon Sep 17 00:00:00 2001 From: colramos425 Date: Tue, 17 Jan 2023 10:26:00 -0600 Subject: [PATCH] Catch any import errors from matplotlib #62 Signed-off-by: colramos425 [ROCm/rocprofiler-compute commit: c3e3c9982b33fa7da7d7e98c6008fe9526316354] --- .../src/utils/plot_roofline.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/projects/rocprofiler-compute/src/utils/plot_roofline.py b/projects/rocprofiler-compute/src/utils/plot_roofline.py index a6e9d2da23..f421dc972e 100644 --- a/projects/rocprofiler-compute/src/utils/plot_roofline.py +++ b/projects/rocprofiler-compute/src/utils/plot_roofline.py @@ -21,14 +21,22 @@ ################################################################################ from linecache import cache -import subprocess -from operator import sub import os import sys from pathlib import Path import numpy -import matplotlib.pyplot as plt +import matplotlib + +try: + + import matplotlib.pyplot as plt +except ImportError: + # other non-interactive options: + # cairo, pdf, pgf, ps, svg, template + matplotlib.use("agg", force=True) + import matplotlib.pyplot as plt + from matplotlib.pyplot import get, text from math import log, pi, sqrt import pandas as pd @@ -511,8 +519,6 @@ def plot_application(inputs, verbose): print(intensities) - # fig, ax = plt.subplots() - plotted_spots = [] labels = [] for i in intensities: