diff --git a/mlir/utils/generate-test-checks.py b/mlir/utils/generate-test-checks.py index 14a790e6d0e6e..f80a1811f418c 100755 --- a/mlir/utils/generate-test-checks.py +++ b/mlir/utils/generate-test-checks.py @@ -229,7 +229,9 @@ def process_attribute_definition(line, attribute_namer): "// CHECK: #[[" + attribute_name + ":.+]] =" - + line[len(m.group(0)) :] + # The rest of the line may contain attribute references, + # so we have to process them. + + process_attribute_references(line[len(m.group(0)) :], attribute_namer) + "\n" ) return None