From 1fe5fc9b3e6e0b68cddb7e1a166e2606221161fc Mon Sep 17 00:00:00 2001
From: "chaofeng.wcf" <chaofeng.wcf@alibaba-inc.com>
Date: Thu, 6 Mar 2025 11:14:08 +0800
Subject: [PATCH] collect all modules info

---
 lib/utils.py | 22 ++++++++++------------
 1 file changed, 10 insertions(+), 12 deletions(-)

diff --git a/lib/utils.py b/lib/utils.py
index 3afaa67..070bbad 100644
--- a/lib/utils.py
+++ b/lib/utils.py
@@ -649,18 +649,16 @@ def save_device_info(comp_insts, dstree, args, logdir):
                 continue
             modules.append(mod)
         driver_info = {}
-        for drv in dstree.drivers:
-            drv = drv.strip()
-            if drv in modules:
-                _, output = run_local_cmd('modinfo %s' % drv, shell=True)
-                info = {drv: {}}
-                for line in output.split('\n'):
-                    if ':' not in line:
-                        continue
-                    key, val = line.split(':', 1)
-                    val = val.strip()
-                    info[drv].update({key: val})
-                driver_info.update(info)
+        for drv in modules:
+            _, output = run_local_cmd('modinfo %s' % drv, shell=True)
+            info = {drv: {}}
+            for line in output.split('\n'):
+                if ':' not in line:
+                    continue
+                key, val = line.split(':', 1)
+                val = val.strip()
+                info[drv].update({key: val})
+            driver_info.update(info)
         return driver_info
 
     all_info = {}
-- 
Gitee