Skip to content

Commit 7a50d7f

Browse files
authored
Merge branch 'develop' into upstream_merge_55
2 parents 74eb560 + 6eb5400 commit 7a50d7f

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

mlir/utils/performance/perfRunner.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -787,6 +787,12 @@ def getAttentionConfigurations(fileName):
787787
oneConfig = line.strip()
788788
for arg, value in zip(args, test_vector):
789789
oneConfig = f"{arg} {value} {oneConfig}"
790+
791+
# Check for valid dtypes
792+
foundDtype = re.search(r"-t\s+(\w+)", oneConfig)
793+
if not foundDtype or foundDtype.group(1) not in DATA_TYPES_ATTENTION:
794+
continue
795+
790796
if oneConfig not in configs:
791797
configs.append(oneConfig)
792798

0 commit comments

Comments
 (0)