diff --git a/src/omniperf b/src/omniperf
index 83722d7f8..073ac2744 100755
--- a/src/omniperf
+++ b/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:
diff --git a/src/soc_params/mi300.csv b/src/soc_params/mi300.csv
index 2098f82fd..97a9ddc9d 100644
--- a/src/soc_params/mi300.csv
+++ b/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
diff --git a/src/utils/specs.py b/src/utils/specs.py
index dc92a972b..a24a7664c 100644
--- a/src/utils/specs.py
+++ b/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