onnxruntime
13d1a3c0 - [ROCm] add SkipLayerNorm vectorize Regular case (#12821)

Commit
3 years ago
[ROCm] add SkipLayerNorm vectorize Regular case (#12821) **Description**: Describe your changes. add SkipLayerNorm vectorize regular case 1. when hidden size <= 1024, SkipLayerNormTunable op can use both small case and regular case 2. when hidden size > 1024, SkipLayerNormTunable op can only use regular case. **Motivation and Context** - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here.
Author
Parents
Loading