search

microsoft / onnxruntime-extensions

onnxruntime-extensions: A specialized pre- and post- processing library for ONNX Runtime
MIT License
295 stars 80 forks source link

sync to flash attention kernel 2.5.9 and add document of how to write custom op #757

Open jslhcl opened 2 days ago

jslhcl commented 2 days ago

sync to flash attention kernel 2.5.9 and add document of how to write custom op