We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 9316d7b commit 6eb5400Copy full SHA for 6eb5400
mlir/utils/performance/perfRunner.py
@@ -787,6 +787,12 @@ def getAttentionConfigurations(fileName):
787
oneConfig = line.strip()
788
for arg, value in zip(args, test_vector):
789
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
796
if oneConfig not in configs:
797
configs.append(oneConfig)
798
0 commit comments