From 622d9b62270712dd25d09e12616fb9c6279b3983 Mon Sep 17 00:00:00 2001 From: coleramos425 Date: Tue, 2 May 2023 15:00:42 -0500 Subject: [PATCH] Add support for Std Dev in CLI parsing logic (#104) Signed-off-by: coleramos425 --- src/omniperf_analyze/utils/parser.py | 9 ++++++++- src/omniperf_analyze/utils/schema.py | 1 + 2 files changed, 9 insertions(+), 1 deletion(-) diff --git a/src/omniperf_analyze/utils/parser.py b/src/omniperf_analyze/utils/parser.py index ace633e35a..406b66b156 100644 --- a/src/omniperf_analyze/utils/parser.py +++ b/src/omniperf_analyze/utils/parser.py @@ -88,6 +88,7 @@ supported_call = { # simple aggr "AVG": "to_avg", "MEDIAN": "to_median", + "STD": "to_std", # functions apply to whole column of df or a single value "TO_INT": "to_int", # Support the below with 2 inputs @@ -134,6 +135,12 @@ def to_median(a): return a.median() else: raise Exception("to_median: unsupported type.") + +def to_std(a): + if isinstance(a, pd.core.series.Series): + return a.std() + else: + raise Exception("to_std: unsupported type.") def to_int(a): @@ -731,7 +738,7 @@ def build_comparable_columns(time_unit): Build comparable columns/headers for display """ comparable_columns = schema.supported_field - top_stat_base = ["Count", "Sum", "Mean", "Median"] + top_stat_base = ["Count", "Sum", "Mean", "Median", "Standard Deviation"] for h in top_stat_base: comparable_columns.append(h + "(" + time_unit + ")") diff --git a/src/omniperf_analyze/utils/schema.py b/src/omniperf_analyze/utils/schema.py index 13d930916d..2c6d7048d7 100644 --- a/src/omniperf_analyze/utils/schema.py +++ b/src/omniperf_analyze/utils/schema.py @@ -81,6 +81,7 @@ supported_field = [ "Count", "Mean", "Pct", + "Std Dev", # Special keywords for Memory chart "Alias", # Special keywords for L2 channel