From 3e6b0ab388b07be8ee53e615d27bb14a69bbe82f Mon Sep 17 00:00:00 2001
From: colramos-amd <colramos@amd.com>
Date: Mon, 17 Jul 2023 13:12:22 -0500
Subject: [PATCH] Filter additional ops in gen_counter_list fucn

Signed-off-by: colramos-amd <colramos@amd.com>


[ROCm/rocprofiler-compute commit: fd55a698057929b2cc9bceb47ec5dac9ea941e18]
---
 .../src/omniperf_analyze/utils/parser.py                  | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/projects/rocprofiler-compute/src/omniperf_analyze/utils/parser.py b/projects/rocprofiler-compute/src/omniperf_analyze/utils/parser.py
index 0328d7aa84..bff3314b2a 100644
--- a/projects/rocprofiler-compute/src/omniperf_analyze/utils/parser.py
+++ b/projects/rocprofiler-compute/src/omniperf_analyze/utils/parser.py
@@ -340,6 +340,8 @@ def gen_counter_list(formula):
         "RW": None,
         "GIOP": None,
         "GFLOPs": None,
+        "CONCAT": None,
+        "MOD": None,
     }
 
     built_in_counter = [
@@ -362,6 +364,12 @@ def gen_counter_list(formula):
         tree = ast.parse(
             formula.replace("$normUnit", "SQ_WAVES")
             .replace("$denom", "SQ_WAVES")
+            .replace(
+                "$numActiveCUs",
+                "TO_INT(MIN((((ROUND(AVG(((4 * SQ_BUSY_CU_CYCLES) / GRBM_GUI_ACTIVE)), \
+              0) / $maxWavesPerCU) * 8) + MIN(MOD(ROUND(AVG(((4 * SQ_BUSY_CU_CYCLES) \
+              / GRBM_GUI_ACTIVE)), 0), $maxWavesPerCU), 8)), $numCU))",
+            )
             .replace("$", "")
         )
         for node in ast.walk(tree):