_scaled_dot_product_flash_attention_for_cpu can't be rewrite in ReduceOpVariantsPass. Because of string not pattern.

[heshuju]

_scaled_dot_product_flash_attention_for_cpu can't be rewrite in ReduceOpVariantsPass.
Because of string not pattern.

torch-mlir/lib/Dialect/Torch/Transforms/ReduceOpVariants.cpp

Line 240 in 0c29ccf

"torch.aten._scaled_dot_product_flash_attention_for_cpu",

I found out the reason is:

torch-mlir/projects/jit_ir_common/csrc/jit_ir_importer/torch_to_mlir_utils.cpp

Line 541 in 0c29ccf

return createMlirOperationAtEnd(

Modify the code and it will work.

  return createMlirOperationAtEnd(
      appendToBlock, "torch.operator", loc, resultTypes, operands,
      toMlirNamedAttribute(
          "name", mlirStringAttrGet(context, toMlirStringRef(opName))));