[mlir][linalg] Add script to update the LinalgNamedStructuredOps.yaml. nfc

[stellaraccident]

Summary

Also adds banners to the files with update instructions.

[stellaraccident]

I'm surprised we didn't have something like this already. Should be uncontroversial. Please post-review.

mlir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOps.yaml

-# Auto-generated file. Do not edit!
+### AUTOGENERATED from core_named_ops.py
+### To regenerate, run: bin/update_core_linalg_named_ops.sh
 --- !LinalgOpConfig
 metadata: !LinalgOpMetadata
   name: matmul
   cpp_class_name: MatmulOp
   doc: |-
     Performs a matrix multiplication of two 2D inputs.
 
     Numeric casting is performed on the operands to the inner multiply, promoting
value: !ScalarExpression
                 fn_name: exp
                 operands:
                 - !ScalarExpression
                   symbolic_cast:
                     type_var: U
                     operands:
                     - !ScalarExpression
                       scalar_arg: I
-

mlir/python/mlir/dialects/linalg/opdsl/dump_oplib.py

for attr_name, value in m.__dict__.items():
           linalg_config = LinalgOpConfig.from_linalg_op_def(value.model)
         except Exception as e:
           raise ValueError(
               f"Could not create LinalgOpConfig from {value.model}") from e
         configs.extend(linalg_config)
 
   # Print.
   if args.format == "yaml":
-    print("# Auto-generated file. Do not edit!")
     print(yaml_dump_all(configs))
   elif args.format == "repr":
     for config in configs:
       print(repr(config))
 
 
 if __name__ == "__main__":
   main(create_arg_parser().parse_args())

mlir/tools/mlir-linalg-ods-gen/CMakeLists.txt

if (${MLIR_LINALG_ODS_YAML_GEN} STREQUAL "mlir-linalg-ods-yaml-gen")
     add_custom_target(mlir-linalg-ods-yaml-gen-host DEPENDS ${MLIR_LINALG_ODS_YAML_GEN_EXE})
     set(MLIR_LINALG_ODS_YAML_GEN_TARGET mlir-linalg-ods-yaml-gen-host DEPENDS PARENT_SCOPE)
 
     if(NOT LLVM_BUILD_UTILS)
       set_target_properties(mlir-linalg-ods-yaml-gen PROPERTIES EXCLUDE_FROM_ALL ON)
     endif()
   endif()
 endif()
+
+configure_file(
+  update_core_linalg_named_ops.sh.in
+  ${MLIR_TOOLS_DIR}/update_core_linalg_named_ops.sh)

mlir/tools/mlir-linalg-ods-gen/update_core_linalg_named_ops.sh.in

+#!/bin/bash
+# Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+# See https://llvm.org/LICENSE.txt for license information.
+# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
+# Updates the LinalgStructuredOps.yaml file based on the
+# core_named_ops.py source of truth.
+
+set -eu -o errtrace
+shopt -s inherit_errexit
+
+mlir_source_dir="@MLIR_SOURCE_DIR@"
+python_exe="@Python3_EXECUTABLE@"
+mlir_binary_dir="@MLIR_BINARY_DIR@"
+python_package_dir="$mlir_binary_dir/python_packages/mlir_core"
+
+if ! [[ -d "$python_package_dir" ]]; then
+  echo "ERROR: Python bindings not found. Build with -DMLIR_ENABLE_BINDINGS_PYTHON=ON"
+  exit 1
+fi
+
+dest_file="$mlir_source_dir/include/mlir/Dialect/Linalg/IR/LinalgNamedStructuredOps.yaml"
+echo "Updating ops in $dest_file"
+
+export PYTHONPATH="$python_package_dir"
+OUTPUT="$(
+  echo "### AUTOGENERATED from core_named_ops.py" && \
+  echo "### To regenerate, run: bin/update_core_linalg_named_ops.sh" && \
+  "$python_exe" -m mlir.dialects.linalg.opdsl.dump_oplib .ops.core_named_ops \
+)"
+echo "$OUTPUT" > "$dest_file"
+echo "Success."