onnxruntime
2092bebc - Fix transformer layer detection for recompute (#20106)

Commit
1 year ago
Fix transformer layer detection for recompute (#20106) ### Fix transformer layer detection for recompute Originally logic miss detecting the layer boudary node in Mistral model. This PR simplifies the searching, by using more strong pattern's match, to make sure it is flexible enough to cover different transformer variants. Also add a UT. Add a warning when user enable layerwise recompute but no layer boudary nodes are found.
Author
Parents
Loading