src/targets/gpu/prefuse_ops.cpp · 5af9aac00a02daa1f2c351bcdc0063e00f30d89a · gaoqiong / MIGraphX

Update matcher to better match layernorm fusions in other models (#1548) · 3e58b1e4

Paul Fultz II authored Feb 06, 2023



* Fuse layernorm with different patterns
* Only match when using the last axis
Co-authored-by: kahmed10 <15948690+kahmed10@users.noreply.github.com>
Co-authored-by: kahmed10 <15948690+kahmed10@users.noreply.github.com>

3e58b1e4

prefuse_ops.cpp 3.98 KB

Replace prefuse_ops.cpp