torch.ops.aten._flash_attention_forward support #2344
Description
🐛 Describe the bug
Cases:
op_ut,third_party.torch-xpu-ops.test.xpu.test_meta_xpu.TestMetaXPU,test_meta_outplace_torch_ops_aten__flash_attention_forward_xpu_float16
Versions
main
{{ message }}