Skip to content

Commit 6eb5400

Browse files
authored
perfRunner: skip f32 attention kernels on Navi (#1902)
Signed-off-by: Djordje Antic <[email protected]>
1 parent 9316d7b commit 6eb5400

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)