Improved SoC detection for gfx940

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

src/omniperf

@@ -220,6 +220,8 @@ def gen_sysinfo(workload_name, workload_dir, ip_blocks, app_cmd, skip_roof):
         param += ["32", "32", "mi200", "56", str(hbmBW)]
         if not skip_roof:
             blocks.append("roofline")
+    elif mspec.GPU == "gfx940":
+        param += ["16", "32", "mi300", "56", str(hbmBW)]
 
     # ip block info
     if ip_blocks == None:

src/soc_params/mi300.csv

@@ -1,2 +1,2 @@
-name,numSE,numCU,numSIMD,numWavesPerCU,numSQC,L2Banks,Freq,mclk
-mi300,8,38,4,32,56,32,1950,1300
+name,numSE,numCU,numSIMD,numWavesPerCU,numSQC,L2Banks,LDSBanks,Freq,mclk
+mi300,8,38,4,32,56,16,32,1950,1300

src/utils/specs.py

@@ -89,9 +89,13 @@ def gpuinfo():
     rocminfo = run(["rocminfo"]).split("\n")
 
     for idx1, linetext in enumerate(rocminfo):
-        gpu_id = search(r"^\s*Name\s*:\s+ ([a-zA-Z0-9]+)\s*$", linetext)
+        gpu_id = search(r"^\s*Name\s*:\s+ ([a-zA-Z0-9:+]+)\s*$", linetext)
         if gpu_id in gpu_list:
             break
+        print(str(gpu_id)[:6])
+        if str(gpu_id)[:6] in gpu_list:
+            gpu_id = str(gpu_id)[:6]
+            break
 
     if not gpu_id in gpu_list:
         return None, None, None, None, None, None, None, None, None, None