From 66b4d640bee4ac3c21161b7130054bc85200489a Mon Sep 17 00:00:00 2001
From: Juee Himalbhai Desai <juee.himalbhai.desai@intel.com>
Date: Fri, 15 Mar 2024 16:46:16 -0700
Subject: [PATCH] Intel/CI: Add OneCCL DSA + shm summary

- Update oneccl summarizer to support DSA + shm summary.
- Add the logic to check whether DSA was actually used or not.

Signed-off-by: Juee Himalbhai Desai <juee.himalbhai.desai@intel.com>
---
 contrib/intel/jenkins/summary.py | 52 ++++++++++++++++++++++++++++----
 1 file changed, 46 insertions(+), 6 deletions(-)

diff --git a/contrib/intel/jenkins/summary.py b/contrib/intel/jenkins/summary.py
index 8488334d12c..fa8a7a37f1a 100755
--- a/contrib/intel/jenkins/summary.py
+++ b/contrib/intel/jenkins/summary.py
@@ -395,12 +395,7 @@ def __init__(self, logger, log_dir, prov, file_name, stage_name):
         self.file_path = os.path.join(self.log_dir, self.file_name)
         self.exists = os.path.exists(self.file_path)
         self.name = 'no_test'
-
-    def read_file(self):
-        with open(self.file_path, 'r') as log_file:
-            self.fast_forward(log_file)
-            for line in log_file:
-                self.check_line(line)
+        self.trace = False
 
     def check_name(self, line):
         #OneCCL GPU tests:
@@ -430,6 +425,45 @@ def check_fail(self, line):
             self.fails += 1
             self.failed_tests.append(self.name)
 
+    def check_trace(self, line):
+        if not self.trace:
+            cmd_count = 0
+            faults_count = 0
+            if ("user to sar buffer" in line):
+                tokens = line.split(' ')
+                for i in range(0, len(tokens)):
+                    if 'cmd' in tokens[i]:
+                        cmd_count += int(tokens[i + 1])
+                    if 'faults' in tokens[i]:
+                        faults_count += int(tokens[i + 1])
+                if (cmd_count > 0 or faults_count > 0):
+                    self.trace = True
+
+    def check_line(self, line):
+        self.check_name(line)
+        if (self.name != 'no_test'):
+            self.check_pass(line)
+            self.check_fail(line)
+            if ('DSA' in self.file_name):
+                self.check_trace(line.lower())
+
+    def read_file(self):
+        with open(self.file_path, 'r') as log_file:
+            self.fast_forward(log_file)
+            for line in log_file:
+                self.check_line(line)
+
+    def summarize(self):
+        if not self.exists:
+            return 0
+
+        self.read_file()
+        self.print_results()
+        if ('DSA' in self.file_name and not self.trace):
+            exit("Expected: DSA to run. Actual: DSA Not Run")
+
+        return int(self.fails)
+
 class ShmemSummarizer(Summarizer):
     def __init__(self, logger, log_dir, prov, file_name, stage_name):
         super().__init__(logger, log_dir, prov, file_name, stage_name)
@@ -819,6 +853,12 @@ def summarize_items(summary_item, logger, log_dir, mode):
                 f"{prov} dsa fabtests {mode}"
             ).summarize()
             err += ret if ret else 0
+            ret = OnecclSummarizer(
+                logger, log_dir, 'oneCCL',
+                f'oneCCL_DSA_shm_oneccl_{mode}',
+                f'oneCCL DSA {prov} {mode}'
+            ).summarize()
+            err += ret if ret else 0
 
     if summary_item == 'dmabuf' or summary_item == 'all':
         for prov in ['verbs-rxm']: